mirror/mbedtls
1
0
Fork 0
mirror of https://github.com/Mbed-TLS/mbedtls.git synced 2025-02-07 06:40:21 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Add testcases for psa_crypto_copy_and_free()

Browse Source 
Signed-off-by: David Horstmann <david.horstmann@arm.com>
This commit is contained in:
David Horstmann 2023-11-01 15:41:54 +00:00
parent 0fee689e57
commit 9700876520
2 changed files with 80 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
tests/suites/test_suite_psa_crypto.data
View File

@ -7457,3 +7457,12 @@ psa_crypto_alloc_and_copy_zero_length:0:1
PSA buffers alloc and copy both zero length
psa_crypto_alloc_and_copy_zero_length:1:1
PSA buffers copy and free
psa_crypto_copy_and_free:0:20:0:20:0:PSA_SUCCESS
PSA buffers copy and free, null input
psa_crypto_copy_and_free:1:0:0:20:0:PSA_SUCCESS
PSA buffers copy and free, null output
psa_crypto_copy_and_free:0:20:1:0:0:PSA_SUCCESS

71
tests/suites/test_suite_psa_crypto.function
View File

@ -10446,3 +10446,74 @@ exit:
mbedtls_free(buffer_copies.output);
}
/* END_CASE */
/* BEGIN_CASE */
void psa_crypto_copy_and_free(int input_null, int input_len,
int output_null, int output_len,
int orig_output_null,
int exp_ret)
{
uint8_t data[] = {0x12, 0x34, 0x56, 0x78};
psa_crypto_buffer_copy_t buffer_copies = { 0 };
uint8_t *input = NULL;
uint8_t *output = NULL;
uint8_t *output_for_comparison = NULL;
uint8_t *orig_output = NULL;
size_t calloc_len;
psa_status_t ret;
if (!input_null) {
/* If zero-length, ensure we actually allocate something
* rather than getting NULL. */
calloc_len = input_len == 0 ? 1 : input_len;
TEST_CALLOC(input, calloc_len);
}
if (!output_null) {
/* If zero-length, ensure we actually allocate something
* rather than getting NULL. */
calloc_len = output_len == 0 ? 1 : output_len;
TEST_CALLOC(output, calloc_len);
TEST_CALLOC(output_for_comparison, calloc_len);
for (int i = 0; i < output_len; i++) {
output[i] = data[i % sizeof(data)];
}
/* We expect the output buffer to be freed, so keep a copy
* for comparison. */
memcpy(output_for_comparison, output, output_len);
}
if (!orig_output_null) {
/* If zero-length, ensure we actually allocate something
* rather than getting NULL. */
calloc_len = output_len == 0 ? 1 : output_len;
TEST_CALLOC(orig_output, calloc_len);
}
buffer_copies.input = input;
buffer_copies.input_len = input_len;
buffer_copies.output = output;
buffer_copies.output_len = output_len;
buffer_copies.output_original = orig_output;
ret = psa_crypto_copy_and_free(&buffer_copies);
TEST_EQUAL((int) ret, exp_ret);
if (exp_ret == PSA_SUCCESS) {
TEST_ASSERT(buffer_copies.input == NULL);
TEST_ASSERT(buffer_copies.output == NULL);
if (!output_null) {
TEST_MEMORY_COMPARE(output_for_comparison, output_len,
buffer_copies.output_original, buffer_copies.output_len);
}
}
exit:
mbedtls_free(buffer_copies.input);
mbedtls_free(buffer_copies.output);
mbedtls_free(output_for_comparison);
mbedtls_free(orig_output);
}
/* END_CASE */