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