Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 1 | /* BEGIN_HEADER */ |
Manuel Pégourié-Gonnard | 7f80997 | 2015-03-09 17:05:11 +0000 | [diff] [blame] | 2 | #include "mbedtls/pkcs5.h" |
Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 3 | /* END_HEADER */ |
Paul Bakker | b0c19a4 | 2013-06-24 19:26:38 +0200 | [diff] [blame] | 4 | |
Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 5 | /* BEGIN_DEPENDENCIES |
Manuel Pégourié-Gonnard | 2cf5a7c | 2015-04-08 12:49:31 +0200 | [diff] [blame] | 6 | * depends_on:MBEDTLS_PKCS5_C |
Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 7 | * END_DEPENDENCIES |
| 8 | */ |
Paul Bakker | b0c19a4 | 2013-06-24 19:26:38 +0200 | [diff] [blame] | 9 | |
Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 10 | /* BEGIN_CASE */ |
Azim Khan | f1aaec9 | 2017-05-30 14:23:15 +0100 | [diff] [blame] | 11 | void pbkdf2_hmac( int hash, uint8_t * pw_str, uint32_t pw_len, |
| 12 | uint8_t * salt_str, uint32_t salt_len, int it_cnt, |
| 13 | int key_len, uint8_t * result_key_string, |
| 14 | uint32_t result_key_string_len ) |
Paul Bakker | b0c19a4 | 2013-06-24 19:26:38 +0200 | [diff] [blame] | 15 | { |
Manuel Pégourié-Gonnard | 2cf5a7c | 2015-04-08 12:49:31 +0200 | [diff] [blame] | 16 | mbedtls_md_context_t ctx; |
| 17 | const mbedtls_md_info_t *info; |
Paul Bakker | b0c19a4 | 2013-06-24 19:26:38 +0200 | [diff] [blame] | 18 | |
Paul Bakker | b0c19a4 | 2013-06-24 19:26:38 +0200 | [diff] [blame] | 19 | unsigned char key[100]; |
| 20 | |
Manuel Pégourié-Gonnard | 2cf5a7c | 2015-04-08 12:49:31 +0200 | [diff] [blame] | 21 | mbedtls_md_init( &ctx ); |
Paul Bakker | d2a2d61 | 2014-07-01 15:45:49 +0200 | [diff] [blame] | 22 | |
Manuel Pégourié-Gonnard | 2cf5a7c | 2015-04-08 12:49:31 +0200 | [diff] [blame] | 23 | info = mbedtls_md_info_from_type( hash ); |
Paul Bakker | b0c19a4 | 2013-06-24 19:26:38 +0200 | [diff] [blame] | 24 | TEST_ASSERT( info != NULL ); |
Manuel Pégourié-Gonnard | 2cf5a7c | 2015-04-08 12:49:31 +0200 | [diff] [blame] | 25 | TEST_ASSERT( mbedtls_md_setup( &ctx, info, 1 ) == 0 ); |
| 26 | TEST_ASSERT( mbedtls_pkcs5_pbkdf2_hmac( &ctx, pw_str, pw_len, salt_str, salt_len, |
Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 27 | it_cnt, key_len, key ) == 0 ); |
Paul Bakker | b0c19a4 | 2013-06-24 19:26:38 +0200 | [diff] [blame] | 28 | |
Azim Khan | f1aaec9 | 2017-05-30 14:23:15 +0100 | [diff] [blame] | 29 | TEST_ASSERT( hexcmp( key, result_key_string, key_len, result_key_string_len ) == 0 ); |
Paul Bakker | bd51b26 | 2014-07-10 15:26:12 +0200 | [diff] [blame] | 30 | |
| 31 | exit: |
Manuel Pégourié-Gonnard | 2cf5a7c | 2015-04-08 12:49:31 +0200 | [diff] [blame] | 32 | mbedtls_md_free( &ctx ); |
Paul Bakker | b0c19a4 | 2013-06-24 19:26:38 +0200 | [diff] [blame] | 33 | } |
Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 34 | /* END_CASE */ |
Manuel Pégourié-Gonnard | 66aca93 | 2014-06-12 13:14:55 +0200 | [diff] [blame] | 35 | |
Andres Amaya Garcia | 576d474 | 2018-03-27 20:53:56 +0100 | [diff] [blame] | 36 | /* BEGIN_CASE depends_on:MBEDTLS_ASN1_PARSE_C */ |
Manuel Pégourié-Gonnard | 2cf5a7c | 2015-04-08 12:49:31 +0200 | [diff] [blame] | 37 | void mbedtls_pkcs5_pbes2( int params_tag, char *params_hex, char *pw_hex, |
Manuel Pégourié-Gonnard | 66aca93 | 2014-06-12 13:14:55 +0200 | [diff] [blame] | 38 | char *data_hex, int ref_ret, char *ref_out_hex ) |
| 39 | { |
| 40 | int my_ret; |
Manuel Pégourié-Gonnard | 2cf5a7c | 2015-04-08 12:49:31 +0200 | [diff] [blame] | 41 | mbedtls_asn1_buf params; |
Paul Bakker | bd51b26 | 2014-07-10 15:26:12 +0200 | [diff] [blame] | 42 | unsigned char *my_out = NULL, *ref_out = NULL, *data = NULL, *pw = NULL; |
Manuel Pégourié-Gonnard | 66aca93 | 2014-06-12 13:14:55 +0200 | [diff] [blame] | 43 | size_t ref_out_len, data_len, pw_len; |
| 44 | |
| 45 | params.tag = params_tag; |
| 46 | params.p = unhexify_alloc( params_hex, ¶ms.len ); |
| 47 | |
| 48 | data = unhexify_alloc( data_hex, &data_len ); |
| 49 | pw = unhexify_alloc( pw_hex, &pw_len ); |
| 50 | ref_out = unhexify_alloc( ref_out_hex, &ref_out_len ); |
Manuel Pégourié-Gonnard | 0dc5e0d | 2014-06-13 21:09:26 +0200 | [diff] [blame] | 51 | my_out = zero_alloc( ref_out_len ); |
Manuel Pégourié-Gonnard | 66aca93 | 2014-06-12 13:14:55 +0200 | [diff] [blame] | 52 | |
Manuel Pégourié-Gonnard | 2cf5a7c | 2015-04-08 12:49:31 +0200 | [diff] [blame] | 53 | my_ret = mbedtls_pkcs5_pbes2( ¶ms, MBEDTLS_PKCS5_DECRYPT, |
Manuel Pégourié-Gonnard | 66aca93 | 2014-06-12 13:14:55 +0200 | [diff] [blame] | 54 | pw, pw_len, data, data_len, my_out ); |
| 55 | TEST_ASSERT( my_ret == ref_ret ); |
| 56 | |
| 57 | if( ref_ret == 0 ) |
| 58 | TEST_ASSERT( memcmp( my_out, ref_out, ref_out_len ) == 0 ); |
| 59 | |
Paul Bakker | bd51b26 | 2014-07-10 15:26:12 +0200 | [diff] [blame] | 60 | exit: |
Manuel Pégourié-Gonnard | 2cf5a7c | 2015-04-08 12:49:31 +0200 | [diff] [blame] | 61 | mbedtls_free( params.p ); |
| 62 | mbedtls_free( data ); |
| 63 | mbedtls_free( pw ); |
| 64 | mbedtls_free( ref_out ); |
| 65 | mbedtls_free( my_out ); |
Manuel Pégourié-Gonnard | 66aca93 | 2014-06-12 13:14:55 +0200 | [diff] [blame] | 66 | } |
| 67 | /* END_CASE */ |
Paul Bakker | 81c6091 | 2016-07-19 14:54:57 +0100 | [diff] [blame] | 68 | |
| 69 | /* BEGIN_CASE depends_on:MBEDTLS_SELF_TEST */ |
Azim Khan | f1aaec9 | 2017-05-30 14:23:15 +0100 | [diff] [blame] | 70 | void pkcs5_selftest( ) |
Paul Bakker | 81c6091 | 2016-07-19 14:54:57 +0100 | [diff] [blame] | 71 | { |
Andres AG | 93012e8 | 2016-09-09 09:10:28 +0100 | [diff] [blame] | 72 | TEST_ASSERT( mbedtls_pkcs5_self_test( 1 ) == 0 ); |
Paul Bakker | 81c6091 | 2016-07-19 14:54:57 +0100 | [diff] [blame] | 73 | } |
| 74 | /* END_CASE */ |