IceOpenConnection: check for malloc failure on connect_to_you too

Fixes: https://gitlab.freedesktop.org/xorg/lib/libice/issues/4

Reported-by: mahendra <mahendra.n@samsung.com>
Signed-off-by: Alan Coopersmith <alan.coopersmith@oracle.com>

1 files changed, 9 insertions, 0 deletions

diff --git a/src/connect.c b/src/connect.c
index 1554ad6..821b727 100644
--- a/src/connect.c
+++ b/src/connect.c
@@ -236,6 +236,15 @@ IceOpenConnection (
     iceConn->ping_waits = NULL;
 
     iceConn->connect_to_you = malloc (sizeof (_IceConnectToYouInfo));
+    if (iceConn->connect_to_you == NULL)
+    {
+	_IceFreeConnection (iceConn);
+	if (errorStringRet && errorLength > 0) {
+	    strncpy (errorStringRet, "Can't malloc", errorLength);
+	    errorStringRet[errorLength - 1] = '\0';
+	}
+	return (NULL);
+    }
     iceConn->connect_to_you->auth_active = 0;
 
     /*