From 6d7d94a821425f0efc7daec621d83aa930715ca8 Mon Sep 17 00:00:00 2001 From: Raef Coles Date: Thu, 13 Oct 2022 17:55:46 +0100 Subject: [PATCH] Use correct export length in LMS export tests Signed-off-by: Raef Coles --- tests/suites/test_suite_lmots.function | 5 ++--- tests/suites/test_suite_lms.function | 5 ++--- 2 files changed, 4 insertions(+), 6 deletions(-) diff --git a/tests/suites/test_suite_lmots.function b/tests/suites/test_suite_lmots.function index b5b18b50f8..53ab02ff6e 100644 --- a/tests/suites/test_suite_lmots.function +++ b/tests/suites/test_suite_lmots.function @@ -182,10 +182,9 @@ void lmots_import_export_test ( data_t * pub_key, int expected_import_rc ) exported_pub_key_buf_size = MBEDTLS_LMOTS_PUBLIC_KEY_LEN(MBEDTLS_LMOTS_SHA256_N32_W8) + 1; ASSERT_ALLOC( exported_pub_key, exported_pub_key_buf_size); TEST_EQUAL( mbedtls_lmots_export_public_key( &ctx, exported_pub_key, - exported_pub_key_buf_size, NULL ), + exported_pub_key_buf_size, + &exported_pub_key_size ), 0 ); - TEST_EQUAL( exported_pub_key_size, - MBEDTLS_LMOTS_PUBLIC_KEY_LEN(MBEDTLS_LMOTS_SHA256_N32_W8) ); ASSERT_COMPARE( pub_key->x, pub_key->len, exported_pub_key, exported_pub_key_size ); mbedtls_free(exported_pub_key); diff --git a/tests/suites/test_suite_lms.function b/tests/suites/test_suite_lms.function index baf74abc97..c5c8aa4b90 100644 --- a/tests/suites/test_suite_lms.function +++ b/tests/suites/test_suite_lms.function @@ -184,10 +184,9 @@ void lms_import_export_test ( data_t * pub_key, int expected_import_rc ) exported_pub_key_buf_size = MBEDTLS_LMS_PUBLIC_KEY_LEN(MBEDTLS_LMS_SHA256_M32_H10) + 1; ASSERT_ALLOC( exported_pub_key, exported_pub_key_buf_size); TEST_EQUAL( mbedtls_lms_export_public_key( &ctx, exported_pub_key, - exported_pub_key_buf_size, NULL ), + exported_pub_key_buf_size, + &exported_pub_key_size ), 0 ); - TEST_EQUAL( exported_pub_key_size, - MBEDTLS_LMS_PUBLIC_KEY_LEN(MBEDTLS_LMS_SHA256_M32_H10 ) ); ASSERT_COMPARE( pub_key->x, pub_key->len, exported_pub_key, exported_pub_key_size ); mbedtls_free(exported_pub_key);