| Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 1 | /* BEGIN_HEADER */ | 
| Paul Bakker | 89e80c9 | 2012-03-20 13:50:09 +0000 | [diff] [blame] | 2 | #include <polarssl/gcm.h> | 
| Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 3 | /* END_HEADER */ | 
| Paul Bakker | 89e80c9 | 2012-03-20 13:50:09 +0000 | [diff] [blame] | 4 |  | 
| Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 5 | /* BEGIN_DEPENDENCIES | 
|  | 6 | * depends_on:POLARSSL_GCM_C | 
|  | 7 | * END_DEPENDENCIES | 
|  | 8 | */ | 
| Paul Bakker | 89e80c9 | 2012-03-20 13:50:09 +0000 | [diff] [blame] | 9 |  | 
| Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 10 | /* BEGIN_CASE */ | 
|  | 11 | void gcm_encrypt_and_tag( char *hex_key_string, char *hex_src_string, | 
|  | 12 | char *hex_iv_string, char *hex_add_string, | 
|  | 13 | char *hex_dst_string, int tag_len_bits, | 
|  | 14 | char *hex_tag_string, int  init_result ) | 
| Paul Bakker | 89e80c9 | 2012-03-20 13:50:09 +0000 | [diff] [blame] | 15 | { | 
|  | 16 | unsigned char key_str[128]; | 
|  | 17 | unsigned char src_str[128]; | 
|  | 18 | unsigned char dst_str[257]; | 
|  | 19 | unsigned char iv_str[128]; | 
|  | 20 | unsigned char add_str[128]; | 
|  | 21 | unsigned char tag_str[128]; | 
|  | 22 | unsigned char output[128]; | 
|  | 23 | unsigned char tag_output[16]; | 
|  | 24 | gcm_context ctx; | 
|  | 25 | unsigned int key_len; | 
| Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 26 | size_t pt_len, iv_len, add_len, tag_len = tag_len_bits / 8; | 
| Paul Bakker | 89e80c9 | 2012-03-20 13:50:09 +0000 | [diff] [blame] | 27 |  | 
|  | 28 | memset(key_str, 0x00, 128); | 
|  | 29 | memset(src_str, 0x00, 128); | 
| Paul Bakker | 68b6d88 | 2012-09-08 14:04:13 +0000 | [diff] [blame] | 30 | memset(dst_str, 0x00, 257); | 
| Paul Bakker | 89e80c9 | 2012-03-20 13:50:09 +0000 | [diff] [blame] | 31 | memset(iv_str, 0x00, 128); | 
|  | 32 | memset(add_str, 0x00, 128); | 
|  | 33 | memset(tag_str, 0x00, 128); | 
|  | 34 | memset(output, 0x00, 128); | 
|  | 35 | memset(tag_output, 0x00, 16); | 
|  | 36 |  | 
| Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 37 | key_len = unhexify( key_str, hex_key_string ); | 
|  | 38 | pt_len = unhexify( src_str, hex_src_string ); | 
|  | 39 | iv_len = unhexify( iv_str, hex_iv_string ); | 
|  | 40 | add_len = unhexify( add_str, hex_add_string ); | 
| Paul Bakker | 89e80c9 | 2012-03-20 13:50:09 +0000 | [diff] [blame] | 41 |  | 
| Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 42 | TEST_ASSERT( gcm_init( &ctx, key_str, key_len * 8 ) == init_result ); | 
|  | 43 | if( init_result == 0 ) | 
| Paul Bakker | 89e80c9 | 2012-03-20 13:50:09 +0000 | [diff] [blame] | 44 | { | 
|  | 45 | TEST_ASSERT( gcm_crypt_and_tag( &ctx, GCM_ENCRYPT, pt_len, iv_str, iv_len, add_str, add_len, src_str, output, tag_len, tag_output ) == 0 ); | 
|  | 46 | hexify( dst_str, output, pt_len ); | 
|  | 47 | hexify( tag_str, tag_output, tag_len ); | 
|  | 48 |  | 
| Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 49 | TEST_ASSERT( strcmp( (char *) dst_str, hex_dst_string ) == 0 ); | 
|  | 50 | TEST_ASSERT( strcmp( (char *) tag_str, hex_tag_string ) == 0 ); | 
| Paul Bakker | 89e80c9 | 2012-03-20 13:50:09 +0000 | [diff] [blame] | 51 | } | 
|  | 52 | } | 
| Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 53 | /* END_CASE */ | 
| Paul Bakker | 89e80c9 | 2012-03-20 13:50:09 +0000 | [diff] [blame] | 54 |  | 
| Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 55 | /* BEGIN_CASE */ | 
|  | 56 | void gcm_decrypt_and_verify( char *hex_key_string, char *hex_src_string, | 
|  | 57 | char *hex_iv_string, char *hex_add_string, | 
|  | 58 | int tag_len_bits, char *hex_tag_string, | 
|  | 59 | char *pt_result, int init_result ) | 
| Paul Bakker | 89e80c9 | 2012-03-20 13:50:09 +0000 | [diff] [blame] | 60 | { | 
|  | 61 | unsigned char key_str[128]; | 
|  | 62 | unsigned char src_str[128]; | 
|  | 63 | unsigned char dst_str[257]; | 
|  | 64 | unsigned char iv_str[128]; | 
|  | 65 | unsigned char add_str[128]; | 
|  | 66 | unsigned char tag_str[128]; | 
|  | 67 | unsigned char output[128]; | 
|  | 68 | gcm_context ctx; | 
|  | 69 | unsigned int key_len; | 
| Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 70 | size_t pt_len, iv_len, add_len, tag_len = tag_len_bits / 8; | 
| Paul Bakker | 89e80c9 | 2012-03-20 13:50:09 +0000 | [diff] [blame] | 71 | int ret; | 
|  | 72 |  | 
|  | 73 | memset(key_str, 0x00, 128); | 
|  | 74 | memset(src_str, 0x00, 128); | 
| Paul Bakker | 68b6d88 | 2012-09-08 14:04:13 +0000 | [diff] [blame] | 75 | memset(dst_str, 0x00, 257); | 
| Paul Bakker | 89e80c9 | 2012-03-20 13:50:09 +0000 | [diff] [blame] | 76 | memset(iv_str, 0x00, 128); | 
|  | 77 | memset(add_str, 0x00, 128); | 
|  | 78 | memset(tag_str, 0x00, 128); | 
|  | 79 | memset(output, 0x00, 128); | 
|  | 80 |  | 
| Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 81 | key_len = unhexify( key_str, hex_key_string ); | 
|  | 82 | pt_len = unhexify( src_str, hex_src_string ); | 
|  | 83 | iv_len = unhexify( iv_str, hex_iv_string ); | 
|  | 84 | add_len = unhexify( add_str, hex_add_string ); | 
|  | 85 | unhexify( tag_str, hex_tag_string ); | 
| Paul Bakker | 89e80c9 | 2012-03-20 13:50:09 +0000 | [diff] [blame] | 86 |  | 
| Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 87 | TEST_ASSERT( gcm_init( &ctx, key_str, key_len * 8 ) == init_result ); | 
|  | 88 | if( init_result == 0 ) | 
| Paul Bakker | 89e80c9 | 2012-03-20 13:50:09 +0000 | [diff] [blame] | 89 | { | 
|  | 90 | ret = gcm_auth_decrypt( &ctx, pt_len, iv_str, iv_len, add_str, add_len, tag_str, tag_len, src_str, output ); | 
|  | 91 |  | 
| Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 92 | if( strcmp( "FAIL", pt_result ) == 0 ) | 
| Paul Bakker | 89e80c9 | 2012-03-20 13:50:09 +0000 | [diff] [blame] | 93 | { | 
|  | 94 | TEST_ASSERT( ret == POLARSSL_ERR_GCM_AUTH_FAILED ); | 
|  | 95 | } | 
|  | 96 | else | 
|  | 97 | { | 
|  | 98 | hexify( dst_str, output, pt_len ); | 
|  | 99 |  | 
| Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 100 | TEST_ASSERT( strcmp( (char *) dst_str, pt_result ) == 0 ); | 
| Paul Bakker | 89e80c9 | 2012-03-20 13:50:09 +0000 | [diff] [blame] | 101 | } | 
|  | 102 | } | 
|  | 103 | } | 
| Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 104 | /* END_CASE */ | 
| Paul Bakker | 89e80c9 | 2012-03-20 13:50:09 +0000 | [diff] [blame] | 105 |  | 
| Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 106 | /* BEGIN_CASE */ | 
|  | 107 | void gcm_selftest() | 
| Paul Bakker | 89e80c9 | 2012-03-20 13:50:09 +0000 | [diff] [blame] | 108 | { | 
|  | 109 | TEST_ASSERT( gcm_self_test( 0 ) == 0 ); | 
|  | 110 | } | 
| Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 111 | /* END_CASE */ |