usr/src/common/net/wanboot/crypt/cbc_test.c

   1 /*
   2  * CDDL HEADER START
   3  *
   4  * The contents of this file are subject to the terms of the
   5  * Common Development and Distribution License, Version 1.0 only
   6  * (the "License").  You may not use this file except in compliance
   7  * with the License.
   8  *
   9  * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
  10  * or http://www.opensolaris.org/os/licensing.
  11  * See the License for the specific language governing permissions
  12  * and limitations under the License.
  13  *
  14  * When distributing Covered Code, include this CDDL HEADER in each
  15  * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
  16  * If applicable, add the following below this CDDL HEADER, with the
  17  * fields enclosed by brackets "[]" replaced with your own identifying
  18  * information: Portions Copyright [yyyy] [name of copyright owner]
  19  *
  20  * CDDL HEADER END
  21  */
  22 /*
  23  * Copyright 2002-2003 Sun Microsystems, Inc.  All rights reserved.
  24  * Use is subject to license terms.
  25  */
  26
  27 #pragma ident   "%Z%%M% %I%     %E% SMI"
  28
  29 /*
  30  * Assertion based test of the CBC implementation.
  31  *
  32  * This test can be used to the CBC implementation using either
  33  * 3DES, AES128, AES192 or AES256. The test string above is encrypted
  34  * and then decrypted using one of the algorithms and keys below. The test
  35  * passes if the decrypted string is the same as the original. Note,
  36  * that this test should not be used to test the underlying algorithms
  37  * and relies on the correctness of those algorithms.
  38  */
  39
  40 #include <stdio.h>
  41 #include <strings.h>
  42
  43 #include "cbc.h"
  44 #include "des3.h"
  45 #include "aes.h"
  46 #include "cbc_test.h"
  47
  48 #define CBC_MAX_KEY_SIZE        AES_256_KEY_SIZE
  49 #define CBC_MAX_BLOCK_SIZE      AES_BLOCK_SIZE
  50 #define CBC_MIN_BLOCK_SIZE      DES3_BLOCK_SIZE
  51 #define CBC_MAX_IV_SIZE         AES_IV_SIZE
  52
  53 #define DES3_KEY        "01234567"
  54 #define AES_128_KEY     "0123456789ABCDEF"
  55 #define AES_192_KEY     "0123456789ABCDEFHIJKLMNO"
  56 #define AES_256_KEY     "0123456789ABCDEFHIJKLMNOPQRSTUVW"
  57
  58 #define TEST_BLOCK_SIZE (CBC_MAX_BLOCK_SIZE * 2)
  59 #define TEST_SIZE       (TEST_BLOCK_SIZE * 2)
  60 #define TEST "This test is successful if this string has a period at the end."
  61
  62 int
  63 cbctest(int type)
  64 {
  65         unsigned char test_string[TEST_SIZE];
  66         char iv[CBC_MAX_IV_SIZE];
  67
  68         cbc_handle_t ch;
  69         void *eh;
  70         int ret;
  71         int i;
  72
  73         switch (type) {
  74         case CBC_DES3_TYPE:
  75                 ret = des3_init(&eh);
  76                 break;
  77         case CBC_AES_128_TYPE:
  78                 ret = aes_init(&eh);
  79                 break;
  80         case CBC_AES_192_TYPE:
  81                 ret = aes_init(&eh);
  82                 break;
  83         case CBC_AES_256_TYPE:
  84                 ret = aes_init(&eh);
  85                 break;
  86         default:
  87                 (void) printf("Illegal encryption type\n");
  88                 return (-1);
  89         }
  90
  91         if (ret != 0) {
  92                 (void) printf("Error initializing encryption algorithm\n");
  93                 return (-1);
  94         }
  95
  96         bzero(iv, CBC_MAX_IV_SIZE);
  97
  98         switch (type) {
  99         case CBC_DES3_TYPE:
 100                 des3_key(eh, (uint8_t *)DES3_KEY);
 101                 cbc_makehandle(&ch, eh, DES3_KEY_SIZE, DES3_BLOCK_SIZE,
 102                     DES3_IV_SIZE, des3_encrypt, des3_decrypt);
 103                 break;
 104         case CBC_AES_128_TYPE:
 105                 aes_key(eh, (uint8_t *)AES_128_KEY, AES_128_KEY_SIZE);
 106                 cbc_makehandle(&ch, eh, AES_128_KEY_SIZE, AES_BLOCK_SIZE,
 107                     AES_IV_SIZE, aes_encrypt, aes_decrypt);
 108                 break;
 109         case CBC_AES_192_TYPE:
 110                 aes_key(eh, (uint8_t *)AES_192_KEY, AES_192_KEY_SIZE);
 111                 cbc_makehandle(&ch, eh, AES_192_KEY_SIZE, AES_BLOCK_SIZE,
 112                     AES_IV_SIZE, aes_encrypt, aes_decrypt);
 113                 break;
 114         case CBC_AES_256_TYPE:
 115                 aes_key(eh, (uint8_t *)AES_256_KEY, AES_256_KEY_SIZE);
 116                 cbc_makehandle(&ch, eh, AES_256_KEY_SIZE, AES_BLOCK_SIZE,
 117                     AES_IV_SIZE, aes_encrypt, aes_decrypt);
 118                 break;
 119         default:
 120                 /* Should not happen */
 121                 (void) printf("Illegal encryption type\n");
 122                 return (-1);
 123         }
 124
 125         (void) strcpy((char *)test_string, TEST);
 126
 127         for (i = 0; i < TEST_SIZE; i += TEST_BLOCK_SIZE) {
 128                 (void) cbc_encrypt(&ch, (uint8_t *)&test_string[i],
 129                     TEST_BLOCK_SIZE, (uint8_t *)iv);
 130         }
 131
 132         if (strcmp((char *)test_string, TEST) == 0) {
 133                 (void) printf("FAILED [Encryption]\n");
 134                 goto out;
 135         }
 136
 137         bzero(iv, CBC_MAX_IV_SIZE);
 138
 139         for (i = 0; i < TEST_SIZE; i += TEST_BLOCK_SIZE) {
 140                 (void) cbc_decrypt(&ch, (uint8_t *)&test_string[i],
 141                     TEST_BLOCK_SIZE, (uint8_t *)iv);
 142         }
 143
 144         if (strcmp((char *)test_string, TEST) == 0) {
 145                 (void) printf("PASSED\n");
 146         } else {
 147                 (void) printf("FAILED [Decryption]\n");
 148         }
 149
 150 out:
 151         switch (type) {
 152         case CBC_DES3_TYPE:
 153                 des3_fini(eh);
 154                 break;
 155         case CBC_AES_128_TYPE:
 156         case CBC_AES_192_TYPE:
 157         case CBC_AES_256_TYPE:
 158                 aes_fini(eh);
 159                 break;
 160         default:
 161                 /* Should not happen */
 162                 (void) printf("Illegal encryption type\n");
 163                 return (-1);
 164         }
 165
 166         return (0);
 167 }