From f071954a76b672e4775780dea9635e95d1a5729d Mon Sep 17 00:00:00 2001
From: William Orr <will@worrbase.com>
Date: Sun, 14 Sep 2014 02:32:24 -0700
Subject: [PATCH] Check return code of connect(2).

Signed-off-by: William Orr <will@worrbase.com>
Reviewed-by: Andreas Schneider <asn@cryptomilk.org>
(cherry picked from commit 250f50648755415a138b1347a17a367ad384b66d)
---
 src/connect.c | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/src/connect.c b/src/connect.c
index 5ddd149f..807b1faf 100644
--- a/src/connect.c
+++ b/src/connect.c
@@ -382,7 +382,15 @@ socket_t ssh_connect_host_nonblocking(ssh_session session, const char *host,
         continue;
     }
 
-    connect(s, itr->ai_addr, itr->ai_addrlen);
+    rc = connect(s, itr->ai_addr, itr->ai_addrlen);
+    if (rc == -1) {
+      ssh_set_error(session, SSH_FATAL,
+          "Failed to connect: %s", strerror(errno));
+      ssh_connect_socket_close(s);
+      s = -1;
+      continue;
+    }
+
     break;
   }