/* 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 */