lib/mpi: added missing NULL check

Added missing NULL check after mpi_alloc_limb_space(). Signed-off-by: Dmitry Kasatkin <[email protected]> Reviewed-by: Tetsuo Handa <[email protected]> Signed-off-by: James Morris <[email protected]>
strager · Feb 1, 2012 · 43b2c0a · 43b2c0a
1 parent a6d68ec
commit 43b2c0a
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/lib/mpi/mpi-div.c b/lib/mpi/mpi-div.c
@@ -210,6 +210,8 @@ int mpi_tdiv_qr(MPI quot, MPI rem, MPI num, MPI den)
 		 * numerator would be gradually overwritten by the quotient limbs.  */
 		if (qp == np) {	/* Copy NP object to temporary space.  */
 			np = marker[markidx++] = mpi_alloc_limb_space(nsize);
+			if (!np)
+				goto nomem;
 			MPN_COPY(np, qp, nsize);
 		}
 	} else			/* Put quotient at top of remainder. */