| /* BEGIN_HEADER */ |
| #include "mbedtls/blowfish.h" |
| /* END_HEADER */ |
| |
| /* BEGIN_DEPENDENCIES |
| * depends_on:MBEDTLS_BLOWFISH_C |
| * END_DEPENDENCIES |
| */ |
| |
| /* BEGIN_CASE */ |
| void blowfish_valid_param() |
| { |
| TEST_VALID_PARAM(mbedtls_blowfish_free(NULL)); |
| } |
| /* END_CASE */ |
| |
| /* BEGIN_CASE depends_on:MBEDTLS_CHECK_PARAMS:!MBEDTLS_PARAM_FAILED_ALT */ |
| void blowfish_invalid_param() |
| { |
| mbedtls_blowfish_context ctx; |
| unsigned char buf[16] = { 0 }; |
| size_t const valid_keylength = sizeof(buf) * 8; |
| size_t valid_mode = MBEDTLS_BLOWFISH_ENCRYPT; |
| size_t invalid_mode = 42; |
| size_t off; |
| ((void) off); |
| |
| TEST_INVALID_PARAM(mbedtls_blowfish_init(NULL)); |
| TEST_VALID_PARAM(mbedtls_blowfish_free(NULL)); |
| |
| TEST_INVALID_PARAM_RET(MBEDTLS_ERR_BLOWFISH_BAD_INPUT_DATA, |
| mbedtls_blowfish_setkey(NULL, |
| buf, |
| valid_keylength)); |
| TEST_INVALID_PARAM_RET(MBEDTLS_ERR_BLOWFISH_BAD_INPUT_DATA, |
| mbedtls_blowfish_setkey(&ctx, |
| NULL, |
| valid_keylength)); |
| |
| TEST_INVALID_PARAM_RET(MBEDTLS_ERR_BLOWFISH_BAD_INPUT_DATA, |
| mbedtls_blowfish_crypt_ecb(NULL, |
| valid_mode, |
| buf, buf)); |
| TEST_INVALID_PARAM_RET(MBEDTLS_ERR_BLOWFISH_BAD_INPUT_DATA, |
| mbedtls_blowfish_crypt_ecb(&ctx, |
| invalid_mode, |
| buf, buf)); |
| TEST_INVALID_PARAM_RET(MBEDTLS_ERR_BLOWFISH_BAD_INPUT_DATA, |
| mbedtls_blowfish_crypt_ecb(&ctx, |
| valid_mode, |
| NULL, buf)); |
| TEST_INVALID_PARAM_RET(MBEDTLS_ERR_BLOWFISH_BAD_INPUT_DATA, |
| mbedtls_blowfish_crypt_ecb(&ctx, |
| valid_mode, |
| buf, NULL)); |
| |
| #if defined(MBEDTLS_CIPHER_MODE_CBC) |
| TEST_INVALID_PARAM_RET(MBEDTLS_ERR_BLOWFISH_BAD_INPUT_DATA, |
| mbedtls_blowfish_crypt_cbc(NULL, |
| valid_mode, |
| sizeof(buf), |
| buf, buf, buf)); |
| TEST_INVALID_PARAM_RET(MBEDTLS_ERR_BLOWFISH_BAD_INPUT_DATA, |
| mbedtls_blowfish_crypt_cbc(&ctx, |
| invalid_mode, |
| sizeof(buf), |
| buf, buf, buf)); |
| TEST_INVALID_PARAM_RET(MBEDTLS_ERR_BLOWFISH_BAD_INPUT_DATA, |
| mbedtls_blowfish_crypt_cbc(&ctx, |
| valid_mode, |
| sizeof(buf), |
| NULL, buf, buf)); |
| TEST_INVALID_PARAM_RET(MBEDTLS_ERR_BLOWFISH_BAD_INPUT_DATA, |
| mbedtls_blowfish_crypt_cbc(&ctx, |
| valid_mode, |
| sizeof(buf), |
| buf, NULL, buf)); |
| TEST_INVALID_PARAM_RET(MBEDTLS_ERR_BLOWFISH_BAD_INPUT_DATA, |
| mbedtls_blowfish_crypt_cbc(&ctx, |
| valid_mode, |
| sizeof(buf), |
| buf, buf, NULL)); |
| #endif /* MBEDTLS_CIPHER_MODE_CBC */ |
| |
| #if defined(MBEDTLS_CIPHER_MODE_CFB) |
| TEST_INVALID_PARAM_RET(MBEDTLS_ERR_BLOWFISH_BAD_INPUT_DATA, |
| mbedtls_blowfish_crypt_cfb64(NULL, |
| valid_mode, |
| sizeof(buf), |
| &off, buf, |
| buf, buf)); |
| TEST_INVALID_PARAM_RET(MBEDTLS_ERR_BLOWFISH_BAD_INPUT_DATA, |
| mbedtls_blowfish_crypt_cfb64(&ctx, |
| invalid_mode, |
| sizeof(buf), |
| &off, buf, |
| buf, buf)); |
| TEST_INVALID_PARAM_RET(MBEDTLS_ERR_BLOWFISH_BAD_INPUT_DATA, |
| mbedtls_blowfish_crypt_cfb64(&ctx, |
| valid_mode, |
| sizeof(buf), |
| NULL, buf, |
| buf, buf)); |
| TEST_INVALID_PARAM_RET(MBEDTLS_ERR_BLOWFISH_BAD_INPUT_DATA, |
| mbedtls_blowfish_crypt_cfb64(&ctx, |
| valid_mode, |
| sizeof(buf), |
| &off, NULL, |
| buf, buf)); |
| TEST_INVALID_PARAM_RET(MBEDTLS_ERR_BLOWFISH_BAD_INPUT_DATA, |
| mbedtls_blowfish_crypt_cfb64(&ctx, |
| valid_mode, |
| sizeof(buf), |
| &off, buf, |
| NULL, buf)); |
| TEST_INVALID_PARAM_RET(MBEDTLS_ERR_BLOWFISH_BAD_INPUT_DATA, |
| mbedtls_blowfish_crypt_cfb64(&ctx, |
| valid_mode, |
| sizeof(buf), |
| &off, buf, |
| buf, NULL)); |
| #endif /* MBEDTLS_CIPHER_MODE_CFB */ |
| |
| #if defined(MBEDTLS_CIPHER_MODE_CTR) |
| TEST_INVALID_PARAM_RET(MBEDTLS_ERR_BLOWFISH_BAD_INPUT_DATA, |
| mbedtls_blowfish_crypt_ctr(NULL, |
| sizeof(buf), |
| &off, |
| buf, buf, |
| buf, buf)); |
| TEST_INVALID_PARAM_RET(MBEDTLS_ERR_BLOWFISH_BAD_INPUT_DATA, |
| mbedtls_blowfish_crypt_ctr(&ctx, |
| sizeof(buf), |
| NULL, |
| buf, buf, |
| buf, buf)); |
| TEST_INVALID_PARAM_RET(MBEDTLS_ERR_BLOWFISH_BAD_INPUT_DATA, |
| mbedtls_blowfish_crypt_ctr(&ctx, |
| sizeof(buf), |
| &off, |
| NULL, buf, |
| buf, buf)); |
| TEST_INVALID_PARAM_RET(MBEDTLS_ERR_BLOWFISH_BAD_INPUT_DATA, |
| mbedtls_blowfish_crypt_ctr(&ctx, |
| sizeof(buf), |
| &off, |
| buf, NULL, |
| buf, buf)); |
| TEST_INVALID_PARAM_RET(MBEDTLS_ERR_BLOWFISH_BAD_INPUT_DATA, |
| mbedtls_blowfish_crypt_ctr(&ctx, |
| sizeof(buf), |
| &off, |
| buf, buf, |
| NULL, buf)); |
| TEST_INVALID_PARAM_RET(MBEDTLS_ERR_BLOWFISH_BAD_INPUT_DATA, |
| mbedtls_blowfish_crypt_ctr(&ctx, |
| sizeof(buf), |
| &off, |
| buf, buf, |
| buf, NULL)); |
| #endif /* MBEDTLS_CIPHER_MODE_CTR */ |
| |
| exit: |
| return; |
| } |
| /* END_CASE */ |
| |
| /* BEGIN_CASE */ |
| void blowfish_encrypt_ecb(data_t *key_str, data_t *src_str, |
| data_t *dst, int setkey_result) |
| { |
| unsigned char output[100]; |
| mbedtls_blowfish_context ctx; |
| |
| memset(output, 0x00, 100); |
| mbedtls_blowfish_init(&ctx); |
| |
| |
| TEST_ASSERT(mbedtls_blowfish_setkey(&ctx, key_str->x, key_str->len * 8) == setkey_result); |
| if (setkey_result == 0) { |
| TEST_ASSERT(mbedtls_blowfish_crypt_ecb(&ctx, MBEDTLS_BLOWFISH_ENCRYPT, src_str->x, |
| output) == 0); |
| |
| TEST_ASSERT(mbedtls_test_hexcmp(output, dst->x, 8, dst->len) == 0); |
| } |
| |
| exit: |
| mbedtls_blowfish_free(&ctx); |
| } |
| /* END_CASE */ |
| |
| /* BEGIN_CASE */ |
| void blowfish_decrypt_ecb(data_t *key_str, data_t *src_str, |
| data_t *dst, int setkey_result) |
| { |
| unsigned char output[100]; |
| mbedtls_blowfish_context ctx; |
| |
| memset(output, 0x00, 100); |
| mbedtls_blowfish_init(&ctx); |
| |
| |
| TEST_ASSERT(mbedtls_blowfish_setkey(&ctx, key_str->x, key_str->len * 8) == setkey_result); |
| if (setkey_result == 0) { |
| TEST_ASSERT(mbedtls_blowfish_crypt_ecb(&ctx, MBEDTLS_BLOWFISH_DECRYPT, src_str->x, |
| output) == 0); |
| |
| TEST_ASSERT(mbedtls_test_hexcmp(output, dst->x, 8, dst->len) == 0); |
| } |
| |
| exit: |
| mbedtls_blowfish_free(&ctx); |
| } |
| /* END_CASE */ |
| |
| /* BEGIN_CASE depends_on:MBEDTLS_CIPHER_MODE_CBC */ |
| void blowfish_encrypt_cbc(data_t *key_str, data_t *iv_str, |
| data_t *src_str, data_t *dst, |
| int cbc_result) |
| { |
| unsigned char output[100]; |
| mbedtls_blowfish_context ctx; |
| |
| memset(output, 0x00, 100); |
| mbedtls_blowfish_init(&ctx); |
| |
| |
| mbedtls_blowfish_setkey(&ctx, key_str->x, key_str->len * 8); |
| |
| TEST_ASSERT(mbedtls_blowfish_crypt_cbc(&ctx, MBEDTLS_BLOWFISH_ENCRYPT, src_str->len, iv_str->x, |
| src_str->x, output) == cbc_result); |
| if (cbc_result == 0) { |
| |
| TEST_ASSERT(mbedtls_test_hexcmp(output, dst->x, |
| src_str->len, dst->len) == 0); |
| } |
| |
| exit: |
| mbedtls_blowfish_free(&ctx); |
| } |
| /* END_CASE */ |
| |
| /* BEGIN_CASE depends_on:MBEDTLS_CIPHER_MODE_CBC */ |
| void blowfish_decrypt_cbc(data_t *key_str, data_t *iv_str, |
| data_t *src_str, data_t *dst, |
| int cbc_result) |
| { |
| unsigned char output[100]; |
| mbedtls_blowfish_context ctx; |
| |
| memset(output, 0x00, 100); |
| mbedtls_blowfish_init(&ctx); |
| |
| |
| mbedtls_blowfish_setkey(&ctx, key_str->x, key_str->len * 8); |
| TEST_ASSERT(mbedtls_blowfish_crypt_cbc(&ctx, MBEDTLS_BLOWFISH_DECRYPT, src_str->len, iv_str->x, |
| src_str->x, output) == cbc_result); |
| if (cbc_result == 0) { |
| |
| TEST_ASSERT(mbedtls_test_hexcmp(output, dst->x, src_str->len, |
| dst->len) == 0); |
| } |
| |
| exit: |
| mbedtls_blowfish_free(&ctx); |
| } |
| /* END_CASE */ |
| |
| /* BEGIN_CASE depends_on:MBEDTLS_CIPHER_MODE_CFB */ |
| void blowfish_encrypt_cfb64(data_t *key_str, data_t *iv_str, |
| data_t *src_str, data_t *dst) |
| { |
| unsigned char output[100]; |
| mbedtls_blowfish_context ctx; |
| size_t iv_offset = 0; |
| |
| memset(output, 0x00, 100); |
| mbedtls_blowfish_init(&ctx); |
| |
| |
| mbedtls_blowfish_setkey(&ctx, key_str->x, key_str->len * 8); |
| TEST_ASSERT(mbedtls_blowfish_crypt_cfb64(&ctx, MBEDTLS_BLOWFISH_ENCRYPT, src_str->len, |
| &iv_offset, iv_str->x, src_str->x, output) == 0); |
| |
| TEST_ASSERT(mbedtls_test_hexcmp(output, dst->x, src_str->len, |
| dst->len) == 0); |
| |
| exit: |
| mbedtls_blowfish_free(&ctx); |
| } |
| /* END_CASE */ |
| |
| /* BEGIN_CASE depends_on:MBEDTLS_CIPHER_MODE_CFB */ |
| void blowfish_decrypt_cfb64(data_t *key_str, data_t *iv_str, |
| data_t *src_str, data_t *dst) |
| { |
| unsigned char output[100]; |
| mbedtls_blowfish_context ctx; |
| size_t iv_offset = 0; |
| |
| memset(output, 0x00, 100); |
| mbedtls_blowfish_init(&ctx); |
| |
| |
| mbedtls_blowfish_setkey(&ctx, key_str->x, key_str->len * 8); |
| TEST_ASSERT(mbedtls_blowfish_crypt_cfb64(&ctx, MBEDTLS_BLOWFISH_DECRYPT, src_str->len, |
| &iv_offset, iv_str->x, src_str->x, output) == 0); |
| |
| TEST_ASSERT(mbedtls_test_hexcmp(output, dst->x, src_str->len, |
| dst->len) == 0); |
| |
| exit: |
| mbedtls_blowfish_free(&ctx); |
| } |
| /* END_CASE */ |
| |
| /* BEGIN_CASE depends_on:MBEDTLS_CIPHER_MODE_CTR */ |
| void blowfish_encrypt_ctr(data_t *key_str, data_t *iv_str, |
| data_t *src_str, data_t *dst) |
| { |
| unsigned char stream_str[100]; |
| unsigned char output[100]; |
| mbedtls_blowfish_context ctx; |
| size_t iv_offset = 0; |
| |
| memset(stream_str, 0x00, 100); |
| memset(output, 0x00, 100); |
| mbedtls_blowfish_init(&ctx); |
| |
| |
| mbedtls_blowfish_setkey(&ctx, key_str->x, key_str->len * 8); |
| TEST_ASSERT(mbedtls_blowfish_crypt_ctr(&ctx, src_str->len, &iv_offset, iv_str->x, stream_str, |
| src_str->x, output) == 0); |
| |
| TEST_ASSERT(mbedtls_test_hexcmp(output, dst->x, src_str->len, |
| dst->len) == 0); |
| |
| exit: |
| mbedtls_blowfish_free(&ctx); |
| } |
| /* END_CASE */ |