fips/aes/fips_gcmtest.c

   1 /* fips/aes/fips_gcmtest.c */
   2 /* Written by Dr Stephen N Henson (steve@openssl.org) for the OpenSSL
   3  * project.
   4  */
   5 /* ====================================================================
   6  * Copyright (c) 2011 The OpenSSL Project.  All rights reserved.
   7  *
   8  * Redistribution and use in source and binary forms, with or without
   9  * modification, are permitted provided that the following conditions
  10  * are met:
  11  *
  12  * 1. Redistributions of source code must retain the above copyright
  13  *    notice, this list of conditions and the following disclaimer.
  14  *
  15  * 2. Redistributions in binary form must reproduce the above copyright
  16  *    notice, this list of conditions and the following disclaimer in
  17  *    the documentation and/or other materials provided with the
  18  *    distribution.
  19  *
  20  * 3. All advertising materials mentioning features or use of this
  21  *    software must display the following acknowledgment:
  22  *    "This product includes software developed by the OpenSSL Project
  23  *    for use in the OpenSSL Toolkit. (http://www.OpenSSL.org/)"
  24  *
  25  * 4. The names "OpenSSL Toolkit" and "OpenSSL Project" must not be used to
  26  *    endorse or promote products derived from this software without
  27  *    prior written permission. For written permission, please contact
  28  *    licensing@OpenSSL.org.
  29  *
  30  * 5. Products derived from this software may not be called "OpenSSL"
  31  *    nor may "OpenSSL" appear in their names without prior written
  32  *    permission of the OpenSSL Project.
  33  *
  34  * 6. Redistributions of any form whatsoever must retain the following
  35  *    acknowledgment:
  36  *    "This product includes software developed by the OpenSSL Project
  37  *    for use in the OpenSSL Toolkit (http://www.OpenSSL.org/)"
  38  *
  39  * THIS SOFTWARE IS PROVIDED BY THE OpenSSL PROJECT ``AS IS'' AND ANY
  40  * EXPRESSED OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  41  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
  42  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE OpenSSL PROJECT OR
  43  * ITS CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
  44  * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
  45  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
  46  * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  47  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
  48  * STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
  49  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED
  50  * OF THE POSSIBILITY OF SUCH DAMAGE.
  51  * ====================================================================
  52  */
  53
  54
  55 #define OPENSSL_FIPSAPI
  56 #include <openssl/opensslconf.h>
  57
  58 #ifndef OPENSSL_FIPS
  59 #include <stdio.h>
  60
  61 int main(int argc, char **argv)
  62 {
  63     printf("No FIPS GCM support\n");
  64     return(0);
  65 }
  66 #else
  67
  68 #include <openssl/bn.h>
  69 #include <openssl/dsa.h>
  70 #include <openssl/fips.h>
  71 #include <openssl/err.h>
  72 #include <openssl/evp.h>
  73 #include <string.h>
  74 #include <ctype.h>
  75
  76 #include "fips_utl.h"
  77
  78 static void gcmtest(FILE *in, FILE *out, int encrypt)
  79         {
  80         char buf[2048];
  81         char lbuf[2048];
  82         char *keyword, *value;
  83         int keylen = -1, ivlen = -1, aadlen = -1, taglen = -1, ptlen = -1;
  84         int rv;
  85         long l;
  86         unsigned char *key = NULL, *iv = NULL, *aad = NULL, *tag = NULL;
  87         unsigned char *ct = NULL, *pt = NULL;
  88         EVP_CIPHER_CTX ctx;
  89         const EVP_CIPHER *gcm = NULL;
  90         FIPS_cipher_ctx_init(&ctx);
  91
  92         while(fgets(buf,sizeof buf,in) != NULL)
  93                 {
  94                 fputs(buf,out);
  95                 if (!parse_line(&keyword, &value, lbuf, buf))
  96                         continue;
  97                 if(!strcmp(keyword,"[Keylen"))
  98                         {
  99                         keylen = atoi(value);
 100                         if (keylen == 128)
 101                                 gcm = EVP_aes_128_gcm();
 102                         else if (keylen == 192)
 103                                 gcm = EVP_aes_192_gcm();
 104                         else if (keylen == 256)
 105                                 gcm = EVP_aes_256_gcm();
 106                         else
 107                                 {
 108                                 fprintf(stderr, "Unsupported keylen %d\n",
 109                                                         keylen);
 110                                 }
 111                         keylen >>= 3;
 112                         }
 113                 else if (!strcmp(keyword, "[IVlen"))
 114                         ivlen = atoi(value) >> 3;
 115                 else if (!strcmp(keyword, "[AADlen"))
 116                         aadlen = atoi(value) >> 3;
 117                 else if (!strcmp(keyword, "[Taglen"))
 118                         taglen = atoi(value) >> 3;
 119                 else if (!strcmp(keyword, "[PTlen"))
 120                         ptlen = atoi(value) >> 3;
 121                 else if(!strcmp(keyword,"Key"))
 122                         {
 123                         key = hex2bin_m(value, &l);
 124                         if (l != keylen)
 125                                 {
 126                                 fprintf(stderr, "Inconsistent Key length\n");
 127                                 exit(1);
 128                                 }
 129                         }
 130                 else if(!strcmp(keyword,"IV"))
 131                         {
 132                         iv = hex2bin_m(value, &l);
 133                         if (l != ivlen)
 134                                 {
 135                                 fprintf(stderr, "Inconsistent IV length\n");
 136                                 exit(1);
 137                                 }
 138                         }
 139                 else if(!strcmp(keyword,"PT"))
 140                         {
 141                         pt = hex2bin_m(value, &l);
 142                         if (l != ptlen)
 143                                 {
 144                                 fprintf(stderr, "Inconsistent PT length\n");
 145                                 exit(1);
 146                                 }
 147                         }
 148                 else if(!strcmp(keyword,"CT"))
 149                         {
 150                         ct = hex2bin_m(value, &l);
 151                         if (l != ptlen)
 152                                 {
 153                                 fprintf(stderr, "Inconsistent CT length\n");
 154                                 exit(1);
 155                                 }
 156                         }
 157                 else if(!strcmp(keyword,"AAD"))
 158                         {
 159                         aad = hex2bin_m(value, &l);
 160                         if (l != aadlen)
 161                                 {
 162                                 fprintf(stderr, "Inconsistent AAD length\n");
 163                                 exit(1);
 164                                 }
 165                         }
 166                 else if(!strcmp(keyword,"Tag"))
 167                         {
 168                         tag = hex2bin_m(value, &l);
 169                         if (l != taglen)
 170                                 {
 171                                 fprintf(stderr, "Inconsistent Tag length\n");
 172                                 exit(1);
 173                                 }
 174                         }
 175                 if (encrypt && pt && aad && (iv || encrypt==1))
 176                         {
 177                         tag = OPENSSL_malloc(taglen);
 178                         FIPS_cipherinit(&ctx, gcm, NULL, NULL, 1);
 179                         /* Relax FIPS constraints for testing */
 180                         M_EVP_CIPHER_CTX_set_flags(&ctx, EVP_CIPH_FLAG_NON_FIPS_ALLOW);
 181                         FIPS_cipher_ctx_ctrl(&ctx, EVP_CTRL_GCM_SET_IVLEN, ivlen, 0);
 182                         if (encrypt == 1)
 183                                 {
 184                                 static unsigned char iv_fixed[4] = {1,2,3,4};
 185                                 if (!iv)
 186                                         iv = OPENSSL_malloc(ivlen);
 187                                 FIPS_cipherinit(&ctx, NULL, key, NULL, 1);
 188                                 FIPS_cipher_ctx_ctrl(&ctx,
 189                                                 EVP_CTRL_GCM_SET_IV_FIXED,
 190                                                 4, iv_fixed);
 191                                 if (!FIPS_cipher_ctx_ctrl(&ctx,
 192                                         EVP_CTRL_GCM_IV_GEN, 0, iv))
 193                                         {
 194                                         fprintf(stderr, "IV gen error\n");
 195                                         exit(1);
 196                                         }
 197                                 OutputValue("IV", iv, ivlen, out, 0);
 198                                 }
 199                         else
 200                                 FIPS_cipherinit(&ctx, NULL, key, iv, 1);
 201
 202
 203                         if (aadlen)
 204                                 FIPS_cipher(&ctx, NULL, aad, aadlen);
 205                         if (ptlen)
 206                                 {
 207                                 ct = OPENSSL_malloc(ptlen);
 208                                 rv = FIPS_cipher(&ctx, ct, pt, ptlen);
 209                                 }
 210                         FIPS_cipher(&ctx, NULL, NULL, 0);
 211                         FIPS_cipher_ctx_ctrl(&ctx, EVP_CTRL_GCM_GET_TAG,
 212                                                                 taglen, tag);
 213                         OutputValue("CT", ct, ptlen, out, 0);
 214                         OutputValue("Tag", tag, taglen, out, 0);
 215                         if (iv)
 216                                 OPENSSL_free(iv);
 217                         if (aad)
 218                                 OPENSSL_free(aad);
 219                         if (ct)
 220                                 OPENSSL_free(ct);
 221                         if (pt)
 222                                 OPENSSL_free(pt);
 223                         if (key)
 224                                 OPENSSL_free(key);
 225                         if (tag)
 226                                 OPENSSL_free(tag);
 227                         iv = aad = ct = pt = key = tag = NULL;
 228                         }
 229                 if (!encrypt && tag)
 230                         {
 231                         FIPS_cipherinit(&ctx, gcm, NULL, NULL, 0);
 232                         /* Relax FIPS constraints for testing */
 233                         M_EVP_CIPHER_CTX_set_flags(&ctx, EVP_CIPH_FLAG_NON_FIPS_ALLOW);
 234                         FIPS_cipher_ctx_ctrl(&ctx, EVP_CTRL_GCM_SET_IVLEN, ivlen, 0);
 235                         FIPS_cipherinit(&ctx, NULL, key, iv, 0);
 236                         FIPS_cipher_ctx_ctrl(&ctx, EVP_CTRL_GCM_SET_TAG, taglen, tag);
 237                         if (aadlen)
 238                                 FIPS_cipher(&ctx, NULL, aad, aadlen);
 239                         if (ptlen)
 240                                 {
 241                                 pt = OPENSSL_malloc(ptlen);
 242                                 rv = FIPS_cipher(&ctx, pt, ct, ptlen);
 243                                 }
 244                         rv = FIPS_cipher(&ctx, NULL, NULL, 0);
 245                         if (rv < 0)
 246                                 fprintf(out, "FAIL\n");
 247                         else
 248                                 OutputValue("PT", pt, ptlen, out, 0);
 249                         if (iv)
 250                                 OPENSSL_free(iv);
 251                         if (aad)
 252                                 OPENSSL_free(aad);
 253                         if (ct)
 254                                 OPENSSL_free(ct);
 255                         if (pt)
 256                                 OPENSSL_free(pt);
 257                         if (key)
 258                                 OPENSSL_free(key);
 259                         if (tag)
 260                                 OPENSSL_free(tag);
 261                         iv = aad = ct = pt = key = tag = NULL;
 262                         }
 263                 }
 264         }
 265
 266 static void xtstest(FILE *in, FILE *out)
 267         {
 268         char buf[2048];
 269         char lbuf[2048];
 270         char *keyword, *value;
 271         int inlen;
 272         int encrypt = 0;
 273         int rv;
 274         long l;
 275         unsigned char *key = NULL, *iv = NULL;
 276         unsigned char *inbuf = NULL, *outbuf = NULL;
 277         EVP_CIPHER_CTX ctx;
 278         const EVP_CIPHER *xts = NULL;
 279         FIPS_cipher_ctx_init(&ctx);
 280
 281         while(fgets(buf,sizeof buf,in) != NULL)
 282                 {
 283                 fputs(buf,out);
 284                 if (buf[0] == '[' && strlen(buf) >= 9)
 285                         {
 286                         if(!strncmp(buf,"[ENCRYPT]", 9))
 287                                 encrypt = 1;
 288                         else if(!strncmp(buf,"[DECRYPT]", 9))
 289                                 encrypt = 0;
 290                         }
 291                 if  (!parse_line(&keyword, &value, lbuf, buf))
 292                         continue;
 293                 else if(!strcmp(keyword,"Key"))
 294                         {
 295                         key = hex2bin_m(value, &l);
 296                         if (l == 32)
 297                                 xts = EVP_aes_128_xts();
 298                         else if (l == 64)
 299                                 xts = EVP_aes_256_xts();
 300                         else
 301                                 {
 302                                 fprintf(stderr, "Inconsistent Key length\n");
 303                                 exit(1);
 304                                 }
 305                         }
 306                 else if(!strcmp(keyword,"i"))
 307                         {
 308                         iv = hex2bin_m(value, &l);
 309                         if (l != 16)
 310                                 {
 311                                 fprintf(stderr, "Inconsistent i length\n");
 312                                 exit(1);
 313                                 }
 314                         }
 315                 else if(encrypt && !strcmp(keyword,"PT"))
 316                         {
 317                         inbuf = hex2bin_m(value, &l);
 318                         inlen = l;
 319                         }
 320                 else if(!encrypt && !strcmp(keyword,"CT"))
 321                         {
 322                         inbuf = hex2bin_m(value, &l);
 323                         inlen = l;
 324                         }
 325                 if (inbuf)
 326                         {
 327                         FIPS_cipherinit(&ctx, xts, key, iv, encrypt);
 328                         outbuf = OPENSSL_malloc(inlen);
 329                         rv = FIPS_cipher(&ctx, outbuf, inbuf, inlen);
 330                         OutputValue(encrypt ? "CT":"PT", outbuf, inlen, out, 0);
 331                         OPENSSL_free(inbuf);
 332                         OPENSSL_free(outbuf);
 333                         OPENSSL_free(key);
 334                         OPENSSL_free(iv);
 335                         iv = key = inbuf = outbuf = NULL;
 336                         }
 337                 }
 338         }
 339
 340 static void ccmencrypt(FILE *in, FILE *out)
 341         {
 342         char buf[2048];
 343         char lbuf[2048];
 344         char *keyword, *value;
 345         long l;
 346         unsigned char *Key = NULL, *Nonce = NULL;
 347         unsigned char *Adata = NULL, *Payload = NULL;
 348         unsigned char *CT = NULL;
 349         int Plen = -1, Nlen = -1, Tlen = -1, Alen = -1;
 350         EVP_CIPHER_CTX ctx;
 351         const EVP_CIPHER *ccm = NULL;
 352         FIPS_cipher_ctx_init(&ctx);
 353
 354         while(fgets(buf,sizeof buf,in) != NULL)
 355                 {
 356                 fputs(buf,out);
 357                 if (!parse_line(&keyword, &value, lbuf, buf))
 358                         continue;
 359
 360                 /* If surrounded by square brackets zap them */
 361                 if (keyword[0] == '[')
 362                         {
 363                         char *p;
 364                         keyword++;
 365                         p = strchr(value, ']');
 366                         if (p)
 367                                 *p = 0;
 368                         }
 369
 370                 if (!strcmp(keyword,"Plen"))
 371                         Plen = atoi(value);
 372                 else if (!strcmp(keyword,"Nlen"))
 373                         Nlen = atoi(value);
 374                 else if (!strcmp(keyword,"Tlen"))
 375                         Tlen = atoi(value);
 376                 else if (!strcmp(keyword,"Alen"))
 377                         Alen = atoi(value);
 378                 else if (!strcmp(keyword,"Key"))
 379                         {
 380                         if (Key)
 381                                 OPENSSL_free(Key);
 382                         Key = hex2bin_m(value, &l);
 383                         if (l == 16)
 384                                 ccm = EVP_aes_128_ccm();
 385                         else if (l == 24)
 386                                 ccm = EVP_aes_192_ccm();
 387                         else if (l == 32)
 388                                 ccm = EVP_aes_256_ccm();
 389                         else
 390                                 {
 391                                 fprintf(stderr, "Inconsistent Key length\n");
 392                                 exit(1);
 393                                 }
 394                         }
 395                 else if (!strcmp(keyword,"Nonce"))
 396                         {
 397                         if (Nonce)
 398                                 OPENSSL_free(Nonce);
 399                         Nonce = hex2bin_m(value, &l);
 400                         if (l != Nlen)
 401                                 {
 402                                 fprintf(stderr, "Inconsistent nonce length\n");
 403                                 exit(1);
 404                                 }
 405                         }
 406                 else if (!strcmp(keyword,"Payload"))
 407                         {
 408                         Payload = hex2bin_m(value, &l);
 409                         if (Plen && l != Plen)
 410                                 {
 411                                 fprintf(stderr, "Inconsistent Payload length\n");
 412                                 exit(1);
 413                                 }
 414                         }
 415                 else if (!strcmp(keyword,"Adata"))
 416                         {
 417                         Adata = hex2bin_m(value, &l);
 418                         if (Alen && l != Alen)
 419                                 {
 420                                 fprintf(stderr, "Inconsistent Payload length\n");
 421                                 exit(1);
 422                                 }
 423                         }
 424                 if (Payload)
 425                         {
 426                         FIPS_cipherinit(&ctx, ccm, NULL, NULL, 1);
 427                         FIPS_cipher_ctx_ctrl(&ctx, EVP_CTRL_CCM_SET_IVLEN, Nlen, 0);
 428                         FIPS_cipher_ctx_ctrl(&ctx, EVP_CTRL_CCM_SET_TAG, Tlen, 0);
 429                         FIPS_cipherinit(&ctx, NULL, Key, Nonce, 1);
 430
 431                         FIPS_cipher(&ctx, NULL, NULL, Plen);
 432                         FIPS_cipher(&ctx, NULL, Adata, Alen);
 433                         CT = OPENSSL_malloc(Plen + Tlen);
 434                         FIPS_cipher(&ctx, CT, Payload, Plen);
 435                         FIPS_cipher_ctx_ctrl(&ctx, EVP_CTRL_CCM_GET_TAG, Tlen,
 436                                                 CT + Plen);
 437                         OutputValue("CT", CT, Plen + Tlen, out, 0);
 438                         OPENSSL_free(CT);
 439                         OPENSSL_free(Payload);
 440                         CT = Payload = NULL;
 441                         }
 442                 }
 443         if (Key)
 444                 OPENSSL_free(Key);
 445         if (Nonce)
 446                 OPENSSL_free(Nonce);
 447         FIPS_cipher_ctx_cleanup(&ctx);
 448         }
 449
 450 int main(int argc,char **argv)
 451         {
 452         int encrypt;
 453         int xts = 0, ccmenc = 0;
 454         FILE *in, *out;
 455         if (argc == 4)
 456                 {
 457                 in = fopen(argv[2], "r");
 458                 if (!in)
 459                         {
 460                         fprintf(stderr, "Error opening input file\n");
 461                         exit(1);
 462                         }
 463                 out = fopen(argv[3], "w");
 464                 if (!out)
 465                         {
 466                         fprintf(stderr, "Error opening output file\n");
 467                         exit(1);
 468                         }
 469                 }
 470         else if (argc == 2)
 471                 {
 472                 in = stdin;
 473                 out = stdout;
 474                 }
 475         else
 476                 {
 477                 fprintf(stderr,"%s [-encrypt|-decrypt]\n",argv[0]);
 478                 exit(1);
 479                 }
 480         fips_algtest_init();
 481         if(!strcmp(argv[1],"-encrypt"))
 482                 encrypt = 1;
 483         else if(!strcmp(argv[1],"-encryptIVext"))
 484                 encrypt = 2;
 485         else if(!strcmp(argv[1],"-decrypt"))
 486                 encrypt = 0;
 487         else if(!strcmp(argv[1],"-ccmencrypt"))
 488                 ccmenc = 1;
 489         else if(!strcmp(argv[1],"-xts"))
 490                 xts = 1;
 491         else
 492                 {
 493                 fprintf(stderr,"Don't know how to %s.\n",argv[1]);
 494                 exit(1);
 495                 }
 496
 497         if (ccmenc)
 498                 ccmencrypt(in, out);
 499         else if (xts)
 500                 xtstest(in, out);
 501         else
 502                 gcmtest(in, out, encrypt);
 503
 504         if (argc == 4)
 505                 {
 506                 fclose(in);
 507                 fclose(out);
 508                 }
 509
 510         return 0;
 511 }
 512
 513 #endif