From 33f7d66304cf7f98205bfa89d16a2122d3f36d42 Mon Sep 17 00:00:00 2001 From: Raef Coles Date: Thu, 13 Oct 2022 14:24:08 +0100 Subject: [PATCH] Add output check to export too-big buffer tests Signed-off-by: Raef Coles --- tests/suites/test_suite_lmots.function | 2 ++ tests/suites/test_suite_lms.function | 2 ++ 2 files changed, 4 insertions(+) diff --git a/tests/suites/test_suite_lmots.function b/tests/suites/test_suite_lmots.function index 0f339259cb..13d1ee4cb4 100644 --- a/tests/suites/test_suite_lmots.function +++ b/tests/suites/test_suite_lmots.function @@ -184,6 +184,8 @@ void lmots_import_export_test ( data_t * pub_key, int expected_import_rc ) TEST_EQUAL( mbedtls_lmots_export_public_key( &ctx, exported_pub_key, exported_pub_key_buf_size, NULL ), 0 ); + ASSERT_COMPARE( pub_key->x, pub_key->len, + exported_pub_key, exported_pub_key_size ); mbedtls_free(exported_pub_key); exported_pub_key = NULL; } diff --git a/tests/suites/test_suite_lms.function b/tests/suites/test_suite_lms.function index ff117ea27f..1e5ac42a65 100644 --- a/tests/suites/test_suite_lms.function +++ b/tests/suites/test_suite_lms.function @@ -186,6 +186,8 @@ void lms_import_export_test ( data_t * pub_key, int expected_import_rc ) TEST_EQUAL( mbedtls_lms_export_public_key( &ctx, exported_pub_key, exported_pub_key_buf_size, NULL ), 0 ); + ASSERT_COMPARE( pub_key->x, pub_key->len, + exported_pub_key, exported_pub_key_size ); mbedtls_free(exported_pub_key); exported_pub_key = NULL; }