commit 6b2c573b3d4e72a17694858dbb4c34a7ee7f8ace
author Raef Coles <raef.coles@arm.com> Thu Oct 13 09:41:39 2022 +0100
committer Raef Coles <raef.coles@arm.com> Thu Oct 13 14:30:24 2022 +0100
tree 639d5d4d498214a9728fa17d539f86913e2b3511
parent 534f66f3f04e549ad9efa9dba2b4edb608b2f229

And export buffer too large test to LMS and LMOTS

Signed-off-by: Raef Coles <raef.coles@arm.com>

tests/suites/test_suite_lms.function

diff --git a/tests/suites/test_suite_lms.function b/tests/suites/test_suite_lms.function
index 97f45e6..51addea 100644
--- a/tests/suites/test_suite_lms.function
+++ b/tests/suites/test_suite_lms.function
@@ -180,6 +180,15 @@
                     MBEDTLS_ERR_LMS_BUFFER_TOO_SMALL );
         mbedtls_free(exported_pub_key);
         exported_pub_key = NULL;
+
+        /* Export into too-large buffer should succeed */
+        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 ),
+                    0 );
+        mbedtls_free(exported_pub_key);
+        exported_pub_key = NULL;
     }
 
 exit: