2dc8b8c158b2c784c78649906591b94811143971
[openssl.git] / crypto / evp / evp_test.c
1 /* Written by Ben Laurie, 2001 */
2 /*
3  * Copyright (c) 2001 The OpenSSL Project.  All rights reserved.
4  *
5  * Redistribution and use in source and binary forms, with or without
6  * modification, are permitted provided that the following conditions
7  * are met:
8  *
9  * 1. Redistributions of source code must retain the above copyright
10  *    notice, this list of conditions and the following disclaimer. 
11  *
12  * 2. Redistributions in binary form must reproduce the above copyright
13  *    notice, this list of conditions and the following disclaimer in
14  *    the documentation and/or other materials provided with the
15  *    distribution.
16  *
17  * 3. All advertising materials mentioning features or use of this
18  *    software must display the following acknowledgment:
19  *    "This product includes software developed by the OpenSSL Project
20  *    for use in the OpenSSL Toolkit. (http://www.openssl.org/)"
21  *
22  * 4. The names "OpenSSL Toolkit" and "OpenSSL Project" must not be used to
23  *    endorse or promote products derived from this software without
24  *    prior written permission. For written permission, please contact
25  *    openssl-core@openssl.org.
26  *
27  * 5. Products derived from this software may not be called "OpenSSL"
28  *    nor may "OpenSSL" appear in their names without prior written
29  *    permission of the OpenSSL Project.
30  *
31  * 6. Redistributions of any form whatsoever must retain the following
32  *    acknowledgment:
33  *    "This product includes software developed by the OpenSSL Project
34  *    for use in the OpenSSL Toolkit (http://www.openssl.org/)"
35  *
36  * THIS SOFTWARE IS PROVIDED BY THE OpenSSL PROJECT ``AS IS'' AND ANY
37  * EXPRESSED OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
38  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
39  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE OpenSSL PROJECT OR
40  * ITS CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
41  * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
42  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
43  * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
44  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
45  * STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
46  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED
47  * OF THE POSSIBILITY OF SUCH DAMAGE.
48  */
49
50 #include <stdio.h>
51 #include <string.h>
52
53 #include "../e_os.h"
54
55 #include <openssl/evp.h>
56 #ifndef OPENSSL_NO_ENGINE
57 #include <openssl/engine.h>
58 #endif
59 #include <openssl/err.h>
60 #include <openssl/conf.h>
61
62 static void hexdump(FILE *f,const char *title,const unsigned char *s,int l)
63     {
64     int n=0;
65
66     fprintf(f,"%s",title);
67     for( ; n < l ; ++n)
68         {
69         if((n%16) == 0)
70             fprintf(f,"\n%04x",n);
71         fprintf(f," %02x",s[n]);
72         }
73     fprintf(f,"\n");
74     }
75
76 static int convert(unsigned char *s)
77     {
78     unsigned char *d;
79
80     for(d=s ; *s ; s+=2,++d)
81         {
82         unsigned int n;
83
84         if(!s[1])
85             {
86             fprintf(stderr,"Odd number of hex digits!");
87             EXIT(4);
88             }
89         sscanf((char *)s,"%2x",&n);
90         *d=(unsigned char)n;
91         }
92     return s-d;
93     }
94
95 static char *sstrsep(char **string, const char *delim)
96     {
97     char isdelim[256];
98     char *token = *string;
99
100     if (**string == 0)
101         return NULL;
102
103     memset(isdelim, 0, 256);
104     isdelim[0] = 1;
105
106     while (*delim)
107         {
108         isdelim[(unsigned char)(*delim)] = 1;
109         delim++;
110         }
111
112     while (!isdelim[(unsigned char)(**string)])
113         {
114         (*string)++;
115         }
116
117     if (**string)
118         {
119         **string = 0;
120         (*string)++;
121         }
122
123     return token;
124     }
125
126 static unsigned char *ustrsep(char **p,const char *sep)
127     { return (unsigned char *)sstrsep(p,sep); }
128
129 static int test1_exit(int ec)
130         {
131         EXIT(ec);
132         return(0);              /* To keep some compilers quiet */
133         }
134
135 static void test1(const EVP_CIPHER *c,const unsigned char *key,int kn,
136                   const unsigned char *iv,int in,
137                   const unsigned char *plaintext,int pn,
138                   const unsigned char *ciphertext,int cn,
139                   int encdec)
140     {
141     EVP_CIPHER_CTX ctx;
142     unsigned char out[4096];
143     int outl,outl2;
144
145     printf("Testing cipher %s%s\n",EVP_CIPHER_name(c),
146            (encdec == 1 ? "(encrypt)" : (encdec == 0 ? "(decrypt)" : "(encrypt/decrypt)")));
147     hexdump(stdout,"Key",key,kn);
148     if(in)
149         hexdump(stdout,"IV",iv,in);
150     hexdump(stdout,"Plaintext",plaintext,pn);
151     hexdump(stdout,"Ciphertext",ciphertext,cn);
152     
153     if(kn != c->key_len)
154         {
155         fprintf(stderr,"Key length doesn't match, got %d expected %d\n",kn,
156                 c->key_len);
157         test1_exit(5);
158         }
159     EVP_CIPHER_CTX_init(&ctx);
160     if (encdec != 0)
161         {
162         if(!EVP_EncryptInit_ex(&ctx,c,NULL,key,iv))
163             {
164             fprintf(stderr,"EncryptInit failed\n");
165             ERR_print_errors_fp(stderr);
166             test1_exit(10);
167             }
168         EVP_CIPHER_CTX_set_padding(&ctx,0);
169
170         if(!EVP_EncryptUpdate(&ctx,out,&outl,plaintext,pn))
171             {
172             fprintf(stderr,"Encrypt failed\n");
173             ERR_print_errors_fp(stderr);
174             test1_exit(6);
175             }
176         if(!EVP_EncryptFinal_ex(&ctx,out+outl,&outl2))
177             {
178             fprintf(stderr,"EncryptFinal failed\n");
179             ERR_print_errors_fp(stderr);
180             test1_exit(7);
181             }
182
183         if(outl+outl2 != cn)
184             {
185             fprintf(stderr,"Ciphertext length mismatch got %d expected %d\n",
186                     outl+outl2,cn);
187             test1_exit(8);
188             }
189
190         if(memcmp(out,ciphertext,cn))
191             {
192             fprintf(stderr,"Ciphertext mismatch\n");
193             hexdump(stderr,"Got",out,cn);
194             hexdump(stderr,"Expected",ciphertext,cn);
195             test1_exit(9);
196             }
197         }
198
199     if (encdec <= 0)
200         {
201         if(!EVP_DecryptInit_ex(&ctx,c,NULL,key,iv))
202             {
203             fprintf(stderr,"DecryptInit failed\n");
204             ERR_print_errors_fp(stderr);
205             test1_exit(11);
206             }
207         EVP_CIPHER_CTX_set_padding(&ctx,0);
208
209         if(!EVP_DecryptUpdate(&ctx,out,&outl,ciphertext,cn))
210             {
211             fprintf(stderr,"Decrypt failed\n");
212             ERR_print_errors_fp(stderr);
213             test1_exit(6);
214             }
215         if(!EVP_DecryptFinal_ex(&ctx,out+outl,&outl2))
216             {
217             fprintf(stderr,"DecryptFinal failed\n");
218             ERR_print_errors_fp(stderr);
219             test1_exit(7);
220             }
221
222         if(outl+outl2 != cn)
223             {
224             fprintf(stderr,"Plaintext length mismatch got %d expected %d\n",
225                     outl+outl2,cn);
226             test1_exit(8);
227             }
228
229         if(memcmp(out,plaintext,cn))
230             {
231             fprintf(stderr,"Plaintext mismatch\n");
232             hexdump(stderr,"Got",out,cn);
233             hexdump(stderr,"Expected",plaintext,cn);
234             test1_exit(9);
235             }
236         }
237
238     EVP_CIPHER_CTX_cleanup(&ctx);
239
240     printf("\n");
241     }
242
243 static int test_cipher(const char *cipher,const unsigned char *key,int kn,
244                        const unsigned char *iv,int in,
245                        const unsigned char *plaintext,int pn,
246                        const unsigned char *ciphertext,int cn,
247                        int encdec)
248     {
249     const EVP_CIPHER *c;
250
251     c=EVP_get_cipherbyname(cipher);
252     if(!c)
253         return 0;
254
255     test1(c,key,kn,iv,in,plaintext,pn,ciphertext,cn,encdec);
256
257     return 1;
258     }
259
260 static int test_digest(const char *digest,
261                        const unsigned char *plaintext,int pn,
262                        const unsigned char *ciphertext, unsigned int cn)
263     {
264     const EVP_MD *d;
265     EVP_MD_CTX ctx;
266     unsigned char md[EVP_MAX_MD_SIZE];
267     unsigned int mdn;
268
269     d=EVP_get_digestbyname(digest);
270     if(!d)
271         return 0;
272
273     printf("Testing digest %s\n",EVP_MD_name(d));
274     hexdump(stdout,"Plaintext",plaintext,pn);
275     hexdump(stdout,"Digest",ciphertext,cn);
276
277     EVP_MD_CTX_init(&ctx);
278     if(!EVP_DigestInit_ex(&ctx,d, NULL))
279         {
280         fprintf(stderr,"DigestInit failed\n");
281         ERR_print_errors_fp(stderr);
282         EXIT(100);
283         }
284     if(!EVP_DigestUpdate(&ctx,plaintext,pn))
285         {
286         fprintf(stderr,"DigestUpdate failed\n");
287         ERR_print_errors_fp(stderr);
288         EXIT(101);
289         }
290     if(!EVP_DigestFinal_ex(&ctx,md,&mdn))
291         {
292         fprintf(stderr,"DigestFinal failed\n");
293         ERR_print_errors_fp(stderr);
294         EXIT(101);
295         }
296     EVP_MD_CTX_cleanup(&ctx);
297
298     if(mdn != cn)
299         {
300         fprintf(stderr,"Digest length mismatch, got %d expected %d\n",mdn,cn);
301         EXIT(102);
302         }
303
304     if(memcmp(md,ciphertext,cn))
305         {
306         fprintf(stderr,"Digest mismatch\n");
307         hexdump(stderr,"Got",md,cn);
308         hexdump(stderr,"Expected",ciphertext,cn);
309         EXIT(103);
310         }
311
312     printf("\n");
313
314     EVP_MD_CTX_cleanup(&ctx);
315
316     return 1;
317     }
318
319 int main(int argc,char **argv)
320     {
321     const char *szTestFile;
322     FILE *f;
323
324     if(argc != 2)
325         {
326         fprintf(stderr,"%s <test file>\n",argv[0]);
327         EXIT(1);
328         }
329     CRYPTO_malloc_debug_init();
330     CRYPTO_set_mem_debug_options(V_CRYPTO_MDEBUG_ALL);
331     CRYPTO_mem_ctrl(CRYPTO_MEM_CHECK_ON);
332
333     szTestFile=argv[1];
334
335     f=fopen(szTestFile,"r");
336     if(!f)
337         {
338         perror(szTestFile);
339         EXIT(2);
340         }
341
342     /* Load up the software EVP_CIPHER and EVP_MD definitions */
343     OpenSSL_add_all_ciphers();
344     OpenSSL_add_all_digests();
345 #ifndef OPENSSL_NO_ENGINE
346     /* Load all compiled-in ENGINEs */
347     ENGINE_load_builtin_engines();
348 #endif
349 #if 0
350     OPENSSL_config();
351 #endif
352 #ifndef OPENSSL_NO_ENGINE
353     /* Register all available ENGINE implementations of ciphers and digests.
354      * This could perhaps be changed to "ENGINE_register_all_complete()"? */
355     ENGINE_register_all_ciphers();
356     ENGINE_register_all_digests();
357     /* If we add command-line options, this statement should be switchable.
358      * It'll prevent ENGINEs being ENGINE_init()ialised for cipher/digest use if
359      * they weren't already initialised. */
360     /* ENGINE_set_cipher_flags(ENGINE_CIPHER_FLAG_NOINIT); */
361 #endif
362
363     for( ; ; )
364         {
365         char line[4096];
366         char *p;
367         char *cipher;
368         unsigned char *iv,*key,*plaintext,*ciphertext;
369         int encdec;
370         int kn,in,pn,cn;
371
372         if(!fgets((char *)line,sizeof line,f))
373             break;
374         if(line[0] == '#' || line[0] == '\n')
375             continue;
376         p=line;
377         cipher=sstrsep(&p,":"); 
378         key=ustrsep(&p,":");
379         iv=ustrsep(&p,":");
380         plaintext=ustrsep(&p,":");
381         ciphertext=ustrsep(&p,":");
382         if (p[-1] == '\n') {
383             p[-1] = '\0';
384             encdec = -1;
385         } else {
386             encdec = atoi(sstrsep(&p,"\n"));
387         }
388               
389
390         kn=convert(key);
391         in=convert(iv);
392         pn=convert(plaintext);
393         cn=convert(ciphertext);
394
395         if(!test_cipher(cipher,key,kn,iv,in,plaintext,pn,ciphertext,cn,encdec)
396            && !test_digest(cipher,plaintext,pn,ciphertext,cn))
397             {
398             fprintf(stderr,"Can't find %s\n",cipher);
399             EXIT(3);
400             }
401         }
402
403 #ifndef OPENSSL_NO_ENGINE
404     ENGINE_cleanup();
405 #endif
406     EVP_cleanup();
407     CRYPTO_cleanup_all_ex_data();
408     ERR_remove_state(0);
409     ERR_free_strings();
410     CRYPTO_mem_leaks_fp(stderr);
411
412     return 0;
413     }