1 files changed, 16 insertions, 2 deletions

diff --git a/packet.c b/packet.c
index 7d8fab1e9..bc1b658ca 100644
--- a/packet.c
+++ b/packet.c
@@ -93,6 +93,8 @@
 static int connection_in = -1;
 static int connection_out = -1;
 
+static int setup_timeout = -1;
+
 /* Protocol flags for the remote side. */
 static u_int remote_protocol_flags = 0;
 
@@ -166,7 +168,7 @@ TAILQ_HEAD(, packet) outgoing;
  * packet_set_encryption_key is called.
  */
 void
-packet_set_connection(int fd_in, int fd_out)
+packet_set_connection(int fd_in, int fd_out, int new_setup_timeout)
 {
         Cipher *none = cipher_by_name("none");
 
@@ -174,6 +176,7 @@ packet_set_connection(int fd_in, int fd_out)
                 fatal("packet_set_connection: cannot load cipher 'none'");
         connection_in = fd_in;
         connection_out = fd_out;
+        setup_timeout = new_setup_timeout;
         cipher_init(&send_context, none, (const u_char *)"",
             0, NULL, 0, CIPHER_ENCRYPT);
         cipher_init(&receive_context, none, (const u_char *)"",
@@ -891,6 +894,7 @@ packet_read_seqnr(u_int32_t *seqnr_p)
         int type, len;
         fd_set *setp;
         char buf[8192];
+        struct timeval tv, *tvp;
         DBG(debug("packet_read()"));
 
         setp = (fd_set *)xcalloc(howmany(connection_in+1, NFDBITS),
@@ -922,11 +926,21 @@ packet_read_seqnr(u_int32_t *seqnr_p)
                     sizeof(fd_mask));
                 FD_SET(connection_in, setp);
 
+                if (setup_timeout > 0) {
+                        tvp = &tv;
+                        tv.tv_sec = setup_timeout;
+                        tv.tv_usec = 0;
+                } else
+                        tvp = NULL;
+
                 /* Wait for some data to arrive. */
-                while (select(connection_in + 1, setp, NULL, NULL, NULL) == -1 &&
+                while (select(connection_in + 1, setp, NULL, NULL, tvp) == -1 &&
                     (errno == EAGAIN || errno == EINTR))
                         ;
 
+                if (!FD_ISSET(connection_in, setp))
+                        fatal("packet_read: Setup timeout expired, giving up");
+
                 /* Read data from the socket. */
                 len = read(connection_in, buf, sizeof(buf));
                 if (len == 0) {