lib/mpi: Add the return value check of kcalloc()

Add the return value check of kcalloc() to avoid potential NULL ptr dereference. Fixes: a8ea8bd ("lib/mpi: Extend the MPI library") Signed-off-by: Zizhuang Deng <[email protected]> Reviewed-by: Tianjia Zhang <[email protected]> Signed-off-by: Herbert Xu <[email protected]>
wafgo · Jan 7, 2022 · dd827ab · dd827ab
1 parent 4cab5df
commit dd827ab
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/lib/mpi/mpi-mod.c b/lib/mpi/mpi-mod.c
@@ -40,6 +40,8 @@ mpi_barrett_t mpi_barrett_init(MPI m, int copy)
 
 	mpi_normalize(m);
 	ctx = kcalloc(1, sizeof(*ctx), GFP_KERNEL);
+	if (!ctx)
+		return NULL;
 
 	if (copy) {
 		ctx->m = mpi_copy(m);