blob: 200bdc37f7fa0209b771d69a2bb9dd715d1a159b [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 Elmelegy79b6e262023-08-29 14:55:03 +010033 data_t *ref_out)
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 Elmelegy4a0a9892023-08-04 11:39:01 +010042 /* Those parameters are only used when MBEDTLS_CIPHER_PADDING_PKCS7 is enabled */
Waleed Elmelegy4a0a9892023-08-04 11:39:01 +010043 (void) outsize;
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
Waleed Elmelegy5d3f3152023-08-01 14:56:30 +010051 if (ref_ret != MBEDTLS_ERR_ASN1_BUF_TOO_SMALL) {
Waleed Elmelegy79b6e262023-08-29 14:55:03 +010052 my_ret = mbedtls_pkcs5_pbes2(&params, MBEDTLS_PKCS5_ENCRYPT,
53 pw->x, pw->len, data->x, data->len, my_out);
Waleed Elmelegy5d3f3152023-08-01 14:56:30 +010054 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,
Waleed Elmelegy79b6e262023-08-29 14:55:03 +010068 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 */
Waleed Elmelegy79b6e262023-08-29 14:55:03 +010079void pbes2_decrypt(int params_tag, data_t *params_hex, data_t *pw,
80 data_t *data, int outsize, int ref_ret,
81 data_t *ref_out)
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 Elmelegy4a0a9892023-08-04 11:39:01 +010091 /* Those parameters are only used when MBEDTLS_CIPHER_PADDING_PKCS7 is enabled */
Waleed Elmelegy4a0a9892023-08-04 11:39:01 +010092 (void) outsize;
Waleed Elmelegy5d3f3152023-08-01 14:56:30 +010093
Manuel Pégourié-Gonnard66aca932014-06-12 13:14:55 +020094 params.tag = params_tag;
Azim Khand30ca132017-06-09 04:32:58 +010095 params.p = params_hex->x;
96 params.len = params_hex->len;
Manuel Pégourié-Gonnard66aca932014-06-12 13:14:55 +020097
Waleed Elmelegy87bc1e12023-08-03 16:14:13 +010098 ASSERT_ALLOC(my_out, ref_out->len);
Manuel Pégourié-Gonnard66aca932014-06-12 13:14:55 +020099
Waleed Elmelegy5d3f3152023-08-01 14:56:30 +0100100 if (ref_ret != MBEDTLS_ERR_ASN1_BUF_TOO_SMALL) {
Waleed Elmelegy79b6e262023-08-29 14:55:03 +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 TEST_EQUAL(my_ret, ref_ret);
104 }
Gilles Peskine449bd832023-01-11 14:50:10 +0100105 if (ref_ret == 0) {
Waleed Elmelegy708d78f2023-07-19 14:01:35 +0100106 ASSERT_COMPARE(my_out, ref_out->len,
107 ref_out->x, ref_out->len);
Gilles Peskine449bd832023-01-11 14:50:10 +0100108 }
Waleed Elmelegy5d3f3152023-08-01 14:56:30 +0100109#if defined(MBEDTLS_CIPHER_PADDING_PKCS7)
110 my_ret = mbedtls_pkcs5_pbes2_ext(&params, MBEDTLS_PKCS5_DECRYPT,
111 pw->x, pw->len, data->x, data->len, my_out,
Waleed Elmelegy87bc1e12023-08-03 16:14:13 +0100112 outsize, &my_out_len);
Waleed Elmelegy5d3f3152023-08-01 14:56:30 +0100113 TEST_EQUAL(my_ret, ref_ret);
114 if (ref_ret == 0) {
Waleed Elmelegy87bc1e12023-08-03 16:14:13 +0100115 ASSERT_COMPARE(my_out, my_out_len,
Waleed Elmelegy79b6e262023-08-29 14:55:03 +0100116 ref_out->x, ref_out->len);
Waleed Elmelegy5d3f3152023-08-01 14:56:30 +0100117 }
118#endif
Manuel Pégourié-Gonnard66aca932014-06-12 13:14:55 +0200119
Paul Bakkerbd51b262014-07-10 15:26:12 +0200120exit:
Gilles Peskine449bd832023-01-11 14:50:10 +0100121 mbedtls_free(my_out);
Manuel Pégourié-Gonnard49e67f82023-03-16 11:39:20 +0100122 MD_PSA_DONE();
Manuel Pégourié-Gonnard66aca932014-06-12 13:14:55 +0200123}
124/* END_CASE */
Paul Bakker81c60912016-07-19 14:54:57 +0100125
126/* BEGIN_CASE depends_on:MBEDTLS_SELF_TEST */
Gilles Peskine449bd832023-01-11 14:50:10 +0100127void pkcs5_selftest()
Paul Bakker81c60912016-07-19 14:54:57 +0100128{
Manuel Pégourié-Gonnard49e67f82023-03-16 11:39:20 +0100129 MD_PSA_INIT();
Gilles Peskine449bd832023-01-11 14:50:10 +0100130 TEST_ASSERT(mbedtls_pkcs5_self_test(1) == 0);
Manuel Pégourié-Gonnard49e67f82023-03-16 11:39:20 +0100131
132exit:
133 MD_PSA_DONE();
Paul Bakker81c60912016-07-19 14:54:57 +0100134}
135/* END_CASE */