blob: 7ce88cbc5e59cab7c3f7c137416302899a14885e [file] [log] [blame]
Paul Bakker33b43f12013-08-20 11:48:36 +02001/* BEGIN_HEADER */
Manuel Pégourié-Gonnard7f809972015-03-09 17:05:11 +00002#include "mbedtls/pkcs5.h"
Waleed Elmelegya2307352023-07-21 16:20:17 +01003#include "mbedtls/cipher.h"
Paul Bakker33b43f12013-08-20 11:48:36 +02004/* END_HEADER */
Paul Bakkerb0c19a42013-06-24 19:26:38 +02005
Paul Bakker33b43f12013-08-20 11:48:36 +02006/* BEGIN_DEPENDENCIES
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02007 * depends_on:MBEDTLS_PKCS5_C
Paul Bakker33b43f12013-08-20 11:48:36 +02008 * END_DEPENDENCIES
9 */
Paul Bakkerb0c19a42013-06-24 19:26:38 +020010
Paul Bakker33b43f12013-08-20 11:48:36 +020011/* BEGIN_CASE */
Gilles Peskine1b6c09a2023-01-11 14:52:35 +010012void 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 Bakkerb0c19a42013-06-24 19:26:38 +020014{
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +020015 mbedtls_md_context_t ctx;
16 const mbedtls_md_info_t *info;
Paul Bakkerb0c19a42013-06-24 19:26:38 +020017
Paul Bakkerb0c19a42013-06-24 19:26:38 +020018 unsigned char key[100];
19
Gilles Peskine1b6c09a2023-01-11 14:52:35 +010020 mbedtls_md_init(&ctx);
Paul Bakkerd2a2d612014-07-01 15:45:49 +020021
Gilles Peskine1b6c09a2023-01-11 14:52:35 +010022 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 Bakkerb0c19a42013-06-24 19:26:38 +020027
Gilles Peskine1b6c09a2023-01-11 14:52:35 +010028 TEST_ASSERT(mbedtls_test_hexcmp(key, result_key_string->x,
29 key_len, result_key_string->len) == 0);
Paul Bakkerbd51b262014-07-10 15:26:12 +020030
31exit:
Gilles Peskine1b6c09a2023-01-11 14:52:35 +010032 mbedtls_md_free(&ctx);
Paul Bakkerb0c19a42013-06-24 19:26:38 +020033}
Paul Bakker33b43f12013-08-20 11:48:36 +020034/* END_CASE */
Manuel Pégourié-Gonnard66aca932014-06-12 13:14:55 +020035
Andres Amaya Garcia576d4742018-03-27 20:53:56 +010036/* BEGIN_CASE depends_on:MBEDTLS_ASN1_PARSE_C */
Waleed Elmelegy412629c2023-07-19 14:01:35 +010037void pbes2_encrypt(int params_tag, data_t *params_hex, data_t *pw,
Waleed Elmelegy23ae4162023-08-03 16:14:13 +010038 data_t *data, int outsize, int ref_ret,
Waleed Elmelegydcad1682023-08-29 14:55:03 +010039 data_t *ref_out)
Waleed Elmelegy412629c2023-07-19 14:01:35 +010040{
41 int my_ret;
42 mbedtls_asn1_buf params;
43 unsigned char *my_out = NULL;
Waleed Elmelegyb66cb652023-08-01 14:56:30 +010044#if defined(MBEDTLS_CIPHER_PADDING_PKCS7)
45 size_t my_out_len = 0;
46#endif
47
Waleed Elmelegy412629c2023-07-19 14:01:35 +010048 params.tag = params_tag;
49 params.p = params_hex->x;
50 params.len = params_hex->len;
51
Waleed Elmelegyd2a03cb2023-08-30 13:12:09 +010052 ASSERT_ALLOC(my_out, outsize);
Waleed Elmelegy412629c2023-07-19 14:01:35 +010053
Waleed Elmelegyb66cb652023-08-01 14:56:30 +010054 if (ref_ret != MBEDTLS_ERR_ASN1_BUF_TOO_SMALL) {
Waleed Elmelegydcad1682023-08-29 14:55:03 +010055 my_ret = mbedtls_pkcs5_pbes2(&params, MBEDTLS_PKCS5_ENCRYPT,
56 pw->x, pw->len, data->x, data->len, my_out);
Waleed Elmelegyb66cb652023-08-01 14:56:30 +010057 TEST_EQUAL(my_ret, ref_ret);
58 }
Waleed Elmelegy412629c2023-07-19 14:01:35 +010059 if (ref_ret == 0) {
60 ASSERT_COMPARE(my_out, ref_out->len,
61 ref_out->x, ref_out->len);
62 }
63
Waleed Elmelegyb66cb652023-08-01 14:56:30 +010064#if defined(MBEDTLS_CIPHER_PADDING_PKCS7)
65 my_ret = mbedtls_pkcs5_pbes2_ext(&params, MBEDTLS_PKCS5_ENCRYPT,
66 pw->x, pw->len, data->x, data->len, my_out,
Waleed Elmelegy23ae4162023-08-03 16:14:13 +010067 outsize, &my_out_len);
Waleed Elmelegyb66cb652023-08-01 14:56:30 +010068 TEST_EQUAL(my_ret, ref_ret);
69 if (ref_ret == 0) {
Waleed Elmelegy23ae4162023-08-03 16:14:13 +010070 ASSERT_COMPARE(my_out, my_out_len,
Waleed Elmelegydcad1682023-08-29 14:55:03 +010071 ref_out->x, ref_out->len);
Waleed Elmelegyb66cb652023-08-01 14:56:30 +010072 }
73#endif
74
Waleed Elmelegy412629c2023-07-19 14:01:35 +010075exit:
76 mbedtls_free(my_out);
Waleed Elmelegy412629c2023-07-19 14:01:35 +010077}
78/* END_CASE */
79
80/* BEGIN_CASE depends_on:MBEDTLS_ASN1_PARSE_C */
Waleed Elmelegydcad1682023-08-29 14:55:03 +010081void 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é-Gonnard66aca932014-06-12 13:14:55 +020084{
85 int my_ret;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +020086 mbedtls_asn1_buf params;
Azim Khand30ca132017-06-09 04:32:58 +010087 unsigned char *my_out = NULL;
Waleed Elmelegyb66cb652023-08-01 14:56:30 +010088#if defined(MBEDTLS_CIPHER_PADDING_PKCS7)
89 size_t my_out_len = 0;
90#endif
91
Manuel Pégourié-Gonnard66aca932014-06-12 13:14:55 +020092 params.tag = params_tag;
Azim Khand30ca132017-06-09 04:32:58 +010093 params.p = params_hex->x;
94 params.len = params_hex->len;
Manuel Pégourié-Gonnard66aca932014-06-12 13:14:55 +020095
Waleed Elmelegyd2a03cb2023-08-30 13:12:09 +010096 ASSERT_ALLOC(my_out, outsize);
Manuel Pégourié-Gonnard66aca932014-06-12 13:14:55 +020097
Waleed Elmelegyb66cb652023-08-01 14:56:30 +010098 if (ref_ret != MBEDTLS_ERR_ASN1_BUF_TOO_SMALL) {
Waleed Elmelegydcad1682023-08-29 14:55:03 +010099 my_ret = mbedtls_pkcs5_pbes2(&params, MBEDTLS_PKCS5_DECRYPT,
100 pw->x, pw->len, data->x, data->len, my_out);
Waleed Elmelegyb66cb652023-08-01 14:56:30 +0100101 TEST_EQUAL(my_ret, ref_ret);
102 }
Gilles Peskine1b6c09a2023-01-11 14:52:35 +0100103 if (ref_ret == 0) {
Waleed Elmelegy412629c2023-07-19 14:01:35 +0100104 ASSERT_COMPARE(my_out, ref_out->len,
105 ref_out->x, ref_out->len);
Gilles Peskine1b6c09a2023-01-11 14:52:35 +0100106 }
Waleed Elmelegyb66cb652023-08-01 14:56:30 +0100107#if defined(MBEDTLS_CIPHER_PADDING_PKCS7)
108 my_ret = mbedtls_pkcs5_pbes2_ext(&params, MBEDTLS_PKCS5_DECRYPT,
109 pw->x, pw->len, data->x, data->len, my_out,
Waleed Elmelegy23ae4162023-08-03 16:14:13 +0100110 outsize, &my_out_len);
Waleed Elmelegyb66cb652023-08-01 14:56:30 +0100111 TEST_EQUAL(my_ret, ref_ret);
112 if (ref_ret == 0) {
Waleed Elmelegy23ae4162023-08-03 16:14:13 +0100113 ASSERT_COMPARE(my_out, my_out_len,
Waleed Elmelegydcad1682023-08-29 14:55:03 +0100114 ref_out->x, ref_out->len);
Waleed Elmelegyb66cb652023-08-01 14:56:30 +0100115 }
116#endif
Manuel Pégourié-Gonnard66aca932014-06-12 13:14:55 +0200117
Paul Bakkerbd51b262014-07-10 15:26:12 +0200118exit:
Gilles Peskine1b6c09a2023-01-11 14:52:35 +0100119 mbedtls_free(my_out);
Manuel Pégourié-Gonnard66aca932014-06-12 13:14:55 +0200120}
121/* END_CASE */
Paul Bakker81c60912016-07-19 14:54:57 +0100122
123/* BEGIN_CASE depends_on:MBEDTLS_SELF_TEST */
Gilles Peskine1b6c09a2023-01-11 14:52:35 +0100124void pkcs5_selftest()
Paul Bakker81c60912016-07-19 14:54:57 +0100125{
Gilles Peskine1b6c09a2023-01-11 14:52:35 +0100126 TEST_ASSERT(mbedtls_pkcs5_self_test(1) == 0);
Paul Bakker81c60912016-07-19 14:54:57 +0100127}
128/* END_CASE */