X-Git-Url: https://git.distorted.org.uk/~mdw/sgt/putty/blobdiff_plain/0e03ceff4e371364f17c49ea370b6475cb0b7230..9711106bffb115afdc60e0ebd48251a094874596:/windows/winhandl.c

diff --git a/windows/winhandl.c b/windows/winhandl.c
index d220e640..222e5b93 100644
--- a/windows/winhandl.c
+++ b/windows/winhandl.c
@@ -75,6 +75,11 @@ struct handle_input {
     void *privdata;		       /* for client to remember who they are */
 
     /*
+     * Data set at initialisation and then read-only.
+     */
+    int flags;
+
+    /*
      * Data set by the input thread before signalling ev_to_main,
      * and read by the main thread after receiving that signal.
      */
@@ -95,13 +100,41 @@ struct handle_input {
 static DWORD WINAPI handle_input_threadfunc(void *param)
 {
     struct handle_input *ctx = (struct handle_input *) param;
+    OVERLAPPED ovl, *povl;
+    HANDLE oev;
+    int readlen;
+
+    if (ctx->flags & HANDLE_FLAG_OVERLAPPED) {
+	povl = &ovl;
+	oev = CreateEvent(NULL, TRUE, FALSE, NULL);
+    } else {
+	povl = NULL;
+    }
+
+    if (ctx->flags & HANDLE_FLAG_UNITBUFFER)
+	readlen = 1;
+    else
+	readlen = sizeof(ctx->buffer);
 
     while (1) {
-	ctx->readret = ReadFile(ctx->h, ctx->buffer, sizeof(ctx->buffer),
-				&ctx->len, NULL);
+	if (povl) {
+	    memset(povl, 0, sizeof(OVERLAPPED));
+	    povl->hEvent = oev;
+	}
+	ctx->readret = ReadFile(ctx->h, ctx->buffer, readlen,
+				&ctx->len, povl);
+	if (povl && !ctx->readret && GetLastError() == ERROR_IO_PENDING) {
+	    WaitForSingleObject(povl->hEvent, INFINITE);
+	    ctx->readret = GetOverlappedResult(ctx->h, povl, &ctx->len, FALSE);
+	}
+
 	if (!ctx->readret)
 	    ctx->len = 0;
 
+	if (ctx->readret && ctx->len == 0 &&
+	    (ctx->flags & HANDLE_FLAG_IGNOREEOF))
+	    continue;
+
 	SetEvent(ctx->ev_to_main);
 
 	if (!ctx->len)
@@ -112,6 +145,9 @@ static DWORD WINAPI handle_input_threadfunc(void *param)
 	    break;		       /* main thread told us to shut down */
     }
 
+    if (povl)
+	CloseHandle(oev);
+
     return 0;
 }
 
@@ -122,7 +158,8 @@ static DWORD WINAPI handle_input_threadfunc(void *param)
  */
 static void handle_throttle(struct handle_input *ctx, int backlog)
 {
-    assert(!ctx->defunct);
+    if (ctx->defunct)
+	return;
 
     /*
      * If there's a read operation already in progress, do nothing:
@@ -163,6 +200,11 @@ struct handle_output {
     void *privdata;		       /* for client to remember who they are */
 
     /*
+     * Data set at initialisation and then read-only.
+     */
+    int flags;
+
+    /*
      * Data set by the main thread before signalling ev_from_main,
      * and read by the input thread after receiving that signal.
      */
@@ -191,6 +233,12 @@ struct handle_output {
 static DWORD WINAPI handle_output_threadfunc(void *param)
 {
     struct handle_output *ctx = (struct handle_output *) param;
+    OVERLAPPED ovl, *povl;
+
+    if (ctx->flags & HANDLE_FLAG_OVERLAPPED)
+	povl = &ovl;
+    else
+	povl = NULL;
 
     while (1) {
 	WaitForSingleObject(ctx->ev_from_main, INFINITE);
@@ -198,8 +246,14 @@ static DWORD WINAPI handle_output_threadfunc(void *param)
 	    SetEvent(ctx->ev_to_main);
 	    break;
 	}
+	if (povl)
+	    memset(povl, 0, sizeof(OVERLAPPED));
 	ctx->writeret = WriteFile(ctx->h, ctx->buffer, ctx->len,
-				  &ctx->lenwritten, NULL);
+				  &ctx->lenwritten, povl);
+	if (povl && !ctx->writeret && GetLastError() == ERROR_IO_PENDING)
+	    ctx->writeret = GetOverlappedResult(ctx->h, povl,
+						&ctx->lenwritten, TRUE);
+
 	SetEvent(ctx->ev_to_main);
 	if (!ctx->writeret)
 	    break;
@@ -264,7 +318,7 @@ static int handle_find_evtomain(void *av, void *bv)
 }
 
 struct handle *handle_input_new(HANDLE handle, handle_inputfn_t gotdata,
-				void *privdata)
+				void *privdata, int flags)
 {
     struct handle *h = snew(struct handle);
 
@@ -273,11 +327,11 @@ struct handle *handle_input_new(HANDLE handle, handle_inputfn_t gotdata,
     h->u.i.ev_to_main = CreateEvent(NULL, FALSE, FALSE, NULL);
     h->u.i.ev_from_main = CreateEvent(NULL, FALSE, FALSE, NULL);
     h->u.i.gotdata = gotdata;
-    h->u.i.busy = FALSE;
     h->u.i.defunct = FALSE;
     h->u.i.moribund = FALSE;
     h->u.i.done = FALSE;
     h->u.i.privdata = privdata;
+    h->u.i.flags = flags;
 
     if (!handles_by_evtomain)
 	handles_by_evtomain = newtree234(handle_cmp_evtomain);
@@ -285,14 +339,13 @@ struct handle *handle_input_new(HANDLE handle, handle_inputfn_t gotdata,
 
     CreateThread(NULL, 0, handle_input_threadfunc,
 		 &h->u.i, 0, NULL);
-
-    handle_throttle(&h->u.i, 0);       /* start first read operation */
+    h->u.i.busy = TRUE;
 
     return h;
 }
 
 struct handle *handle_output_new(HANDLE handle, handle_outputfn_t sentdata,
-				 void *privdata)
+				 void *privdata, int flags)
 {
     struct handle *h = snew(struct handle);
 
@@ -307,6 +360,7 @@ struct handle *handle_output_new(HANDLE handle, handle_outputfn_t sentdata,
     h->u.o.privdata = privdata;
     bufchain_init(&h->u.o.queued_data);
     h->u.o.sentdata = sentdata;
+    h->u.o.flags = flags;
 
     if (!handles_by_evtomain)
 	handles_by_evtomain = newtree234(handle_cmp_evtomain);
@@ -393,6 +447,7 @@ void handle_free(struct handle *h)
 	 */
 	h->u.g.moribund = TRUE;
 	h->u.g.done = TRUE;
+	h->u.g.busy = TRUE;
 	SetEvent(h->u.g.ev_from_main);
     }
 }
@@ -426,6 +481,7 @@ void handle_got_event(HANDLE event)
 	    handle_destroy(h);
 	} else {
 	    h->u.g.done = TRUE;
+	    h->u.g.busy = TRUE;
 	    SetEvent(h->u.g.ev_from_main);
 	}
 	return;