7c5453d7e9d2fe52b5f03d806ad51725c7676e98
[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 #include <openssl/evp.h>
53 #include <openssl/engine.h>
54 #include <openssl/conf.h>
55
56 static void hexdump(FILE *f,const char *title,const unsigned char *s,int l)
57     {
58     int n=0;
59
60     fprintf(f,"%s",title);
61     for( ; n < l ; ++n)
62         {
63         if((n%16) == 0)
64             fprintf(f,"\n%04x",n);
65         fprintf(f," %02x",s[n]);
66         }
67     fprintf(f,"\n");
68     }
69
70 static int convert(unsigned char *s)
71     {
72     unsigned char *d;
73
74     for(d=s ; *s ; s+=2,++d)
75         {
76         unsigned int n;
77
78         if(!s[1])
79             {
80             fprintf(stderr,"Odd number of hex digits!");
81             exit(4);
82             }
83         sscanf((char *)s,"%2x",&n);
84         *d=(unsigned char)n;
85         }
86     return s-d;
87     }
88
89 static char *sstrsep(char **string, const char *delim)
90     {
91     char isdelim[256];
92     char *token = *string;
93
94     if (**string == 0)
95         return NULL;
96
97     memset(isdelim, 0, 256);
98     isdelim[0] = 1;
99
100     while (*delim)
101         {
102         isdelim[(unsigned char)(*delim)] = 1;
103         delim++;
104         }
105
106     while (!isdelim[(unsigned char)(**string)])
107         {
108         (*string)++;
109         }
110
111     if (**string)
112         {
113         **string = 0;
114         (*string)++;
115         }
116
117     return token;
118     }
119
120 static unsigned char *ustrsep(char **p,const char *sep)
121     { return (unsigned char *)sstrsep((char **)p,sep); }
122
123 static void test1(const EVP_CIPHER *c,const unsigned char *key,int kn,
124                   const unsigned char *iv,int in,
125                   const unsigned char *plaintext,int pn,
126                   const unsigned char *ciphertext,int cn,
127                   int encdec)
128     {
129     EVP_CIPHER_CTX ctx;
130     unsigned char out[4096];
131     int outl,outl2;
132
133     printf("Testing cipher %s%s\n",EVP_CIPHER_name(c),
134            (encdec == 1 ? "(encrypt)" : (encdec == 0 ? "(decrypt)" : "(encrypt/decrypt)")));
135     hexdump(stdout,"Key",key,kn);
136     if(in)
137         hexdump(stdout,"IV",iv,in);
138     hexdump(stdout,"Plaintext",plaintext,pn);
139     hexdump(stdout,"Ciphertext",ciphertext,cn);
140     
141     if(kn != c->key_len)
142         {
143         fprintf(stderr,"Key length doesn't match, got %d expected %d\n",kn,
144                 c->key_len);
145         exit(5);
146         }
147     EVP_CIPHER_CTX_init(&ctx);
148     if (encdec != 0)
149         {
150         if(!EVP_EncryptInit_ex(&ctx,c,NULL,key,iv))
151             {
152             fprintf(stderr,"EncryptInit failed\n");
153             exit(10);
154             }
155         EVP_CIPHER_CTX_set_padding(&ctx,0);
156
157         if(!EVP_EncryptUpdate(&ctx,out,&outl,plaintext,pn))
158             {
159             fprintf(stderr,"Encrypt failed\n");
160             exit(6);
161             }
162         if(!EVP_EncryptFinal_ex(&ctx,out+outl,&outl2))
163             {
164             fprintf(stderr,"EncryptFinal failed\n");
165             exit(7);
166             }
167
168         if(outl+outl2 != cn)
169             {
170             fprintf(stderr,"Ciphertext length mismatch got %d expected %d\n",
171                     outl+outl2,cn);
172             exit(8);
173             }
174
175         if(memcmp(out,ciphertext,cn))
176             {
177             fprintf(stderr,"Ciphertext mismatch\n");
178             hexdump(stderr,"Got",out,cn);
179             hexdump(stderr,"Expected",ciphertext,cn);
180             exit(9);
181             }
182         }
183
184     if (encdec <= 0)
185         {
186         if(!EVP_DecryptInit_ex(&ctx,c,NULL,key,iv))
187             {
188             fprintf(stderr,"DecryptInit failed\n");
189             exit(11);
190             }
191         EVP_CIPHER_CTX_set_padding(&ctx,0);
192
193         if(!EVP_DecryptUpdate(&ctx,out,&outl,ciphertext,cn))
194             {
195             fprintf(stderr,"Decrypt failed\n");
196             exit(6);
197             }
198         if(!EVP_DecryptFinal_ex(&ctx,out+outl,&outl2))
199             {
200             fprintf(stderr,"DecryptFinal failed\n");
201             exit(7);
202             }
203
204         if(outl+outl2 != cn)
205             {
206             fprintf(stderr,"Plaintext length mismatch got %d expected %d\n",
207                     outl+outl2,cn);
208             exit(8);
209             }
210
211         if(memcmp(out,plaintext,cn))
212             {
213             fprintf(stderr,"Plaintext mismatch\n");
214             hexdump(stderr,"Got",out,cn);
215             hexdump(stderr,"Expected",plaintext,cn);
216             exit(9);
217             }
218         }
219
220     EVP_CIPHER_CTX_cleanup(&ctx);
221
222     printf("\n");
223     }
224
225 static int test_cipher(const char *cipher,const unsigned char *key,int kn,
226                        const unsigned char *iv,int in,
227                        const unsigned char *plaintext,int pn,
228                        const unsigned char *ciphertext,int cn,
229                        int encdec)
230     {
231     const EVP_CIPHER *c;
232
233     c=EVP_get_cipherbyname(cipher);
234     if(!c)
235         return 0;
236
237     test1(c,key,kn,iv,in,plaintext,pn,ciphertext,cn,encdec);
238
239     return 1;
240     }
241
242 static int test_digest(const char *digest,
243                        const unsigned char *plaintext,int pn,
244                        const unsigned char *ciphertext, unsigned int cn)
245     {
246     const EVP_MD *d;
247     EVP_MD_CTX ctx;
248     unsigned char md[EVP_MAX_MD_SIZE];
249     unsigned int mdn;
250
251     d=EVP_get_digestbyname(digest);
252     if(!d)
253         return 0;
254
255     printf("Testing digest %s\n",EVP_MD_name(d));
256     hexdump(stdout,"Plaintext",plaintext,pn);
257     hexdump(stdout,"Digest",ciphertext,cn);
258
259     EVP_MD_CTX_init(&ctx);
260     if(!EVP_DigestInit_ex(&ctx,d, NULL))
261         {
262         fprintf(stderr,"DigestInit failed\n");
263         exit(100);
264         }
265     if(!EVP_DigestUpdate(&ctx,plaintext,pn))
266         {
267         fprintf(stderr,"DigestUpdate failed\n");
268         exit(101);
269         }
270     if(!EVP_DigestFinal_ex(&ctx,md,&mdn))
271         {
272         fprintf(stderr,"DigestFinal failed\n");
273         exit(101);
274         }
275     EVP_MD_CTX_cleanup(&ctx);
276
277     if(mdn != cn)
278         {
279         fprintf(stderr,"Digest length mismatch, got %d expected %d\n",mdn,cn);
280         exit(102);
281         }
282
283     if(memcmp(md,ciphertext,cn))
284         {
285         fprintf(stderr,"Digest mismatch\n");
286         hexdump(stderr,"Got",md,cn);
287         hexdump(stderr,"Expected",ciphertext,cn);
288         exit(103);
289         }
290
291     printf("\n");
292
293     EVP_MD_CTX_cleanup(&ctx);
294
295     return 1;
296     }
297
298 int main(int argc,char **argv)
299     {
300     const char *szTestFile;
301     FILE *f;
302
303     if(argc != 2)
304         {
305         fprintf(stderr,"%s <test file>\n",argv[0]);
306         exit(1);
307         }
308     CRYPTO_malloc_debug_init();
309     CRYPTO_set_mem_debug_options(V_CRYPTO_MDEBUG_ALL);
310     CRYPTO_mem_ctrl(CRYPTO_MEM_CHECK_ON);
311
312     szTestFile=argv[1];
313
314     f=fopen(szTestFile,"r");
315     if(!f)
316         {
317         perror(szTestFile);
318         exit(2);
319         }
320
321     /* Load up the software EVP_CIPHER and EVP_MD definitions */
322     OpenSSL_add_all_ciphers();
323     OpenSSL_add_all_digests();
324     /* Load all compiled-in ENGINEs */
325     ENGINE_load_builtin_engines();
326 #if 0
327     OPENSSL_config();
328 #endif
329     /* Register all available ENGINE implementations of ciphers and digests.
330      * This could perhaps be changed to "ENGINE_register_all_complete()"? */
331     ENGINE_register_all_ciphers();
332     ENGINE_register_all_digests();
333     /* If we add command-line options, this statement should be switchable.
334      * It'll prevent ENGINEs being ENGINE_init()ialised for cipher/digest use if
335      * they weren't already initialised. */
336     /* ENGINE_set_cipher_flags(ENGINE_CIPHER_FLAG_NOINIT); */
337
338     for( ; ; )
339         {
340         char line[4096];
341         char *p;
342         char *cipher;
343         unsigned char *iv,*key,*plaintext,*ciphertext;
344         int encdec;
345         int kn,in,pn,cn;
346
347         if(!fgets((char *)line,sizeof line,f))
348             break;
349         if(line[0] == '#' || line[0] == '\n')
350             continue;
351         p=line;
352         cipher=sstrsep(&p,":"); 
353         key=ustrsep(&p,":");
354         iv=ustrsep(&p,":");
355         plaintext=ustrsep(&p,":");
356         ciphertext=ustrsep(&p,":");
357         if (p[-1] == '\n') {
358             p[-1] = '\0';
359             encdec = -1;
360         } else {
361             encdec = atoi(ustrsep(&p,"\n"));
362         }
363               
364
365         kn=convert(key);
366         in=convert(iv);
367         pn=convert(plaintext);
368         cn=convert(ciphertext);
369
370         if(!test_cipher(cipher,key,kn,iv,in,plaintext,pn,ciphertext,cn,encdec)
371            && !test_digest(cipher,plaintext,pn,ciphertext,cn))
372             {
373             fprintf(stderr,"Can't find %s\n",cipher);
374             exit(3);
375             }
376         }
377
378     ENGINE_cleanup();
379     EVP_cleanup();
380     CRYPTO_cleanup_all_ex_data();
381     ERR_remove_state(0);
382     ERR_free_strings();
383     CRYPTO_mem_leaks_fp(stderr);
384
385     return 0;
386     }