blob: 05ce1db8519da03ab650b10575ae132592a7b326 [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 Elmelegyf3fafc32023-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 Peskine449bd832023-01-11 14:50:10 +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{
Paul Bakkerb0c19a42013-06-24 19:26:38 +020015 unsigned char key[100];
16
Manuel Pégourié-Gonnard49e67f82023-03-16 11:39:20 +010017 MD_PSA_INIT();
Gilles Peskine449bd832023-01-11 14:50:10 +010018 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 Bakkerb0c19a42013-06-24 19:26:38 +020021
Gilles Peskine449bd832023-01-11 14:50:10 +010022 TEST_ASSERT(mbedtls_test_hexcmp(key, result_key_string->x,
23 key_len, result_key_string->len) == 0);
Manuel Pégourié-Gonnard49e67f82023-03-16 11:39:20 +010024
25exit:
26 MD_PSA_DONE();
Paul Bakkerb0c19a42013-06-24 19:26:38 +020027}
Paul Bakker33b43f12013-08-20 11:48:36 +020028/* END_CASE */
Manuel Pégourié-Gonnard66aca932014-06-12 13:14:55 +020029
Andres Amaya Garcia576d4742018-03-27 20:53:56 +010030/* BEGIN_CASE depends_on:MBEDTLS_ASN1_PARSE_C */
Waleed Elmelegy708d78f2023-07-19 14:01:35 +010031void pbes2_encrypt(int params_tag, data_t *params_hex, data_t *pw,
Waleed Elmelegy87bc1e12023-08-03 16:14:13 +010032 data_t *data, int outsize, int ref_ret,
Waleed Elmelegy5d3f3152023-08-01 14:56:30 +010033 data_t *ref_out, int ref_out_len)
Waleed Elmelegy708d78f2023-07-19 14:01:35 +010034{
35 int my_ret;
36 mbedtls_asn1_buf params;
37 unsigned char *my_out = NULL;
Waleed Elmelegy5d3f3152023-08-01 14:56:30 +010038#if defined(MBEDTLS_CIPHER_PADDING_PKCS7)
39 size_t my_out_len = 0;
40#endif
Waleed Elmelegy708d78f2023-07-19 14:01:35 +010041 MD_PSA_INIT();
Waleed Elmelegy5d3f3152023-08-01 14:56:30 +010042 /* This parameter is only used when MBEDTLS_CIPHER_PADDING_PKCS7 is enabled */
43 (void) ref_out_len;
Waleed Elmelegy708d78f2023-07-19 14:01:35 +010044
45 params.tag = params_tag;
46 params.p = params_hex->x;
47 params.len = params_hex->len;
48
Waleed Elmelegy87bc1e12023-08-03 16:14:13 +010049 ASSERT_ALLOC(my_out, ref_out->len);
Waleed Elmelegy708d78f2023-07-19 14:01:35 +010050
51 my_ret = mbedtls_pkcs5_pbes2(&params, MBEDTLS_PKCS5_ENCRYPT,
52 pw->x, pw->len, data->x, data->len, my_out);
Waleed Elmelegy5d3f3152023-08-01 14:56:30 +010053 if (ref_ret != MBEDTLS_ERR_ASN1_BUF_TOO_SMALL) {
54 TEST_EQUAL(my_ret, ref_ret);
55 }
Waleed Elmelegy708d78f2023-07-19 14:01:35 +010056 if (ref_ret == 0) {
57 ASSERT_COMPARE(my_out, ref_out->len,
58 ref_out->x, ref_out->len);
59 }
60
Waleed Elmelegy5d3f3152023-08-01 14:56:30 +010061#if defined(MBEDTLS_CIPHER_PADDING_PKCS7)
62 my_ret = mbedtls_pkcs5_pbes2_ext(&params, MBEDTLS_PKCS5_ENCRYPT,
63 pw->x, pw->len, data->x, data->len, my_out,
Waleed Elmelegy87bc1e12023-08-03 16:14:13 +010064 outsize, &my_out_len);
Waleed Elmelegy5d3f3152023-08-01 14:56:30 +010065 TEST_EQUAL(my_ret, ref_ret);
66 if (ref_ret == 0) {
Waleed Elmelegy87bc1e12023-08-03 16:14:13 +010067 ASSERT_COMPARE(my_out, my_out_len,
68 ref_out->x, ref_out_len);
Waleed Elmelegy5d3f3152023-08-01 14:56:30 +010069 }
70#endif
71
Waleed Elmelegy708d78f2023-07-19 14:01:35 +010072exit:
73 mbedtls_free(my_out);
74 MD_PSA_DONE();
75}
76/* END_CASE */
77
78/* BEGIN_CASE depends_on:MBEDTLS_ASN1_PARSE_C */
Gilles Peskine449bd832023-01-11 14:50:10 +010079void mbedtls_pkcs5_pbes2(int params_tag, data_t *params_hex, data_t *pw,
Waleed Elmelegy87bc1e12023-08-03 16:14:13 +010080 data_t *data, int outsize, int ref_ret,
Waleed Elmelegy5d3f3152023-08-01 14:56:30 +010081 data_t *ref_out, int ref_out_len)
Manuel Pégourié-Gonnard66aca932014-06-12 13:14:55 +020082{
83 int my_ret;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +020084 mbedtls_asn1_buf params;
Azim Khand30ca132017-06-09 04:32:58 +010085 unsigned char *my_out = NULL;
Waleed Elmelegy5d3f3152023-08-01 14:56:30 +010086#if defined(MBEDTLS_CIPHER_PADDING_PKCS7)
87 size_t my_out_len = 0;
88#endif
Manuel Pégourié-Gonnard66aca932014-06-12 13:14:55 +020089
Manuel Pégourié-Gonnard49e67f82023-03-16 11:39:20 +010090 MD_PSA_INIT();
Waleed Elmelegy5d3f3152023-08-01 14:56:30 +010091 /* This parameter is only used when MBEDTLS_CIPHER_PADDING_PKCS7 is enabled */
92 (void) ref_out_len;
93
Andrzej Kurek26909f32022-08-31 15:10:50 -040094
Manuel Pégourié-Gonnard66aca932014-06-12 13:14:55 +020095 params.tag = params_tag;
Azim Khand30ca132017-06-09 04:32:58 +010096 params.p = params_hex->x;
97 params.len = params_hex->len;
Manuel Pégourié-Gonnard66aca932014-06-12 13:14:55 +020098
Waleed Elmelegy87bc1e12023-08-03 16:14:13 +010099 ASSERT_ALLOC(my_out, ref_out->len);
Manuel Pégourié-Gonnard66aca932014-06-12 13:14:55 +0200100
Gilles Peskine449bd832023-01-11 14:50:10 +0100101 my_ret = mbedtls_pkcs5_pbes2(&params, MBEDTLS_PKCS5_DECRYPT,
102 pw->x, pw->len, data->x, data->len, my_out);
Waleed Elmelegy5d3f3152023-08-01 14:56:30 +0100103 if (ref_ret != MBEDTLS_ERR_ASN1_BUF_TOO_SMALL) {
104 TEST_EQUAL(my_ret, ref_ret);
105 }
Gilles Peskine449bd832023-01-11 14:50:10 +0100106 if (ref_ret == 0) {
Waleed Elmelegy708d78f2023-07-19 14:01:35 +0100107 ASSERT_COMPARE(my_out, ref_out->len,
108 ref_out->x, ref_out->len);
Gilles Peskine449bd832023-01-11 14:50:10 +0100109 }
Waleed Elmelegy5d3f3152023-08-01 14:56:30 +0100110#if defined(MBEDTLS_CIPHER_PADDING_PKCS7)
111 my_ret = mbedtls_pkcs5_pbes2_ext(&params, MBEDTLS_PKCS5_DECRYPT,
112 pw->x, pw->len, data->x, data->len, my_out,
Waleed Elmelegy87bc1e12023-08-03 16:14:13 +0100113 outsize, &my_out_len);
Waleed Elmelegy5d3f3152023-08-01 14:56:30 +0100114 TEST_EQUAL(my_ret, ref_ret);
115 if (ref_ret == 0) {
Waleed Elmelegy87bc1e12023-08-03 16:14:13 +0100116 ASSERT_COMPARE(my_out, my_out_len,
117 ref_out->x, ref_out_len);
Waleed Elmelegy5d3f3152023-08-01 14:56:30 +0100118 }
119#endif
Manuel Pégourié-Gonnard66aca932014-06-12 13:14:55 +0200120
Paul Bakkerbd51b262014-07-10 15:26:12 +0200121exit:
Gilles Peskine449bd832023-01-11 14:50:10 +0100122 mbedtls_free(my_out);
Manuel Pégourié-Gonnard49e67f82023-03-16 11:39:20 +0100123 MD_PSA_DONE();
Manuel Pégourié-Gonnard66aca932014-06-12 13:14:55 +0200124}
125/* END_CASE */
Paul Bakker81c60912016-07-19 14:54:57 +0100126
127/* BEGIN_CASE depends_on:MBEDTLS_SELF_TEST */
Gilles Peskine449bd832023-01-11 14:50:10 +0100128void pkcs5_selftest()
Paul Bakker81c60912016-07-19 14:54:57 +0100129{
Manuel Pégourié-Gonnard49e67f82023-03-16 11:39:20 +0100130 MD_PSA_INIT();
Gilles Peskine449bd832023-01-11 14:50:10 +0100131 TEST_ASSERT(mbedtls_pkcs5_self_test(1) == 0);
Manuel Pégourié-Gonnard49e67f82023-03-16 11:39:20 +0100132
133exit:
134 MD_PSA_DONE();
Paul Bakker81c60912016-07-19 14:54:57 +0100135}
136/* END_CASE */