From c0fdc4655aa6c359abedc5704bd338bbcc50902e Mon Sep 17 00:00:00 2001 From: Li kunyu Date: Wed, 14 Dec 2022 14:47:05 +0800 Subject: [PATCH] connect: Add check after malloc allocation --- lib/connect.c | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/lib/connect.c b/lib/connect.c index 74b7251..38b8d19 100644 --- a/lib/connect.c +++ b/lib/connect.c @@ -469,6 +469,10 @@ static int reconnect(struct iscsi_context *iscsi, int force) tmp_iscsi->old_iscsi = iscsi->old_iscsi; } else { tmp_iscsi->old_iscsi = malloc(sizeof(struct iscsi_context)); + if (!tmp_iscsi->old_iscsi) { + free(tmp_iscsi); + return -1; + } memcpy(tmp_iscsi->old_iscsi, iscsi, sizeof(struct iscsi_context)); } memcpy(iscsi, tmp_iscsi, sizeof(struct iscsi_context));