40 lines
1.2 KiB
Text
40 lines
1.2 KiB
Text
|
/* BEGIN_HEADER */
|
||
|
|
#include "mbedtls/hkdf.h"
|
||
|
|
/* END_HEADER */
|
||
|
|
|
||
|
|
/* BEGIN_DEPENDENCIES
|
||
|
|
* depends_on:MBEDTLS_HKDF_C
|
||
|
|
* END_DEPENDENCIES
|
||
|
|
*/
|
||
|
|
|
||
|
|
/* BEGIN_CASE */
|
||
|
|
void test_hkdf( int md_alg, char *hex_ikm_string, char *hex_salt_string,
|
||
|
|
char *hex_info_string, char *hex_okm_string )
|
||
|
|
{
|
||
|
|
int ret;
|
||
|
|
size_t ikm_len, salt_len, info_len, okm_len;
|
||
|
|
unsigned char ikm[1024] = { '\0' };
|
||
|
|
unsigned char salt[1024] = { '\0' };
|
||
|
|
unsigned char info[1024] = { '\0' };
|
||
|
|
unsigned char expected_okm[1024] = { '\0' };
|
||
|
|
unsigned char okm[1024] = { '\0' };
|
||
|
|
unsigned char okm_string[1000] = { '\0' };
|
||
|
|
|
||
|
|
const mbedtls_md_info_t *md = mbedtls_md_info_from_type( md_alg );
|
||
|
|
TEST_ASSERT( md != NULL );
|
||
|
|
|
||
|
|
ikm_len = unhexify( ikm, hex_ikm_string );
|
||
|
|
salt_len = unhexify( salt, hex_salt_string );
|
||
|
|
info_len = unhexify( info, hex_info_string );
|
||
|
|
okm_len = unhexify( expected_okm, hex_okm_string );
|
||
|
|
|
||
|
|
ret = mbedtls_hkdf( md, salt, salt_len, ikm, ikm_len, info, info_len, okm,
|
||
|
|
okm_len);
|
||
|
|
TEST_ASSERT( ret == 0 );
|
||
|
|
|
||
|
|
// Run hexify on it so that it looks nicer if the assertion fails
|
||
|
|
hexify( okm_string, okm, okm_len );
|
||
|
|
TEST_ASSERT( !strcmp( (char *)okm_string, hex_okm_string ) );
|
||
|
|
}
|
||
|
|
/* END_CASE */
|