/* BEGIN_HEADER */
#include "mbedtls/cipher.h"
#include "mbedtls/cmac.h"
/* END_HEADER */

/* BEGIN_DEPENDENCIES
 * depends_on:MBEDTLS_CMAC_C
 * END_DEPENDENCIES
 */

/* BEGIN_CASE depends_on:MBEDTLS_SELF_TEST */
void mbedtls_cmac_self_test(  )
{
    TEST_ASSERT( mbedtls_cmac_self_test( 1 ) == 0 );
}
/* END_CASE */

/* BEGIN_CASE */
void mbedtls_cmac_null_args(  )
{
    mbedtls_cipher_context_t ctx;
    const mbedtls_cipher_info_t *cipher_info;
    unsigned char test_key[MBEDTLS_CIPHER_BLKSIZE_MAX];
    unsigned char test_data[MBEDTLS_CIPHER_BLKSIZE_MAX];
    unsigned char test_output[MBEDTLS_CIPHER_BLKSIZE_MAX];

    mbedtls_cipher_init( &ctx );

    /* Test NULL cipher info */
    TEST_ASSERT( mbedtls_cipher_cmac_update( &ctx, test_data, 16 ) ==
                                         MBEDTLS_ERR_CIPHER_BAD_INPUT_DATA );

    cipher_info = mbedtls_cipher_info_from_type( MBEDTLS_CIPHER_AES_128_ECB );
    TEST_ASSERT( mbedtls_cipher_setup( &ctx, cipher_info ) == 0 );

    TEST_ASSERT( mbedtls_cipher_cmac_starts( NULL, test_key, 128 ) ==
                                         MBEDTLS_ERR_CIPHER_BAD_INPUT_DATA );

    TEST_ASSERT( mbedtls_cipher_cmac_starts( &ctx, NULL, 128 ) ==
                                         MBEDTLS_ERR_CIPHER_BAD_INPUT_DATA );

    TEST_ASSERT( mbedtls_cipher_cmac_update( NULL, test_data, 16 ) ==
                                         MBEDTLS_ERR_CIPHER_BAD_INPUT_DATA );

    TEST_ASSERT( mbedtls_cipher_cmac_update( &ctx, NULL, 16 ) ==
                                         MBEDTLS_ERR_CIPHER_BAD_INPUT_DATA );

    TEST_ASSERT( mbedtls_cipher_cmac_finish( NULL, test_output ) ==
                                         MBEDTLS_ERR_CIPHER_BAD_INPUT_DATA );

    TEST_ASSERT( mbedtls_cipher_cmac_finish( &ctx, NULL ) ==
                                         MBEDTLS_ERR_CIPHER_BAD_INPUT_DATA );

    TEST_ASSERT( mbedtls_cipher_cmac_reset( NULL ) ==
                                         MBEDTLS_ERR_CIPHER_BAD_INPUT_DATA );

    TEST_ASSERT( mbedtls_cipher_cmac( NULL,
                                      test_key, 128,
                                      test_data, 16,
                                      test_output ) ==
                                            MBEDTLS_ERR_CIPHER_BAD_INPUT_DATA );

    TEST_ASSERT( mbedtls_cipher_cmac( cipher_info,
                                      NULL, 128,
                                      test_data, 16,
                                      test_output ) ==
                                            MBEDTLS_ERR_CIPHER_BAD_INPUT_DATA );

    TEST_ASSERT( mbedtls_cipher_cmac( cipher_info,
                                      test_key, 128,
                                      NULL, 16,
                                      test_output ) ==
                                            MBEDTLS_ERR_CIPHER_BAD_INPUT_DATA );

    TEST_ASSERT( mbedtls_cipher_cmac( cipher_info,
                                      test_key, 128,
                                      test_data, 16,
                                      NULL ) ==
                                            MBEDTLS_ERR_CIPHER_BAD_INPUT_DATA );

    TEST_ASSERT( mbedtls_aes_cmac_prf_128( NULL, 16,
                                           test_data, 16,
                                           test_output ) ==
                                           MBEDTLS_ERR_CIPHER_BAD_INPUT_DATA );

    TEST_ASSERT( mbedtls_aes_cmac_prf_128( test_key, 16,
                                           NULL, 16,
                                           test_output ) ==
                                              MBEDTLS_ERR_CIPHER_BAD_INPUT_DATA );

    TEST_ASSERT( mbedtls_aes_cmac_prf_128( test_key, 16,
                                           test_data, 16,
                                           NULL ) ==
                                              MBEDTLS_ERR_CIPHER_BAD_INPUT_DATA );

exit:
    mbedtls_cipher_free( &ctx );
}
/* END_CASE */

/* BEGIN_CASE */
void mbedtls_cmac_setkey( int cipher_type, int key_size, int result )
{
    const mbedtls_cipher_info_t *cipher_info;
    unsigned char key[32];
    unsigned char buf[16];
    unsigned char tmp[16];

    memset( key, 0x2A, sizeof( key ) );
    TEST_ASSERT( (unsigned) key_size <= 8 * sizeof( key ) );

    TEST_ASSERT( ( cipher_info = mbedtls_cipher_info_from_type( cipher_type ) )
                    != NULL );

    memset( buf, 0x2A, sizeof( buf ) );
    TEST_ASSERT( ( result == mbedtls_cipher_cmac( cipher_info, key, key_size,
                                                buf, 16, tmp ) ) != 0 );
}
/* END_CASE */

/* BEGIN_CASE */
void mbedtls_cmac_multiple_blocks( int cipher_type, data_t * key,
                                   int keybits, int block_size,
                                   data_t * block1, int block1_len,
                                   data_t * block2, int block2_len,
                                   data_t * block3, int block3_len,
                                   data_t * block4, int block4_len,
                                   data_t * expected_result )
{
    const mbedtls_cipher_info_t *cipher_info;
    mbedtls_cipher_context_t ctx;
    unsigned char output[MBEDTLS_CIPHER_BLKSIZE_MAX];

    /* Convert the test parameters to binary data */

    mbedtls_cipher_init( &ctx );

    /* Validate the test inputs */
    TEST_ASSERT( block1_len <= 100 );
    TEST_ASSERT( block2_len <= 100 );
    TEST_ASSERT( block3_len <= 100 );
    TEST_ASSERT( block4_len <= 100 );

    /* Set up */
    TEST_ASSERT( ( cipher_info = mbedtls_cipher_info_from_type( cipher_type ) )
                    != NULL );

    TEST_ASSERT( mbedtls_cipher_setup( &ctx, cipher_info ) == 0 );

    TEST_ASSERT( mbedtls_cipher_cmac_starts( &ctx,
                                             (const unsigned char*)key->x,
                                             keybits ) == 0 );

    /* Multiple partial and complete blocks. A negative length means skip the
     * update operation */
    if( block1_len >= 0)
        TEST_ASSERT( mbedtls_cipher_cmac_update( &ctx,
                                                 (unsigned char*)block1->x,
                                                 block1_len ) == 0);

    if( block2_len >= 0 )
        TEST_ASSERT( mbedtls_cipher_cmac_update( &ctx,
                                                 (unsigned char*)block2->x,
                                                 block2_len ) == 0);

    if( block3_len >= 0 )
        TEST_ASSERT( mbedtls_cipher_cmac_update( &ctx,
                                                 (unsigned char*)block3->x,
                                                 block3_len ) == 0);

    if( block4_len >= 0 )
        TEST_ASSERT( mbedtls_cipher_cmac_update( &ctx,
                                                 (unsigned char*)block4->x,
                                                 block4_len ) == 0);

    TEST_ASSERT( mbedtls_cipher_cmac_finish( &ctx, output ) == 0 );

    TEST_ASSERT( memcmp( output, expected_result->x, block_size )  == 0 );

exit:
    mbedtls_cipher_free( &ctx );
}
/* END_CASE */

/* BEGIN_CASE */
void mbedtls_cmac_multiple_operations_same_key( int cipher_type,
                                                data_t * key, int keybits,
                                                int block_size,
                                                data_t * block_a1,
                                                int block_a1_len,
                                                data_t * block_a2,
                                                int block_a2_len,
                                                data_t * block_a3,
                                                int block_a3_len,
                                                data_t * expected_result_a,
                                                data_t * block_b1,
                                                int block_b1_len,
                                                data_t * block_b2,
                                                int block_b2_len,
                                                data_t * block_b3,
                                                int block_b3_len,
                                                data_t * expected_result_b
                                                )
{
    const mbedtls_cipher_info_t *cipher_info;
    mbedtls_cipher_context_t ctx;
    unsigned char output[MBEDTLS_CIPHER_BLKSIZE_MAX];

    /* Convert the test parameters to binary data */


    mbedtls_cipher_init( &ctx );

    /* Validate the test inputs */
    TEST_ASSERT( block_a1_len <= 100 );
    TEST_ASSERT( block_a2_len <= 100 );
    TEST_ASSERT( block_a3_len <= 100 );

    TEST_ASSERT( block_b1_len <= 100 );
    TEST_ASSERT( block_b2_len <= 100 );
    TEST_ASSERT( block_b3_len <= 100 );

    /* Set up */
    TEST_ASSERT( ( cipher_info = mbedtls_cipher_info_from_type( cipher_type ) )
                    != NULL );

    TEST_ASSERT( mbedtls_cipher_setup( &ctx, cipher_info ) == 0 );

    TEST_ASSERT( mbedtls_cipher_cmac_starts( &ctx,
                                             (const unsigned char*)key->x,
                                             keybits ) == 0 );

    /* Sequence A */

    /* Multiple partial and complete blocks. A negative length means skip the
     * update operation */
    if( block_a1_len >= 0 )
        TEST_ASSERT( mbedtls_cipher_cmac_update( &ctx,
                                                 (unsigned char*)block_a1->x,
                                                 block_a1_len ) == 0);

    if( block_a2_len >= 0 )
        TEST_ASSERT( mbedtls_cipher_cmac_update( &ctx,
                                                 (unsigned char*)block_a2->x,
                                                 block_a2_len ) == 0);

    if( block_a3_len >= 0 )
        TEST_ASSERT( mbedtls_cipher_cmac_update( &ctx,
                                                 (unsigned char*)block_a3->x,
                                                  block_a3_len ) == 0);

    TEST_ASSERT( mbedtls_cipher_cmac_finish( &ctx, output ) == 0 );

    TEST_ASSERT( memcmp( output, expected_result_a->x, block_size )  == 0 );

    TEST_ASSERT( mbedtls_cipher_cmac_reset( &ctx ) == 0 );

    /* Sequence B */

    /* Multiple partial and complete blocks. A negative length means skip the
     * update operation */
    if( block_b1_len >= 0)
        TEST_ASSERT( mbedtls_cipher_cmac_update( &ctx,
                                                 (unsigned char*)block_b1->x,
                                                 block_b1_len ) == 0);

    if( block_b2_len >= 0 )
        TEST_ASSERT( mbedtls_cipher_cmac_update( &ctx,
                                                 (unsigned char*)block_b2->x,
                                                 block_b2_len ) == 0);

    if( block_b3_len >= 0 )
        TEST_ASSERT( mbedtls_cipher_cmac_update( &ctx,
                                                 (unsigned char*)block_b3->x,
                                                 block_b3_len ) == 0);

    TEST_ASSERT( mbedtls_cipher_cmac_finish( &ctx, output ) == 0 );

    TEST_ASSERT( memcmp( output, expected_result_b->x, block_size )  == 0 );

exit:
    mbedtls_cipher_free( &ctx );
}
/* END_CASE */