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     {
 128     EVP_CIPHER_CTX ctx;
 129     unsigned char out[4096];
 130     int outl,outl2;
 131
 132     printf("Testing cipher %s\n",EVP_CIPHER_name(c));
 133     hexdump(stdout,"Key",key,kn);
 134     if(in)
 135         hexdump(stdout,"IV",iv,in);
 136     hexdump(stdout,"Plaintext",plaintext,pn);
 137     hexdump(stdout,"Ciphertext",ciphertext,cn);
 138
 139     if(kn != c->key_len)
 140         {
 141         fprintf(stderr,"Key length doesn't match, got %d expected %d\n",kn,
 142                 c->key_len);
 143         exit(5);
 144         }
 145     EVP_CIPHER_CTX_init(&ctx);
 146     if(!EVP_EncryptInit_ex(&ctx,c,NULL,key,iv))
 147         {
 148         fprintf(stderr,"EncryptInit failed\n");
 149         exit(10);
 150         }
 151     EVP_CIPHER_CTX_set_padding(&ctx,0);
 152
 153     if(!EVP_EncryptUpdate(&ctx,out,&outl,plaintext,pn))
 154         {
 155         fprintf(stderr,"Encrypt failed\n");
 156         exit(6);
 157         }
 158     if(!EVP_EncryptFinal_ex(&ctx,out+outl,&outl2))
 159         {
 160         fprintf(stderr,"EncryptFinal failed\n");
 161         exit(7);
 162         }
 163
 164     if(outl+outl2 != cn)
 165         {
 166         fprintf(stderr,"Ciphertext length mismatch got %d expected %d\n",
 167                 outl+outl2,cn);
 168         exit(8);
 169         }
 170
 171     if(memcmp(out,ciphertext,cn))
 172         {
 173         fprintf(stderr,"Ciphertext mismatch\n");
 174         hexdump(stderr,"Got",out,cn);
 175         hexdump(stderr,"Expected",ciphertext,cn);
 176         exit(9);
 177         }
 178
 179     if(!EVP_DecryptInit_ex(&ctx,c,NULL,key,iv))
 180         {
 181         fprintf(stderr,"DecryptInit failed\n");
 182         exit(11);
 183         }
 184     EVP_CIPHER_CTX_set_padding(&ctx,0);
 185
 186     if(!EVP_DecryptUpdate(&ctx,out,&outl,ciphertext,pn))
 187         {
 188         fprintf(stderr,"Decrypt failed\n");
 189         exit(6);
 190         }
 191     if(!EVP_DecryptFinal_ex(&ctx,out+outl,&outl2))
 192         {
 193         fprintf(stderr,"DecryptFinal failed\n");
 194         exit(7);
 195         }
 196
 197     if(outl+outl2 != cn)
 198         {
 199         fprintf(stderr,"Plaintext length mismatch got %d expected %d\n",
 200                 outl+outl2,cn);
 201         exit(8);
 202         }
 203
 204     if(memcmp(out,plaintext,cn))
 205         {
 206         fprintf(stderr,"Plaintext mismatch\n");
 207         hexdump(stderr,"Got",out,cn);
 208         hexdump(stderr,"Expected",plaintext,cn);
 209         exit(9);
 210         }
 211
 212     EVP_CIPHER_CTX_cleanup(&ctx);
 213
 214     printf("\n");
 215     }
 216
 217 static int test_cipher(const char *cipher,const unsigned char *key,int kn,
 218                        const unsigned char *iv,int in,
 219                        const unsigned char *plaintext,int pn,
 220                        const unsigned char *ciphertext,int cn)
 221     {
 222     const EVP_CIPHER *c;
 223
 224     c=EVP_get_cipherbyname(cipher);
 225     if(!c)
 226         return 0;
 227
 228     test1(c,key,kn,iv,in,plaintext,pn,ciphertext,cn);
 229
 230     return 1;
 231     }
 232
 233 static int test_digest(const char *digest,
 234                        const unsigned char *plaintext,int pn,
 235                        const unsigned char *ciphertext, unsigned int cn)
 236     {
 237     const EVP_MD *d;
 238     EVP_MD_CTX ctx;
 239     unsigned char md[EVP_MAX_MD_SIZE];
 240     unsigned int mdn;
 241
 242     d=EVP_get_digestbyname(digest);
 243     if(!d)
 244         return 0;
 245
 246     printf("Testing digest %s\n",EVP_MD_name(d));
 247     hexdump(stdout,"Plaintext",plaintext,pn);
 248     hexdump(stdout,"Digest",ciphertext,cn);
 249
 250     EVP_MD_CTX_init(&ctx);
 251     if(!EVP_DigestInit_ex(&ctx,d, NULL))
 252         {
 253         fprintf(stderr,"DigestInit failed\n");
 254         exit(100);
 255         }
 256     if(!EVP_DigestUpdate(&ctx,plaintext,pn))
 257         {
 258         fprintf(stderr,"DigestUpdate failed\n");
 259         exit(101);
 260         }
 261     if(!EVP_DigestFinal_ex(&ctx,md,&mdn))
 262         {
 263         fprintf(stderr,"DigestFinal failed\n");
 264         exit(101);
 265         }
 266     EVP_MD_CTX_cleanup(&ctx);
 267
 268     if(mdn != cn)
 269         {
 270         fprintf(stderr,"Digest length mismatch, got %d expected %d\n",mdn,cn);
 271         exit(102);
 272         }
 273
 274     if(memcmp(md,ciphertext,cn))
 275         {
 276         fprintf(stderr,"Digest mismatch\n");
 277         hexdump(stderr,"Got",md,cn);
 278         hexdump(stderr,"Expected",ciphertext,cn);
 279         exit(103);
 280         }
 281
 282     printf("\n");
 283
 284     EVP_MD_CTX_cleanup(&ctx);
 285
 286     return 1;
 287     }
 288
 289 int main(int argc,char **argv)
 290     {
 291     const char *szTestFile;
 292     FILE *f;
 293
 294     if(argc != 2)
 295         {
 296         fprintf(stderr,"%s <test file>\n",argv[0]);
 297         exit(1);
 298         }
 299     CRYPTO_malloc_debug_init();
 300     CRYPTO_set_mem_debug_options(V_CRYPTO_MDEBUG_ALL);
 301     CRYPTO_mem_ctrl(CRYPTO_MEM_CHECK_ON);
 302
 303     szTestFile=argv[1];
 304
 305     f=fopen(szTestFile,"r");
 306     if(!f)
 307         {
 308         perror(szTestFile);
 309         exit(2);
 310         }
 311
 312     /* Load up the software EVP_CIPHER and EVP_MD definitions */
 313     OpenSSL_add_all_ciphers();
 314     OpenSSL_add_all_digests();
 315     /* Load all compiled-in ENGINEs */
 316     ENGINE_load_builtin_engines();
 317 #if 0
 318     OPENSSL_config();
 319 #endif
 320     /* Register all available ENGINE implementations of ciphers and digests.
 321      * This could perhaps be changed to "ENGINE_register_all_complete()"? */
 322     ENGINE_register_all_ciphers();
 323     ENGINE_register_all_digests();
 324     /* If we add command-line options, this statement should be switchable.
 325      * It'll prevent ENGINEs being ENGINE_init()ialised for cipher/digest use if
 326      * they weren't already initialised. */
 327     /* ENGINE_set_cipher_flags(ENGINE_CIPHER_FLAG_NOINIT); */
 328
 329     for( ; ; )
 330         {
 331         char line[4096];
 332         char *p;
 333         char *cipher;
 334         unsigned char *iv,*key,*plaintext,*ciphertext;
 335         int kn,in,pn,cn;
 336
 337         if(!fgets((char *)line,sizeof line,f))
 338             break;
 339         if(line[0] == '#' || line[0] == '\n')
 340             continue;
 341         p=line;
 342         cipher=sstrsep(&p,":");
 343         key=ustrsep(&p,":");
 344         iv=ustrsep(&p,":");
 345         plaintext=ustrsep(&p,":");
 346         ciphertext=ustrsep(&p,"\n");
 347
 348         kn=convert(key);
 349         in=convert(iv);
 350         pn=convert(plaintext);
 351         cn=convert(ciphertext);
 352
 353         if(!test_cipher(cipher,key,kn,iv,in,plaintext,pn,ciphertext,cn)
 354            && !test_digest(cipher,plaintext,pn,ciphertext,cn))
 355             {
 356             fprintf(stderr,"Can't find %s\n",cipher);
 357             exit(3);
 358             }
 359         }
 360
 361     ENGINE_cleanup();
 362     EVP_cleanup();
 363     CRYPTO_cleanup_all_ex_data();
 364     ERR_remove_state(0);
 365     ERR_free_strings();
 366     CRYPTO_mem_leaks_fp(stderr);
 367
 368     return 0;
 369     }