From c0b14e3df431e322873b30a0d3d89f4723793f23 Mon Sep 17 00:00:00 2001
From: Francois Perrad <francois.perrad@gadz.org>
Date: Fri, 13 Nov 2015 17:46:32 +0100
Subject: [PATCH] handle the return value of function

---
 bn_mp_sqrtmod_prime.c | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/bn_mp_sqrtmod_prime.c b/bn_mp_sqrtmod_prime.c
index 2499626..968729e 100644
--- a/bn_mp_sqrtmod_prime.c
+++ b/bn_mp_sqrtmod_prime.c
@@ -64,7 +64,7 @@ int mp_sqrtmod_prime(mp_int *n, mp_int *prime, mp_int *ret)
   }
 
   /* find a Z such that the Legendre symbol (Z|prime) == -1 */
-  mp_set_int(&Z, 2);
+  if ((res = mp_set_int(&Z, 2)) != MP_OKAY)                     goto cleanup;
   /* Z = 2 */
   while(1) {
     if ((res = mp_jacobi(&Z, prime, &legendre)) != MP_OKAY)     goto cleanup;
@@ -96,7 +96,7 @@ int mp_sqrtmod_prime(mp_int *n, mp_int *prime, mp_int *ret)
       i++;
     }
     if (i == 0) {
-      mp_copy(&R, ret);
+      if ((res = mp_copy(&R, ret)) != MP_OKAY)                  goto cleanup;
       res = MP_OKAY;
       goto cleanup;
     }