X-Git-Url: https://git.distorted.org.uk/~mdw/sgt/putty/blobdiff_plain/6da411554bfe4c2a8ddfbb0616b0030ea5e813f5..055817455466c8eb60392f30bb7c689763962e17:/windows/winplink.c

diff --git a/windows/winplink.c b/windows/winplink.c
index b99646af..c1366979 100644
--- a/windows/winplink.c
+++ b/windows/winplink.c
@@ -231,6 +231,7 @@ static void usage(void)
     printf("  -A -a     enable / disable agent forwarding\n");
     printf("  -t -T     enable / disable pty allocation\n");
     printf("  -1 -2     force use of particular protocol version\n");
+    printf("  -4 -6     force use of IPv4 or IPv6\n");
     printf("  -C        enable compression\n");
     printf("  -i key    private key file for authentication\n");
     printf("  -s        remote command is an SSH subsystem (SSH-2 only)\n");
@@ -279,6 +280,7 @@ int main(int argc, char **argv)
     int exitcode;
     int errors;
     int use_subsystem = 0;
+    long now, next;
 
     ssh_get_line = console_get_line;
 
@@ -631,8 +633,11 @@ int main(int argc, char **argv)
 	cleanup_exit(1);
     }
 
+    now = GETTICKCOUNT();
+
     while (1) {
 	int n;
+	DWORD ticks;
 
 	if (!sending && back->sendok(backhandle)) {
 	    /*
@@ -661,9 +666,16 @@ int main(int argc, char **argv)
 	    sending = TRUE;
 	}
 
-	n = MsgWaitForMultipleObjects(4, handles, FALSE, INFINITE,
+	if (run_timers(now, &next)) {
+	    ticks = next - GETTICKCOUNT();
+	    if (ticks < 0) ticks = 0;  /* just in case */
+	} else {
+	    ticks = INFINITE;
+	}
+
+	n = MsgWaitForMultipleObjects(4, handles, FALSE, ticks,
 				      QS_POSTMESSAGE);
-	if (n == 0) {
+	if (n == WAIT_OBJECT_0 + 0) {
 	    WSANETWORKEVENTS things;
 	    SOCKET socket;
 	    extern SOCKET first_socket(int *), next_socket(int *);
@@ -724,7 +736,7 @@ int main(int argc, char **argv)
                         }
 		}
 	    }
-	} else if (n == 1) {
+	} else if (n == WAIT_OBJECT_0 + 1) {
 	    reading = 0;
 	    noise_ultralight(idata.len);
 	    if (connopen && back->socket(backhandle) != NULL) {
@@ -734,7 +746,7 @@ int main(int argc, char **argv)
 		    back->special(backhandle, TS_EOF);
 		}
 	    }
-	} else if (n == 2) {
+	} else if (n == WAIT_OBJECT_0 + 2) {
 	    odata.busy = 0;
 	    if (!odata.writeret) {
 		fprintf(stderr, "Unable to write to standard output\n");
@@ -747,7 +759,7 @@ int main(int argc, char **argv)
 		back->unthrottle(backhandle, bufchain_size(&stdout_data) +
 				 bufchain_size(&stderr_data));
 	    }
-	} else if (n == 3) {
+	} else if (n == WAIT_OBJECT_0 + 3) {
 	    edata.busy = 0;
 	    if (!edata.writeret) {
 		fprintf(stderr, "Unable to write to standard output\n");
@@ -760,7 +772,7 @@ int main(int argc, char **argv)
 		back->unthrottle(backhandle, bufchain_size(&stdout_data) +
 				 bufchain_size(&stderr_data));
 	    }
-	} else if (n == 4) {
+	} else if (n == WAIT_OBJECT_0 + 4) {
 	    MSG msg;
 	    while (PeekMessage(&msg, INVALID_HANDLE_VALUE,
 			       WM_AGENT_CALLBACK, WM_AGENT_CALLBACK,
@@ -770,6 +782,13 @@ int main(int argc, char **argv)
 		sfree(c);
 	    }
 	}
+
+	if (n == WAIT_TIMEOUT) {
+	    now = next;
+	} else {
+	    now = GETTICKCOUNT();
+	}
+
 	if (!reading && back->sendbuffer(backhandle) < MAX_STDIN_BACKLOG) {
 	    SetEvent(idata.eventback);
 	    reading = 1;