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