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.
*/
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)
break; /* main thread told us to shut down */
}
+ if (povl)
+ CloseHandle(oev);
+
return 0;
}
*/
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:
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.
*/
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);
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;
}
struct handle *handle_input_new(HANDLE handle, handle_inputfn_t gotdata,
- void *privdata)
+ void *privdata, int flags)
{
struct handle *h = snew(struct handle);
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);
}
struct handle *handle_output_new(HANDLE handle, handle_outputfn_t sentdata,
- void *privdata)
+ void *privdata, int flags)
{
struct handle *h = snew(struct handle);
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);
*/
h->u.g.moribund = TRUE;
h->u.g.done = TRUE;
+ h->u.g.busy = TRUE;
SetEvent(h->u.g.ev_from_main);
}
}
handle_destroy(h);
} else {
h->u.g.done = TRUE;
+ h->u.g.busy = TRUE;
SetEvent(h->u.g.ev_from_main);
}
return;