lib/mpi: added missing NULL check

Added missing NULL check after mpi_alloc(). Signed-off-by: Dmitry Kasatkin <[email protected]> Reviewed-by: Tetsuo Handa <[email protected]> Signed-off-by: James Morris <[email protected]>
sfionov · Feb 1, 2012 · c70c471 · c70c471
1 parent 62aa2b5
commit c70c471
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/lib/mpi/mpicoder.c b/lib/mpi/mpicoder.c
@@ -79,7 +79,8 @@ MPI do_encode_md(const void *sha_buffer, unsigned nbits)
 	}
 
 	a = mpi_alloc((nframe + BYTES_PER_MPI_LIMB - 1) / BYTES_PER_MPI_LIMB);
-	mpi_set_buffer(a, frame, nframe, 0);
+	if (a)
+		mpi_set_buffer(a, frame, nframe, 0);
 	kfree(frame);
 
 	return a;