/* 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); #if defined(MBEDTLS_AES_C) 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); #endif 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 */