X-Git-Url: https://git.openssl.org/?p=openssl.git;a=blobdiff_plain;f=test%2Fsm2crypttest.c;h=961256bc9027128cfeaadca9250c0bc7f83663cd;hp=e1e00d8aa4a6b80fc5fb1d87866e4c81c4b56716;hb=4e6647506331fc3b3ef5b23e5dbe188279ddd575;hpb=3d328a445c2ad0eff2e9e843c384711be58a7c2f

diff --git a/test/sm2crypttest.c b/test/sm2crypttest.c
index e1e00d8aa4..961256bc90 100644
--- a/test/sm2crypttest.c
+++ b/test/sm2crypttest.c
@@ -139,6 +139,7 @@ static int test_sm2(const EC_GROUP *group,
     unsigned char *expected = OPENSSL_hexstr2buf(ctext_hex, NULL);
 
     size_t ctext_len = 0;
+    size_t ptext_len = 0;
     uint8_t *ctext = NULL;
     uint8_t *recovered = NULL;
     size_t recovered_len = msg_len;
@@ -172,7 +173,11 @@ static int test_sm2(const EC_GROUP *group,
     if (rc == 0)
         goto done;
 
-    recovered = OPENSSL_zalloc(msg_len);
+    ptext_len = SM2_plaintext_size(key, digest, ctext_len);
+
+    TEST_int_eq(ptext_len, msg_len);
+
+    recovered = OPENSSL_zalloc(ptext_len);
     if (recovered == NULL)
         goto done;
     rc = SM2_decrypt(key, digest, ctext, ctext_len, recovered, &recovered_len);