Example program to test AES-GCM functionality. Used for a workshop

Dependencies:   mbed

Committer:
HannesTschofenig
Date:
Thu Sep 27 06:34:22 2018 +0000
Revision:
0:796d0f61a05b
Example AES-GCM test program

Who changed what in which revision?

UserRevisionLine numberNew contents of line
HannesTschofenig 0:796d0f61a05b 1 /*
HannesTschofenig 0:796d0f61a05b 2 * RFC 1521 base64 encoding/decoding
HannesTschofenig 0:796d0f61a05b 3 *
HannesTschofenig 0:796d0f61a05b 4 * Copyright (C) 2006-2014, Brainspark B.V.
HannesTschofenig 0:796d0f61a05b 5 *
HannesTschofenig 0:796d0f61a05b 6 * This file is part of PolarSSL (http://www.polarssl.org)
HannesTschofenig 0:796d0f61a05b 7 * Lead Maintainer: Paul Bakker <polarssl_maintainer at polarssl.org>
HannesTschofenig 0:796d0f61a05b 8 *
HannesTschofenig 0:796d0f61a05b 9 * All rights reserved.
HannesTschofenig 0:796d0f61a05b 10 *
HannesTschofenig 0:796d0f61a05b 11 * This program is free software; you can redistribute it and/or modify
HannesTschofenig 0:796d0f61a05b 12 * it under the terms of the GNU General Public License as published by
HannesTschofenig 0:796d0f61a05b 13 * the Free Software Foundation; either version 2 of the License, or
HannesTschofenig 0:796d0f61a05b 14 * (at your option) any later version.
HannesTschofenig 0:796d0f61a05b 15 *
HannesTschofenig 0:796d0f61a05b 16 * This program is distributed in the hope that it will be useful,
HannesTschofenig 0:796d0f61a05b 17 * but WITHOUT ANY WARRANTY; without even the implied warranty of
HannesTschofenig 0:796d0f61a05b 18 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
HannesTschofenig 0:796d0f61a05b 19 * GNU General Public License for more details.
HannesTschofenig 0:796d0f61a05b 20 *
HannesTschofenig 0:796d0f61a05b 21 * You should have received a copy of the GNU General Public License along
HannesTschofenig 0:796d0f61a05b 22 * with this program; if not, write to the Free Software Foundation, Inc.,
HannesTschofenig 0:796d0f61a05b 23 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
HannesTschofenig 0:796d0f61a05b 24 */
HannesTschofenig 0:796d0f61a05b 25
HannesTschofenig 0:796d0f61a05b 26 #if !defined(POLARSSL_CONFIG_FILE)
HannesTschofenig 0:796d0f61a05b 27 #include "polarssl/config.h"
HannesTschofenig 0:796d0f61a05b 28 #else
HannesTschofenig 0:796d0f61a05b 29 #include POLARSSL_CONFIG_FILE
HannesTschofenig 0:796d0f61a05b 30 #endif
HannesTschofenig 0:796d0f61a05b 31
HannesTschofenig 0:796d0f61a05b 32 #if defined(POLARSSL_BASE64_C)
HannesTschofenig 0:796d0f61a05b 33
HannesTschofenig 0:796d0f61a05b 34 #include "polarssl/base64.h"
HannesTschofenig 0:796d0f61a05b 35
HannesTschofenig 0:796d0f61a05b 36 #if defined(_MSC_VER) && !defined(EFIX64) && !defined(EFI32)
HannesTschofenig 0:796d0f61a05b 37 #include <basetsd.h>
HannesTschofenig 0:796d0f61a05b 38 typedef UINT32 uint32_t;
HannesTschofenig 0:796d0f61a05b 39 #else
HannesTschofenig 0:796d0f61a05b 40 #include <inttypes.h>
HannesTschofenig 0:796d0f61a05b 41 #endif
HannesTschofenig 0:796d0f61a05b 42
HannesTschofenig 0:796d0f61a05b 43 #if defined(POLARSSL_PLATFORM_C)
HannesTschofenig 0:796d0f61a05b 44 #include "polarssl/platform.h"
HannesTschofenig 0:796d0f61a05b 45 #else
HannesTschofenig 0:796d0f61a05b 46 #define polarssl_printf printf
HannesTschofenig 0:796d0f61a05b 47 #endif
HannesTschofenig 0:796d0f61a05b 48
HannesTschofenig 0:796d0f61a05b 49 static const unsigned char base64_enc_map[64] =
HannesTschofenig 0:796d0f61a05b 50 {
HannesTschofenig 0:796d0f61a05b 51 'A', 'B', 'C', 'D', 'E', 'F', 'G', 'H', 'I', 'J',
HannesTschofenig 0:796d0f61a05b 52 'K', 'L', 'M', 'N', 'O', 'P', 'Q', 'R', 'S', 'T',
HannesTschofenig 0:796d0f61a05b 53 'U', 'V', 'W', 'X', 'Y', 'Z', 'a', 'b', 'c', 'd',
HannesTschofenig 0:796d0f61a05b 54 'e', 'f', 'g', 'h', 'i', 'j', 'k', 'l', 'm', 'n',
HannesTschofenig 0:796d0f61a05b 55 'o', 'p', 'q', 'r', 's', 't', 'u', 'v', 'w', 'x',
HannesTschofenig 0:796d0f61a05b 56 'y', 'z', '0', '1', '2', '3', '4', '5', '6', '7',
HannesTschofenig 0:796d0f61a05b 57 '8', '9', '+', '/'
HannesTschofenig 0:796d0f61a05b 58 };
HannesTschofenig 0:796d0f61a05b 59
HannesTschofenig 0:796d0f61a05b 60 static const unsigned char base64_dec_map[128] =
HannesTschofenig 0:796d0f61a05b 61 {
HannesTschofenig 0:796d0f61a05b 62 127, 127, 127, 127, 127, 127, 127, 127, 127, 127,
HannesTschofenig 0:796d0f61a05b 63 127, 127, 127, 127, 127, 127, 127, 127, 127, 127,
HannesTschofenig 0:796d0f61a05b 64 127, 127, 127, 127, 127, 127, 127, 127, 127, 127,
HannesTschofenig 0:796d0f61a05b 65 127, 127, 127, 127, 127, 127, 127, 127, 127, 127,
HannesTschofenig 0:796d0f61a05b 66 127, 127, 127, 62, 127, 127, 127, 63, 52, 53,
HannesTschofenig 0:796d0f61a05b 67 54, 55, 56, 57, 58, 59, 60, 61, 127, 127,
HannesTschofenig 0:796d0f61a05b 68 127, 64, 127, 127, 127, 0, 1, 2, 3, 4,
HannesTschofenig 0:796d0f61a05b 69 5, 6, 7, 8, 9, 10, 11, 12, 13, 14,
HannesTschofenig 0:796d0f61a05b 70 15, 16, 17, 18, 19, 20, 21, 22, 23, 24,
HannesTschofenig 0:796d0f61a05b 71 25, 127, 127, 127, 127, 127, 127, 26, 27, 28,
HannesTschofenig 0:796d0f61a05b 72 29, 30, 31, 32, 33, 34, 35, 36, 37, 38,
HannesTschofenig 0:796d0f61a05b 73 39, 40, 41, 42, 43, 44, 45, 46, 47, 48,
HannesTschofenig 0:796d0f61a05b 74 49, 50, 51, 127, 127, 127, 127, 127
HannesTschofenig 0:796d0f61a05b 75 };
HannesTschofenig 0:796d0f61a05b 76
HannesTschofenig 0:796d0f61a05b 77 /*
HannesTschofenig 0:796d0f61a05b 78 * Encode a buffer into base64 format
HannesTschofenig 0:796d0f61a05b 79 */
HannesTschofenig 0:796d0f61a05b 80 int base64_encode( unsigned char *dst, size_t *dlen,
HannesTschofenig 0:796d0f61a05b 81 const unsigned char *src, size_t slen )
HannesTschofenig 0:796d0f61a05b 82 {
HannesTschofenig 0:796d0f61a05b 83 size_t i, n;
HannesTschofenig 0:796d0f61a05b 84 int C1, C2, C3;
HannesTschofenig 0:796d0f61a05b 85 unsigned char *p;
HannesTschofenig 0:796d0f61a05b 86
HannesTschofenig 0:796d0f61a05b 87 if( slen == 0 )
HannesTschofenig 0:796d0f61a05b 88 return( 0 );
HannesTschofenig 0:796d0f61a05b 89
HannesTschofenig 0:796d0f61a05b 90 n = (slen << 3) / 6;
HannesTschofenig 0:796d0f61a05b 91
HannesTschofenig 0:796d0f61a05b 92 switch( (slen << 3) - (n * 6) )
HannesTschofenig 0:796d0f61a05b 93 {
HannesTschofenig 0:796d0f61a05b 94 case 2: n += 3; break;
HannesTschofenig 0:796d0f61a05b 95 case 4: n += 2; break;
HannesTschofenig 0:796d0f61a05b 96 default: break;
HannesTschofenig 0:796d0f61a05b 97 }
HannesTschofenig 0:796d0f61a05b 98
HannesTschofenig 0:796d0f61a05b 99 if( *dlen < n + 1 )
HannesTschofenig 0:796d0f61a05b 100 {
HannesTschofenig 0:796d0f61a05b 101 *dlen = n + 1;
HannesTschofenig 0:796d0f61a05b 102 return( POLARSSL_ERR_BASE64_BUFFER_TOO_SMALL );
HannesTschofenig 0:796d0f61a05b 103 }
HannesTschofenig 0:796d0f61a05b 104
HannesTschofenig 0:796d0f61a05b 105 n = (slen / 3) * 3;
HannesTschofenig 0:796d0f61a05b 106
HannesTschofenig 0:796d0f61a05b 107 for( i = 0, p = dst; i < n; i += 3 )
HannesTschofenig 0:796d0f61a05b 108 {
HannesTschofenig 0:796d0f61a05b 109 C1 = *src++;
HannesTschofenig 0:796d0f61a05b 110 C2 = *src++;
HannesTschofenig 0:796d0f61a05b 111 C3 = *src++;
HannesTschofenig 0:796d0f61a05b 112
HannesTschofenig 0:796d0f61a05b 113 *p++ = base64_enc_map[(C1 >> 2) & 0x3F];
HannesTschofenig 0:796d0f61a05b 114 *p++ = base64_enc_map[(((C1 & 3) << 4) + (C2 >> 4)) & 0x3F];
HannesTschofenig 0:796d0f61a05b 115 *p++ = base64_enc_map[(((C2 & 15) << 2) + (C3 >> 6)) & 0x3F];
HannesTschofenig 0:796d0f61a05b 116 *p++ = base64_enc_map[C3 & 0x3F];
HannesTschofenig 0:796d0f61a05b 117 }
HannesTschofenig 0:796d0f61a05b 118
HannesTschofenig 0:796d0f61a05b 119 if( i < slen )
HannesTschofenig 0:796d0f61a05b 120 {
HannesTschofenig 0:796d0f61a05b 121 C1 = *src++;
HannesTschofenig 0:796d0f61a05b 122 C2 = ((i + 1) < slen) ? *src++ : 0;
HannesTschofenig 0:796d0f61a05b 123
HannesTschofenig 0:796d0f61a05b 124 *p++ = base64_enc_map[(C1 >> 2) & 0x3F];
HannesTschofenig 0:796d0f61a05b 125 *p++ = base64_enc_map[(((C1 & 3) << 4) + (C2 >> 4)) & 0x3F];
HannesTschofenig 0:796d0f61a05b 126
HannesTschofenig 0:796d0f61a05b 127 if( (i + 1) < slen )
HannesTschofenig 0:796d0f61a05b 128 *p++ = base64_enc_map[((C2 & 15) << 2) & 0x3F];
HannesTschofenig 0:796d0f61a05b 129 else *p++ = '=';
HannesTschofenig 0:796d0f61a05b 130
HannesTschofenig 0:796d0f61a05b 131 *p++ = '=';
HannesTschofenig 0:796d0f61a05b 132 }
HannesTschofenig 0:796d0f61a05b 133
HannesTschofenig 0:796d0f61a05b 134 *dlen = p - dst;
HannesTschofenig 0:796d0f61a05b 135 *p = 0;
HannesTschofenig 0:796d0f61a05b 136
HannesTschofenig 0:796d0f61a05b 137 return( 0 );
HannesTschofenig 0:796d0f61a05b 138 }
HannesTschofenig 0:796d0f61a05b 139
HannesTschofenig 0:796d0f61a05b 140 /*
HannesTschofenig 0:796d0f61a05b 141 * Decode a base64-formatted buffer
HannesTschofenig 0:796d0f61a05b 142 */
HannesTschofenig 0:796d0f61a05b 143 int base64_decode( unsigned char *dst, size_t *dlen,
HannesTschofenig 0:796d0f61a05b 144 const unsigned char *src, size_t slen )
HannesTschofenig 0:796d0f61a05b 145 {
HannesTschofenig 0:796d0f61a05b 146 size_t i, n;
HannesTschofenig 0:796d0f61a05b 147 uint32_t j, x;
HannesTschofenig 0:796d0f61a05b 148 unsigned char *p;
HannesTschofenig 0:796d0f61a05b 149
HannesTschofenig 0:796d0f61a05b 150 for( i = n = j = 0; i < slen; i++ )
HannesTschofenig 0:796d0f61a05b 151 {
HannesTschofenig 0:796d0f61a05b 152 if( ( slen - i ) >= 2 &&
HannesTschofenig 0:796d0f61a05b 153 src[i] == '\r' && src[i + 1] == '\n' )
HannesTschofenig 0:796d0f61a05b 154 continue;
HannesTschofenig 0:796d0f61a05b 155
HannesTschofenig 0:796d0f61a05b 156 if( src[i] == '\n' )
HannesTschofenig 0:796d0f61a05b 157 continue;
HannesTschofenig 0:796d0f61a05b 158
HannesTschofenig 0:796d0f61a05b 159 if( src[i] == '=' && ++j > 2 )
HannesTschofenig 0:796d0f61a05b 160 return( POLARSSL_ERR_BASE64_INVALID_CHARACTER );
HannesTschofenig 0:796d0f61a05b 161
HannesTschofenig 0:796d0f61a05b 162 if( src[i] > 127 || base64_dec_map[src[i]] == 127 )
HannesTschofenig 0:796d0f61a05b 163 return( POLARSSL_ERR_BASE64_INVALID_CHARACTER );
HannesTschofenig 0:796d0f61a05b 164
HannesTschofenig 0:796d0f61a05b 165 if( base64_dec_map[src[i]] < 64 && j != 0 )
HannesTschofenig 0:796d0f61a05b 166 return( POLARSSL_ERR_BASE64_INVALID_CHARACTER );
HannesTschofenig 0:796d0f61a05b 167
HannesTschofenig 0:796d0f61a05b 168 n++;
HannesTschofenig 0:796d0f61a05b 169 }
HannesTschofenig 0:796d0f61a05b 170
HannesTschofenig 0:796d0f61a05b 171 if( n == 0 )
HannesTschofenig 0:796d0f61a05b 172 return( 0 );
HannesTschofenig 0:796d0f61a05b 173
HannesTschofenig 0:796d0f61a05b 174 n = ((n * 6) + 7) >> 3;
HannesTschofenig 0:796d0f61a05b 175
HannesTschofenig 0:796d0f61a05b 176 if( dst == NULL || *dlen < n )
HannesTschofenig 0:796d0f61a05b 177 {
HannesTschofenig 0:796d0f61a05b 178 *dlen = n;
HannesTschofenig 0:796d0f61a05b 179 return( POLARSSL_ERR_BASE64_BUFFER_TOO_SMALL );
HannesTschofenig 0:796d0f61a05b 180 }
HannesTschofenig 0:796d0f61a05b 181
HannesTschofenig 0:796d0f61a05b 182 for( j = 3, n = x = 0, p = dst; i > 0; i--, src++ )
HannesTschofenig 0:796d0f61a05b 183 {
HannesTschofenig 0:796d0f61a05b 184 if( *src == '\r' || *src == '\n' )
HannesTschofenig 0:796d0f61a05b 185 continue;
HannesTschofenig 0:796d0f61a05b 186
HannesTschofenig 0:796d0f61a05b 187 j -= ( base64_dec_map[*src] == 64 );
HannesTschofenig 0:796d0f61a05b 188 x = (x << 6) | ( base64_dec_map[*src] & 0x3F );
HannesTschofenig 0:796d0f61a05b 189
HannesTschofenig 0:796d0f61a05b 190 if( ++n == 4 )
HannesTschofenig 0:796d0f61a05b 191 {
HannesTschofenig 0:796d0f61a05b 192 n = 0;
HannesTschofenig 0:796d0f61a05b 193 if( j > 0 ) *p++ = (unsigned char)( x >> 16 );
HannesTschofenig 0:796d0f61a05b 194 if( j > 1 ) *p++ = (unsigned char)( x >> 8 );
HannesTschofenig 0:796d0f61a05b 195 if( j > 2 ) *p++ = (unsigned char)( x );
HannesTschofenig 0:796d0f61a05b 196 }
HannesTschofenig 0:796d0f61a05b 197 }
HannesTschofenig 0:796d0f61a05b 198
HannesTschofenig 0:796d0f61a05b 199 *dlen = p - dst;
HannesTschofenig 0:796d0f61a05b 200
HannesTschofenig 0:796d0f61a05b 201 return( 0 );
HannesTschofenig 0:796d0f61a05b 202 }
HannesTschofenig 0:796d0f61a05b 203
HannesTschofenig 0:796d0f61a05b 204 #if defined(POLARSSL_SELF_TEST)
HannesTschofenig 0:796d0f61a05b 205
HannesTschofenig 0:796d0f61a05b 206 #include <string.h>
HannesTschofenig 0:796d0f61a05b 207 #include <stdio.h>
HannesTschofenig 0:796d0f61a05b 208
HannesTschofenig 0:796d0f61a05b 209 static const unsigned char base64_test_dec[64] =
HannesTschofenig 0:796d0f61a05b 210 {
HannesTschofenig 0:796d0f61a05b 211 0x24, 0x48, 0x6E, 0x56, 0x87, 0x62, 0x5A, 0xBD,
HannesTschofenig 0:796d0f61a05b 212 0xBF, 0x17, 0xD9, 0xA2, 0xC4, 0x17, 0x1A, 0x01,
HannesTschofenig 0:796d0f61a05b 213 0x94, 0xED, 0x8F, 0x1E, 0x11, 0xB3, 0xD7, 0x09,
HannesTschofenig 0:796d0f61a05b 214 0x0C, 0xB6, 0xE9, 0x10, 0x6F, 0x22, 0xEE, 0x13,
HannesTschofenig 0:796d0f61a05b 215 0xCA, 0xB3, 0x07, 0x05, 0x76, 0xC9, 0xFA, 0x31,
HannesTschofenig 0:796d0f61a05b 216 0x6C, 0x08, 0x34, 0xFF, 0x8D, 0xC2, 0x6C, 0x38,
HannesTschofenig 0:796d0f61a05b 217 0x00, 0x43, 0xE9, 0x54, 0x97, 0xAF, 0x50, 0x4B,
HannesTschofenig 0:796d0f61a05b 218 0xD1, 0x41, 0xBA, 0x95, 0x31, 0x5A, 0x0B, 0x97
HannesTschofenig 0:796d0f61a05b 219 };
HannesTschofenig 0:796d0f61a05b 220
HannesTschofenig 0:796d0f61a05b 221 static const unsigned char base64_test_enc[] =
HannesTschofenig 0:796d0f61a05b 222 "JEhuVodiWr2/F9mixBcaAZTtjx4Rs9cJDLbpEG8i7hPK"
HannesTschofenig 0:796d0f61a05b 223 "swcFdsn6MWwINP+Nwmw4AEPpVJevUEvRQbqVMVoLlw==";
HannesTschofenig 0:796d0f61a05b 224
HannesTschofenig 0:796d0f61a05b 225 /*
HannesTschofenig 0:796d0f61a05b 226 * Checkup routine
HannesTschofenig 0:796d0f61a05b 227 */
HannesTschofenig 0:796d0f61a05b 228 int base64_self_test( int verbose )
HannesTschofenig 0:796d0f61a05b 229 {
HannesTschofenig 0:796d0f61a05b 230 size_t len;
HannesTschofenig 0:796d0f61a05b 231 const unsigned char *src;
HannesTschofenig 0:796d0f61a05b 232 unsigned char buffer[128];
HannesTschofenig 0:796d0f61a05b 233
HannesTschofenig 0:796d0f61a05b 234 if( verbose != 0 )
HannesTschofenig 0:796d0f61a05b 235 polarssl_printf( " Base64 encoding test: " );
HannesTschofenig 0:796d0f61a05b 236
HannesTschofenig 0:796d0f61a05b 237 len = sizeof( buffer );
HannesTschofenig 0:796d0f61a05b 238 src = base64_test_dec;
HannesTschofenig 0:796d0f61a05b 239
HannesTschofenig 0:796d0f61a05b 240 if( base64_encode( buffer, &len, src, 64 ) != 0 ||
HannesTschofenig 0:796d0f61a05b 241 memcmp( base64_test_enc, buffer, 88 ) != 0 )
HannesTschofenig 0:796d0f61a05b 242 {
HannesTschofenig 0:796d0f61a05b 243 if( verbose != 0 )
HannesTschofenig 0:796d0f61a05b 244 polarssl_printf( "failed\n" );
HannesTschofenig 0:796d0f61a05b 245
HannesTschofenig 0:796d0f61a05b 246 return( 1 );
HannesTschofenig 0:796d0f61a05b 247 }
HannesTschofenig 0:796d0f61a05b 248
HannesTschofenig 0:796d0f61a05b 249 if( verbose != 0 )
HannesTschofenig 0:796d0f61a05b 250 polarssl_printf( "passed\n Base64 decoding test: " );
HannesTschofenig 0:796d0f61a05b 251
HannesTschofenig 0:796d0f61a05b 252 len = sizeof( buffer );
HannesTschofenig 0:796d0f61a05b 253 src = base64_test_enc;
HannesTschofenig 0:796d0f61a05b 254
HannesTschofenig 0:796d0f61a05b 255 if( base64_decode( buffer, &len, src, 88 ) != 0 ||
HannesTschofenig 0:796d0f61a05b 256 memcmp( base64_test_dec, buffer, 64 ) != 0 )
HannesTschofenig 0:796d0f61a05b 257 {
HannesTschofenig 0:796d0f61a05b 258 if( verbose != 0 )
HannesTschofenig 0:796d0f61a05b 259 polarssl_printf( "failed\n" );
HannesTschofenig 0:796d0f61a05b 260
HannesTschofenig 0:796d0f61a05b 261 return( 1 );
HannesTschofenig 0:796d0f61a05b 262 }
HannesTschofenig 0:796d0f61a05b 263
HannesTschofenig 0:796d0f61a05b 264 if( verbose != 0 )
HannesTschofenig 0:796d0f61a05b 265 polarssl_printf( "passed\n\n" );
HannesTschofenig 0:796d0f61a05b 266
HannesTschofenig 0:796d0f61a05b 267 return( 0 );
HannesTschofenig 0:796d0f61a05b 268 }
HannesTschofenig 0:796d0f61a05b 269
HannesTschofenig 0:796d0f61a05b 270 #endif /* POLARSSL_SELF_TEST */
HannesTschofenig 0:796d0f61a05b 271
HannesTschofenig 0:796d0f61a05b 272 #endif /* POLARSSL_BASE64_C */
HannesTschofenig 0:796d0f61a05b 273
HannesTschofenig 0:796d0f61a05b 274