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" |
Waleed Elmelegy | a230735 | 2023-07-21 16:20:17 +0100 | [diff] [blame] | 3 | #include "mbedtls/cipher.h" |
Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 4 | /* END_HEADER */ |
Paul Bakker | b0c19a4 | 2013-06-24 19:26:38 +0200 | [diff] [blame] | 5 | |
Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 6 | /* BEGIN_DEPENDENCIES |
Manuel Pégourié-Gonnard | 2cf5a7c | 2015-04-08 12:49:31 +0200 | [diff] [blame] | 7 | * depends_on:MBEDTLS_PKCS5_C |
Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 8 | * END_DEPENDENCIES |
| 9 | */ |
Paul Bakker | b0c19a4 | 2013-06-24 19:26:38 +0200 | [diff] [blame] | 10 | |
Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 11 | /* BEGIN_CASE */ |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 12 | void pbkdf2_hmac(int hash, data_t *pw_str, data_t *salt_str, |
| 13 | int it_cnt, int key_len, data_t *result_key_string) |
Paul Bakker | b0c19a4 | 2013-06-24 19:26:38 +0200 | [diff] [blame] | 14 | { |
Manuel Pégourié-Gonnard | 2cf5a7c | 2015-04-08 12:49:31 +0200 | [diff] [blame] | 15 | mbedtls_md_context_t ctx; |
| 16 | const mbedtls_md_info_t *info; |
Paul Bakker | b0c19a4 | 2013-06-24 19:26:38 +0200 | [diff] [blame] | 17 | |
Paul Bakker | b0c19a4 | 2013-06-24 19:26:38 +0200 | [diff] [blame] | 18 | unsigned char key[100]; |
| 19 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 20 | mbedtls_md_init(&ctx); |
Paul Bakker | d2a2d61 | 2014-07-01 15:45:49 +0200 | [diff] [blame] | 21 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 22 | info = mbedtls_md_info_from_type(hash); |
| 23 | TEST_ASSERT(info != NULL); |
| 24 | TEST_ASSERT(mbedtls_md_setup(&ctx, info, 1) == 0); |
| 25 | TEST_ASSERT(mbedtls_pkcs5_pbkdf2_hmac(&ctx, pw_str->x, pw_str->len, salt_str->x, salt_str->len, |
| 26 | it_cnt, key_len, key) == 0); |
Paul Bakker | b0c19a4 | 2013-06-24 19:26:38 +0200 | [diff] [blame] | 27 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 28 | TEST_ASSERT(mbedtls_test_hexcmp(key, result_key_string->x, |
| 29 | key_len, result_key_string->len) == 0); |
Paul Bakker | bd51b26 | 2014-07-10 15:26:12 +0200 | [diff] [blame] | 30 | |
| 31 | exit: |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [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 */ |
Waleed Elmelegy | 412629c | 2023-07-19 14:01:35 +0100 | [diff] [blame] | 37 | void pbes2_encrypt(int params_tag, data_t *params_hex, data_t *pw, |
Waleed Elmelegy | 23ae416 | 2023-08-03 16:14:13 +0100 | [diff] [blame] | 38 | data_t *data, int outsize, int ref_ret, |
Waleed Elmelegy | dcad168 | 2023-08-29 14:55:03 +0100 | [diff] [blame] | 39 | data_t *ref_out) |
Waleed Elmelegy | 412629c | 2023-07-19 14:01:35 +0100 | [diff] [blame] | 40 | { |
| 41 | int my_ret; |
| 42 | mbedtls_asn1_buf params; |
| 43 | unsigned char *my_out = NULL; |
Waleed Elmelegy | b66cb65 | 2023-08-01 14:56:30 +0100 | [diff] [blame] | 44 | #if defined(MBEDTLS_CIPHER_PADDING_PKCS7) |
| 45 | size_t my_out_len = 0; |
| 46 | #endif |
| 47 | |
Waleed Elmelegy | 412629c | 2023-07-19 14:01:35 +0100 | [diff] [blame] | 48 | params.tag = params_tag; |
| 49 | params.p = params_hex->x; |
| 50 | params.len = params_hex->len; |
| 51 | |
Waleed Elmelegy | d2a03cb | 2023-08-30 13:12:09 +0100 | [diff] [blame] | 52 | ASSERT_ALLOC(my_out, outsize); |
Waleed Elmelegy | 412629c | 2023-07-19 14:01:35 +0100 | [diff] [blame] | 53 | |
Waleed Elmelegy | b66cb65 | 2023-08-01 14:56:30 +0100 | [diff] [blame] | 54 | if (ref_ret != MBEDTLS_ERR_ASN1_BUF_TOO_SMALL) { |
Waleed Elmelegy | dcad168 | 2023-08-29 14:55:03 +0100 | [diff] [blame] | 55 | my_ret = mbedtls_pkcs5_pbes2(¶ms, MBEDTLS_PKCS5_ENCRYPT, |
| 56 | pw->x, pw->len, data->x, data->len, my_out); |
Waleed Elmelegy | b66cb65 | 2023-08-01 14:56:30 +0100 | [diff] [blame] | 57 | TEST_EQUAL(my_ret, ref_ret); |
| 58 | } |
Waleed Elmelegy | 412629c | 2023-07-19 14:01:35 +0100 | [diff] [blame] | 59 | if (ref_ret == 0) { |
| 60 | ASSERT_COMPARE(my_out, ref_out->len, |
| 61 | ref_out->x, ref_out->len); |
| 62 | } |
| 63 | |
Waleed Elmelegy | b66cb65 | 2023-08-01 14:56:30 +0100 | [diff] [blame] | 64 | #if defined(MBEDTLS_CIPHER_PADDING_PKCS7) |
| 65 | my_ret = mbedtls_pkcs5_pbes2_ext(¶ms, MBEDTLS_PKCS5_ENCRYPT, |
| 66 | pw->x, pw->len, data->x, data->len, my_out, |
Waleed Elmelegy | 23ae416 | 2023-08-03 16:14:13 +0100 | [diff] [blame] | 67 | outsize, &my_out_len); |
Waleed Elmelegy | b66cb65 | 2023-08-01 14:56:30 +0100 | [diff] [blame] | 68 | TEST_EQUAL(my_ret, ref_ret); |
| 69 | if (ref_ret == 0) { |
Waleed Elmelegy | 23ae416 | 2023-08-03 16:14:13 +0100 | [diff] [blame] | 70 | ASSERT_COMPARE(my_out, my_out_len, |
Waleed Elmelegy | dcad168 | 2023-08-29 14:55:03 +0100 | [diff] [blame] | 71 | ref_out->x, ref_out->len); |
Waleed Elmelegy | b66cb65 | 2023-08-01 14:56:30 +0100 | [diff] [blame] | 72 | } |
| 73 | #endif |
| 74 | |
Waleed Elmelegy | 412629c | 2023-07-19 14:01:35 +0100 | [diff] [blame] | 75 | exit: |
| 76 | mbedtls_free(my_out); |
Waleed Elmelegy | 412629c | 2023-07-19 14:01:35 +0100 | [diff] [blame] | 77 | } |
| 78 | /* END_CASE */ |
| 79 | |
| 80 | /* BEGIN_CASE depends_on:MBEDTLS_ASN1_PARSE_C */ |
Waleed Elmelegy | dcad168 | 2023-08-29 14:55:03 +0100 | [diff] [blame] | 81 | void pbes2_decrypt(int params_tag, data_t *params_hex, data_t *pw, |
| 82 | data_t *data, int outsize, int ref_ret, |
| 83 | data_t *ref_out) |
Manuel Pégourié-Gonnard | 66aca93 | 2014-06-12 13:14:55 +0200 | [diff] [blame] | 84 | { |
| 85 | int my_ret; |
Manuel Pégourié-Gonnard | 2cf5a7c | 2015-04-08 12:49:31 +0200 | [diff] [blame] | 86 | mbedtls_asn1_buf params; |
Azim Khan | d30ca13 | 2017-06-09 04:32:58 +0100 | [diff] [blame] | 87 | unsigned char *my_out = NULL; |
Waleed Elmelegy | b66cb65 | 2023-08-01 14:56:30 +0100 | [diff] [blame] | 88 | #if defined(MBEDTLS_CIPHER_PADDING_PKCS7) |
| 89 | size_t my_out_len = 0; |
| 90 | #endif |
| 91 | |
Manuel Pégourié-Gonnard | 66aca93 | 2014-06-12 13:14:55 +0200 | [diff] [blame] | 92 | params.tag = params_tag; |
Azim Khan | d30ca13 | 2017-06-09 04:32:58 +0100 | [diff] [blame] | 93 | params.p = params_hex->x; |
| 94 | params.len = params_hex->len; |
Manuel Pégourié-Gonnard | 66aca93 | 2014-06-12 13:14:55 +0200 | [diff] [blame] | 95 | |
Waleed Elmelegy | d2a03cb | 2023-08-30 13:12:09 +0100 | [diff] [blame] | 96 | ASSERT_ALLOC(my_out, outsize); |
Manuel Pégourié-Gonnard | 66aca93 | 2014-06-12 13:14:55 +0200 | [diff] [blame] | 97 | |
Waleed Elmelegy | b66cb65 | 2023-08-01 14:56:30 +0100 | [diff] [blame] | 98 | if (ref_ret != MBEDTLS_ERR_ASN1_BUF_TOO_SMALL) { |
Waleed Elmelegy | dcad168 | 2023-08-29 14:55:03 +0100 | [diff] [blame] | 99 | my_ret = mbedtls_pkcs5_pbes2(¶ms, MBEDTLS_PKCS5_DECRYPT, |
| 100 | pw->x, pw->len, data->x, data->len, my_out); |
Waleed Elmelegy | b66cb65 | 2023-08-01 14:56:30 +0100 | [diff] [blame] | 101 | TEST_EQUAL(my_ret, ref_ret); |
| 102 | } |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 103 | if (ref_ret == 0) { |
Waleed Elmelegy | 412629c | 2023-07-19 14:01:35 +0100 | [diff] [blame] | 104 | ASSERT_COMPARE(my_out, ref_out->len, |
| 105 | ref_out->x, ref_out->len); |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 106 | } |
Waleed Elmelegy | b66cb65 | 2023-08-01 14:56:30 +0100 | [diff] [blame] | 107 | #if defined(MBEDTLS_CIPHER_PADDING_PKCS7) |
| 108 | my_ret = mbedtls_pkcs5_pbes2_ext(¶ms, MBEDTLS_PKCS5_DECRYPT, |
| 109 | pw->x, pw->len, data->x, data->len, my_out, |
Waleed Elmelegy | 23ae416 | 2023-08-03 16:14:13 +0100 | [diff] [blame] | 110 | outsize, &my_out_len); |
Waleed Elmelegy | b66cb65 | 2023-08-01 14:56:30 +0100 | [diff] [blame] | 111 | TEST_EQUAL(my_ret, ref_ret); |
| 112 | if (ref_ret == 0) { |
Waleed Elmelegy | 23ae416 | 2023-08-03 16:14:13 +0100 | [diff] [blame] | 113 | ASSERT_COMPARE(my_out, my_out_len, |
Waleed Elmelegy | dcad168 | 2023-08-29 14:55:03 +0100 | [diff] [blame] | 114 | ref_out->x, ref_out->len); |
Waleed Elmelegy | b66cb65 | 2023-08-01 14:56:30 +0100 | [diff] [blame] | 115 | } |
| 116 | #endif |
Manuel Pégourié-Gonnard | 66aca93 | 2014-06-12 13:14:55 +0200 | [diff] [blame] | 117 | |
Paul Bakker | bd51b26 | 2014-07-10 15:26:12 +0200 | [diff] [blame] | 118 | exit: |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 119 | mbedtls_free(my_out); |
Manuel Pégourié-Gonnard | 66aca93 | 2014-06-12 13:14:55 +0200 | [diff] [blame] | 120 | } |
| 121 | /* END_CASE */ |
Paul Bakker | 81c6091 | 2016-07-19 14:54:57 +0100 | [diff] [blame] | 122 | |
| 123 | /* BEGIN_CASE depends_on:MBEDTLS_SELF_TEST */ |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 124 | void pkcs5_selftest() |
Paul Bakker | 81c6091 | 2016-07-19 14:54:57 +0100 | [diff] [blame] | 125 | { |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 126 | TEST_ASSERT(mbedtls_pkcs5_self_test(1) == 0); |
Paul Bakker | 81c6091 | 2016-07-19 14:54:57 +0100 | [diff] [blame] | 127 | } |
| 128 | /* END_CASE */ |