GitLab repository storage has been migrated to hashed layout. Please contact Infrastructure team if you notice any issues with repositories or hooks.

giowin32.c 56.4 KB
Newer Older
1 2 3 4 5
/* GLIB - Library of useful routines for C programming
 * Copyright (C) 1995-1997  Peter Mattis, Spencer Kimball and Josh MacDonald
 *
 * giowin32.c: IO Channels for Win32.
 * Copyright 1998 Owen Taylor and Tor Lillqvist
6
 * Copyright 1999-2000 Tor Lillqvist and Craig Setera
7
 * Copyright 2001-2003 Andrew Lanoix
8 9
 *
 * This library is free software; you can redistribute it and/or
10
 * modify it under the terms of the GNU Lesser General Public
11 12 13 14 15
 * License as published by the Free Software Foundation; either
 * version 2 of the License, or (at your option) any later version.
 *
 * This library is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
16
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
17
 * Lesser General Public License for more details.
18
 *
19
 * You should have received a copy of the GNU Lesser General Public
20 21 22 23 24
 * License along with this library; if not, write to the
 * Free Software Foundation, Inc., 59 Temple Place - Suite 330,
 * Boston, MA 02111-1307, USA.
 */

25
/*
26
 * Modified by the GLib Team and others 1997-2000.  See the AUTHORS
27 28
 * file for a list of people on the GLib Team.  See the ChangeLog
 * files for a list of changes.  These files are distributed with
29
 * GLib at ftp://ftp.gtk.org/pub/gtk/.
30 31
 */

Owen Taylor's avatar
Owen Taylor committed
32 33
#include "config.h"

34
#include "glib.h"
35 36

#include <stdlib.h>
37
#include <winsock2.h>
38
#include <windows.h>
39
#include <conio.h>
40 41
#include <fcntl.h>
#include <io.h>
42
#include <process.h>
43
#include <errno.h>
44
#include <sys/stat.h>
45

46
#include "gstdio.h"
47 48
#include "glibintl.h"

49 50
#include "galias.h"

51 52 53
typedef struct _GIOWin32Channel GIOWin32Channel;
typedef struct _GIOWin32Watch GIOWin32Watch;

54
#define BUFFER_SIZE 4096
55 56

typedef enum {
57 58
  G_IO_WIN32_WINDOWS_MESSAGES,	/* Windows messages */
  G_IO_WIN32_FILE_DESC,		/* Unix-like file descriptors from
59
				 * _open() or _pipe(), except for console IO.
60 61
				 * Have to create separate thread to read.
				 */
62
  G_IO_WIN32_CONSOLE,		/* Console IO (usually stdin, stdout, stderr) */
63
  G_IO_WIN32_SOCKET		/* Sockets. No separate thread */
64 65 66 67 68 69 70 71 72
} GIOWin32ChannelType;

struct _GIOWin32Channel {
  GIOChannel channel;
  gint fd;			/* Either a Unix-like file handle as provided
				 * by the Microsoft C runtime, or a SOCKET
				 * as provided by WinSock.
				 */
  GIOWin32ChannelType type;
73
  
74 75
  gboolean debug;

76
  /* This is used by G_IO_WIN32_WINDOWS_MESSAGES channels */
77
  HWND hwnd;			/* handle of window, or NULL */
78
  
79 80 81
  /* Following fields are used by fd channels. */
  CRITICAL_SECTION mutex;

82 83 84 85
  int direction;		/* 0 means we read from it,
				 * 1 means we write to it.
				 */

86 87 88 89 90 91 92 93 94 95 96 97 98
  gboolean running;		/* Is reader thread running. FALSE if
				 * EOF has been reached.
				 */
  gboolean needs_close;		/* If the channel has been closed while
				 * the reader thread was still running.
				 */
  guint thread_id;		/* If non-NULL has a reader thread, or has
				 * had.*/
  HANDLE data_avail_event;

  gushort revents;

  /* Following fields used by fd channels for input */
99 100 101 102 103 104 105 106 107 108 109 110 111 112
  
  /* Data is kept in a circular buffer. To be able to distinguish between
   * empty and full buffer, we cannot fill it completely, but have to
   * leave a one character gap.
   *
   * Data available is between indexes rdp and wrp-1 (modulo BUFFER_SIZE).
   *
   * Empty:    wrp == rdp
   * Full:     (wrp + 1) % BUFFER_SIZE == rdp
   * Partial:  otherwise
   */
  guchar *buffer;		/* (Circular) buffer */
  gint wrp, rdp;		/* Buffer indices for writing and reading */
  HANDLE space_avail_event;
113 114

  /* Following fields used by socket channels */
115 116 117 118
  int event_mask;
  int last_events;
  int event;
  gboolean write_would_have_blocked;
119 120
};

121 122 123
#define LOCK(mutex) EnterCriticalSection (&mutex)
#define UNLOCK(mutex) LeaveCriticalSection (&mutex)

124
struct _GIOWin32Watch {
125
  GSource       source;
126 127 128 129 130
  GPollFD       pollfd;
  GIOChannel   *channel;
  GIOCondition  condition;
};

131
static void
132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165
g_win32_print_access_mode (int flags)
{
  g_print ("%s%s%s%s%s%s%s%s%s%s",
	   ((flags & 0x3) == _O_RDWR ? "O_RDWR" :
	    ((flags & 0x3) == _O_RDONLY ? "O_RDONLY" :
	     ((flags & 0x3) == _O_WRONLY ? "O_WRONLY" : "0"))),
	   (flags & _O_APPEND ? "|O_APPEND" : ""),
	   (flags & _O_RANDOM ? "|O_RANDOM" : ""),
	   (flags & _O_SEQUENTIAL ? "|O_SEQUENTIAL" : ""),
	   (flags & _O_TEMPORARY ? "|O_TEMPORARY" : ""),
	   (flags & _O_CREAT ? "|O_CREAT" : ""),
	   (flags & _O_TRUNC ? "|O_TRUNC" : ""),
	   (flags & _O_EXCL ? "|O_EXCL" : ""),
	   (flags & _O_TEXT ? "|O_TEXT" : ""),
	   (flags & _O_BINARY ? "|O_BINARY" : ""));
}

static void
g_win32_print_gioflags (GIOFlags flags)
{
  char *bar = "";

  if (flags & G_IO_FLAG_APPEND)
    bar = "|", g_print ("APPEND");
  if (flags & G_IO_FLAG_NONBLOCK)
    g_print ("%sNONBLOCK", bar), bar = "|";
  if (flags & G_IO_FLAG_IS_READABLE)
    g_print ("%sREADABLE", bar), bar = "|";
  if (flags & G_IO_FLAG_IS_WRITEABLE)
    g_print ("%sWRITEABLE", bar), bar = "|";
  if (flags & G_IO_FLAG_IS_SEEKABLE)
    g_print ("%sSEEKABLE", bar), bar = "|";
}

166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223
static const char *
event_mask_to_string (int mask)
{
  char buf[100];
  int checked_bits = 0;
  char *bufp = buf;

  if (mask == 0)
    return "";

#define BIT(n) checked_bits |= FD_##n; if (mask & FD_##n) bufp += sprintf (bufp, "%s" #n, (bufp>buf ? "|" : ""))

  BIT (READ);
  BIT (WRITE);
  BIT (OOB);
  BIT (ACCEPT);
  BIT (CONNECT);
  BIT (CLOSE);
  BIT (QOS);
  BIT (GROUP_QOS);
  BIT (ROUTING_INTERFACE_CHANGE);
  BIT (ADDRESS_LIST_CHANGE);
  
#undef BIT

  if ((mask & ~checked_bits) != 0)
	  bufp += sprintf (bufp, "|%#x", mask & ~checked_bits);
  
  return g_quark_to_string (g_quark_from_string (buf));
}

static const char *
condition_to_string (GIOCondition condition)
{
  char buf[100];
  int checked_bits = 0;
  char *bufp = buf;

  if (condition == 0)
    return "";

#define BIT(n) checked_bits |= G_IO_##n; if (condition & G_IO_##n) bufp += sprintf (bufp, "%s" #n, (bufp>buf ? "|" : ""))

  BIT (IN);
  BIT (OUT);
  BIT (PRI);
  BIT (ERR);
  BIT (HUP);
  BIT (NVAL);
  
#undef BIT

  if ((condition & ~checked_bits) != 0)
	  bufp += sprintf (bufp, "|%#x", condition & ~checked_bits);
  
  return g_quark_to_string (g_quark_from_string (buf));
}

224 225
static gboolean
g_io_win32_get_debug_flag (void)
226
{
227 228 229 230 231 232 233 234 235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268 269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 288 289 290 291 292 293 294 295 296 297 298 299 300 301 302 303 304 305 306 307 308 309 310 311 312 313 314 315 316 317 318 319 320 321 322 323 324 325 326 327 328 329 330 331 332 333 334 335 336 337 338 339 340 341 342 343 344
  return (getenv ("G_IO_WIN32_DEBUG") != NULL);
}

static char *
winsock_error_message (int number)
{
  static char unk[100];

  switch (number) {
  case WSAEINTR:
    return "Interrupted function call";
  case WSAEACCES:
    return "Permission denied";
  case WSAEFAULT:
    return "Bad address";
  case WSAEINVAL:
    return "Invalid argument";
  case WSAEMFILE:
    return "Too many open sockets";
  case WSAEWOULDBLOCK:
    return "Resource temporarily unavailable";
  case WSAEINPROGRESS:
    return "Operation now in progress";
  case WSAEALREADY:
    return "Operation already in progress";
  case WSAENOTSOCK:
    return "Socket operation on nonsocket";
  case WSAEDESTADDRREQ:
    return "Destination address required";
  case WSAEMSGSIZE:
    return "Message too long";
  case WSAEPROTOTYPE:
    return "Protocol wrong type for socket";
  case WSAENOPROTOOPT:
    return "Bad protocol option";
  case WSAEPROTONOSUPPORT:
    return "Protocol not supported";
  case WSAESOCKTNOSUPPORT:
    return "Socket type not supported";
  case WSAEOPNOTSUPP:
    return "Operation not supported on transport endpoint";
  case WSAEPFNOSUPPORT:
    return "Protocol family not supported";
  case WSAEAFNOSUPPORT:
    return "Address family not supported by protocol family";
  case WSAEADDRINUSE:
    return "Address already in use";
  case WSAEADDRNOTAVAIL:
    return "Address not available";
  case WSAENETDOWN:
    return "Network interface is not configured";
  case WSAENETUNREACH:
    return "Network is unreachable";
  case WSAENETRESET:
    return "Network dropped connection on reset";
  case WSAECONNABORTED:
    return "Software caused connection abort";
  case WSAECONNRESET:
    return "Connection reset by peer";
  case WSAENOBUFS:
    return "No buffer space available";
  case WSAEISCONN:
    return "Socket is already connected";
  case WSAENOTCONN:
    return "Socket is not connected";
  case WSAESHUTDOWN:
    return "Can't send after socket shutdown";
  case WSAETIMEDOUT:
    return "Connection timed out";
  case WSAECONNREFUSED:
    return "Connection refused";
  case WSAEHOSTDOWN:
    return "Host is down";
  case WSAEHOSTUNREACH:
    return "Host is unreachable";
  case WSAEPROCLIM:
    return "Too many processes";
  case WSASYSNOTREADY:
    return "Network subsystem is unavailable";
  case WSAVERNOTSUPPORTED:
    return "Winsock.dll version out of range";
  case WSANOTINITIALISED:
    return "Successful WSAStartup not yet performed";
  case WSAEDISCON:
    return "Graceful shutdown in progress";
  case WSATYPE_NOT_FOUND:
    return "Class type not found";
  case WSAHOST_NOT_FOUND:
    return "Host not found";
  case WSATRY_AGAIN:
    return "Nonauthoritative host not found";
  case WSANO_RECOVERY:
    return "This is a nonrecoverable error";
  case WSANO_DATA:
    return "Valid name, no data record of requested type";
  case WSA_INVALID_HANDLE:
    return "Specified event object handle is invalid";
  case WSA_INVALID_PARAMETER:
    return "One or more parameters are invalid";
  case WSA_IO_INCOMPLETE:
    return "Overlapped I/O event object not in signaled state";
  case WSA_NOT_ENOUGH_MEMORY:
    return "Insufficient memory available";
  case WSA_OPERATION_ABORTED:
    return "Overlapped operation aborted";
  case WSAEINVALIDPROCTABLE:
    return "Invalid procedure table from service provider";
  case WSAEINVALIDPROVIDER:
    return "Invalid service provider version number";
  case WSAEPROVIDERFAILEDINIT:
    return "Unable to initialize a service provider";
  case WSASYSCALLFAILURE:
    return "System call failure";
  default:
    sprintf (unk, "Unknown WinSock error %d", number);
    return unk;
  }
}
345 346 347 348 349

static void
g_io_channel_win32_init (GIOWin32Channel *channel)
{
  channel->debug = g_io_win32_get_debug_flag ();
350 351
  channel->buffer = NULL;
  channel->running = FALSE;
352
  channel->needs_close = FALSE;
353
  channel->thread_id = 0;
354
  channel->data_avail_event = NULL;
355
  channel->revents = 0;
356
  channel->space_avail_event = NULL;
357 358 359 360
  channel->event_mask = 0;
  channel->last_events = 0;
  channel->event = 0;
  channel->write_would_have_blocked = FALSE;
361
  InitializeCriticalSection (&channel->mutex);
362 363
}

364 365 366 367 368
static void
create_events (GIOWin32Channel *channel)
{
  SECURITY_ATTRIBUTES sec_attrs;
  
369
  sec_attrs.nLength = sizeof (SECURITY_ATTRIBUTES);
370 371
  sec_attrs.lpSecurityDescriptor = NULL;
  sec_attrs.bInheritHandle = FALSE;
372

373 374 375 376
  /* The data available event is manual reset, the space available event
   * is automatic reset.
   */
  if (!(channel->data_avail_event = CreateEvent (&sec_attrs, TRUE, FALSE, NULL))
377
      || !(channel->space_avail_event = CreateEvent (&sec_attrs, FALSE, FALSE, NULL)))
378
    {
379 380 381
      gchar *emsg = g_win32_error_message (GetLastError ());
      g_error ("Error creating event: %s", emsg);
      g_free (emsg);
382 383
    }
}
384

385
static unsigned __stdcall
386
read_thread (void *parameter)
387 388 389 390
{
  GIOWin32Channel *channel = parameter;
  guchar *buffer;
  guint nbytes;
391

392
  g_io_channel_ref ((GIOChannel *)channel);
393

394
  if (channel->debug)
395
    g_print ("read_thread %#x: start fd=%d, data_avail=%#x space_avail=%#x\n",
396 397 398 399
	     channel->thread_id,
	     channel->fd,
	     (guint) channel->data_avail_event,
	     (guint) channel->space_avail_event);
400 401

  channel->direction = 0;
402 403 404
  channel->buffer = g_malloc (BUFFER_SIZE);
  channel->rdp = channel->wrp = 0;
  channel->running = TRUE;
405

406 407
  SetEvent (channel->space_avail_event);
  
408
  LOCK (channel->mutex);
409 410
  while (channel->running)
    {
411
      if (channel->debug)
412
	g_print ("read_thread %#x: rdp=%d, wrp=%d\n",
413
		 channel->thread_id, channel->rdp, channel->wrp);
414 415 416
      if ((channel->wrp + 1) % BUFFER_SIZE == channel->rdp)
	{
	  /* Buffer is full */
417
	  if (channel->debug)
418
	    g_print ("read_thread %#x: resetting space_avail\n",
419
		     channel->thread_id);
420
	  ResetEvent (channel->space_avail_event);
421
	  if (channel->debug)
422 423
	    g_print ("read_thread %#x: waiting for space\n",
		     channel->thread_id);
424 425 426
	  UNLOCK (channel->mutex);
	  WaitForSingleObject (channel->space_avail_event, INFINITE);
	  LOCK (channel->mutex);
427
	  if (channel->debug)
428
	    g_print ("read_thread %#x: rdp=%d, wrp=%d\n",
429
		     channel->thread_id, channel->rdp, channel->wrp);
430 431 432 433 434 435 436 437 438
	}
      
      buffer = channel->buffer + channel->wrp;
      
      /* Always leave at least one byte unused gap to be able to
       * distinguish between the full and empty condition...
       */
      nbytes = MIN ((channel->rdp + BUFFER_SIZE - channel->wrp - 1) % BUFFER_SIZE,
		    BUFFER_SIZE - channel->wrp);
439

440
      if (channel->debug)
441
	g_print ("read_thread %#x: calling read() for %d bytes\n",
442 443
		 channel->thread_id, nbytes);

444
      UNLOCK (channel->mutex);
445

446
      nbytes = read (channel->fd, buffer, nbytes);
447 448
      
      LOCK (channel->mutex);
449

450 451 452 453 454 455
      channel->revents = G_IO_IN;
      if (nbytes == 0)
	channel->revents |= G_IO_HUP;
      else if (nbytes < 0)
	channel->revents |= G_IO_ERR;

456
      if (channel->debug)
457
	g_print ("read_thread %#x: read() returned %d, rdp=%d, wrp=%d\n",
458
		 channel->thread_id, nbytes, channel->rdp, channel->wrp);
459 460 461 462

      if (nbytes <= 0)
	break;

463
      channel->wrp = (channel->wrp + nbytes) % BUFFER_SIZE;
464
      if (channel->debug)
465
	g_print ("read_thread %#x: rdp=%d, wrp=%d, setting data_avail\n",
466
		 channel->thread_id, channel->rdp, channel->wrp);
467 468 469 470
      SetEvent (channel->data_avail_event);
    }
  
  channel->running = FALSE;
471 472 473
  if (channel->needs_close)
    {
      if (channel->debug)
474
	g_print ("read_thread %#x: channel fd %d needs closing\n",
475
		 channel->thread_id, channel->fd);
476
      close (channel->fd);
477 478 479
      channel->fd = -1;
    }

480 481 482
  if (channel->debug)
    g_print ("read_thread %#x: EOF, rdp=%d, wrp=%d, setting data_avail\n",
	     channel->thread_id, channel->rdp, channel->wrp);
483 484 485
  SetEvent (channel->data_avail_event);
  UNLOCK (channel->mutex);
  
486
  g_io_channel_unref ((GIOChannel *)channel);
487
  
488 489 490
  /* No need to call _endthreadex(), the actual thread starter routine
   * in MSVCRT (see crt/src/threadex.c:_threadstartex) calls
   * _endthreadex() for us.
491
   */
492

493 494
  return 0;
}
495

496 497 498 499 500 501 502 503 504 505 506 507 508 509 510 511 512 513 514 515 516 517 518 519 520 521 522 523 524 525 526 527 528 529 530 531 532 533 534 535 536 537 538 539 540 541 542 543 544 545 546 547 548 549 550 551 552 553 554 555 556 557 558 559 560 561 562 563 564 565 566 567 568 569 570 571 572 573 574 575 576 577 578 579 580 581 582 583 584 585 586 587 588 589 590 591 592 593 594 595 596 597 598 599 600 601 602 603 604 605 606
static unsigned __stdcall
write_thread (void *parameter)
{
  GIOWin32Channel *channel = parameter;
  guchar *buffer;
  guint nbytes;

  g_io_channel_ref ((GIOChannel *)channel);

  if (channel->debug)
    g_print ("write_thread %#x: start fd=%d, data_avail=%#x space_avail=%#x\n",
	     channel->thread_id,
	     channel->fd,
	     (guint) channel->data_avail_event,
	     (guint) channel->space_avail_event);
  
  channel->direction = 1;
  channel->buffer = g_malloc (BUFFER_SIZE);
  channel->rdp = channel->wrp = 0;
  channel->running = TRUE;

  SetEvent (channel->space_avail_event);

  /* We use the same event objects as for a reader thread, but with
   * reversed meaning. So, space_avail is used if data is available
   * for writing, and data_avail is used if space is available in the
   * write buffer.
   */

  LOCK (channel->mutex);
  while (channel->running || channel->rdp != channel->wrp)
    {
      if (channel->debug)
	g_print ("write_thread %#x: rdp=%d, wrp=%d\n",
		 channel->thread_id, channel->rdp, channel->wrp);
      if (channel->wrp == channel->rdp)
	{
	  /* Buffer is empty. */
	  if (channel->debug)
	    g_print ("write_thread %#x: resetting space_avail\n",
		     channel->thread_id);
	  ResetEvent (channel->space_avail_event);
	  if (channel->debug)
	    g_print ("write_thread %#x: waiting for data\n",
		     channel->thread_id);
	  channel->revents = G_IO_OUT;
	  SetEvent (channel->data_avail_event);
	  UNLOCK (channel->mutex);
	  WaitForSingleObject (channel->space_avail_event, INFINITE);

	  LOCK (channel->mutex);
	  if (channel->rdp == channel->wrp)
	    break;

	  if (channel->debug)
	    g_print ("write_thread %#x: rdp=%d, wrp=%d\n",
		     channel->thread_id, channel->rdp, channel->wrp);
	}
      
      buffer = channel->buffer + channel->rdp;
      if (channel->rdp < channel->wrp)
	nbytes = channel->wrp - channel->rdp;
      else
	nbytes = BUFFER_SIZE - channel->rdp;

      if (channel->debug)
	g_print ("write_thread %#x: calling write() for %d bytes\n",
		 channel->thread_id, nbytes);

      UNLOCK (channel->mutex);
      nbytes = write (channel->fd, buffer, nbytes);
      LOCK (channel->mutex);

      if (channel->debug)
	g_print ("write_thread %#x: write(%i) returned %d, rdp=%d, wrp=%d\n",
		 channel->thread_id, channel->fd, nbytes, channel->rdp, channel->wrp);

      channel->revents = 0;
      if (nbytes > 0)
	channel->revents |= G_IO_OUT;
      else if (nbytes <= 0)
	channel->revents |= G_IO_ERR;

      channel->rdp = (channel->rdp + nbytes) % BUFFER_SIZE;

      if (nbytes <= 0)
	break;

      if (channel->debug)
	g_print ("write_thread: setting data_avail for thread %#x\n",
		 channel->thread_id);
      SetEvent (channel->data_avail_event);
    }
  
  channel->running = FALSE;
  if (channel->needs_close)
    {
      if (channel->debug)
	g_print ("write_thread %#x: channel fd %d needs closing\n",
		 channel->thread_id, channel->fd);
      close (channel->fd);
      channel->fd = -1;
    }

  UNLOCK (channel->mutex);
  
  g_io_channel_unref ((GIOChannel *)channel);
  
  return 0;
}

607
static void
608 609 610
create_thread (GIOWin32Channel     *channel,
	       GIOCondition         condition,
	       unsigned (__stdcall *thread) (void *parameter))
611
{
612 613 614 615 616
  HANDLE thread_handle;

  thread_handle = (HANDLE) _beginthreadex (NULL, 0, thread, channel, 0,
					   &channel->thread_id);
  if (thread_handle == 0)
617
    g_warning (G_STRLOC ": Error creating reader thread: %s",
Owen Taylor's avatar
Owen Taylor committed
618
	       g_strerror (errno));
619 620 621 622
  else if (!CloseHandle (thread_handle))
    g_warning (G_STRLOC ": Error closing thread handle: %s\n",
	       g_win32_error_message (GetLastError ()));

623 624
  WaitForSingleObject (channel->space_avail_event, INFINITE);
}
625

626
static GIOStatus
627 628
buffer_read (GIOWin32Channel *channel,
	     guchar          *dest,
629 630 631
	     gsize            count,
	     gsize           *bytes_read,
	     GError         **err)
632
{
633 634 635 636
  guint nbytes;
  guint left = count;
  
  LOCK (channel->mutex);
637 638 639
  if (channel->debug)
    g_print ("reading from thread %#x %d bytes, rdp=%d, wrp=%d\n",
	     channel->thread_id, count, channel->rdp, channel->wrp);
640
  
641
  if (channel->wrp == channel->rdp)
642 643
    {
      UNLOCK (channel->mutex);
644 645 646
      if (channel->debug)
	g_print ("waiting for data from thread %#x\n", channel->thread_id);
      WaitForSingleObject (channel->data_avail_event, INFINITE);
647 648
      if (channel->debug)
	g_print ("done waiting for data from thread %#x\n", channel->thread_id);
649
      LOCK (channel->mutex);
650
      if (channel->wrp == channel->rdp && !channel->running)
651
	{
652 653
	  if (channel->debug)
	    g_print ("wrp==rdp, !running\n");
654
	  UNLOCK (channel->mutex);
655
          *bytes_read = 0;
656
	  return G_IO_STATUS_EOF;
657
	}
658
    }
659 660 661 662 663 664 665 666 667 668 669 670 671 672 673 674
  
  if (channel->rdp < channel->wrp)
    nbytes = channel->wrp - channel->rdp;
  else
    nbytes = BUFFER_SIZE - channel->rdp;
  UNLOCK (channel->mutex);
  nbytes = MIN (left, nbytes);
  if (channel->debug)
    g_print ("moving %d bytes from thread %#x\n",
	     nbytes, channel->thread_id);
  memcpy (dest, channel->buffer + channel->rdp, nbytes);
  dest += nbytes;
  left -= nbytes;
  LOCK (channel->mutex);
  channel->rdp = (channel->rdp + nbytes) % BUFFER_SIZE;
  if (channel->debug)
675
    g_print ("setting space_avail for thread %#x\n", channel->thread_id);
676 677 678 679
  SetEvent (channel->space_avail_event);
  if (channel->debug)
    g_print ("for thread %#x: rdp=%d, wrp=%d\n",
	     channel->thread_id, channel->rdp, channel->wrp);
680
  if (channel->running && channel->wrp == channel->rdp)
681 682
    {
      if (channel->debug)
683
	g_print ("resetting data_avail of thread %#x\n",
684 685 686
		 channel->thread_id);
      ResetEvent (channel->data_avail_event);
    };
687 688
  UNLOCK (channel->mutex);
  
689 690 691
  /* We have no way to indicate any errors form the actual
   * read() or recv() call in the reader thread. Should we have?
   */
692 693
  *bytes_read = count - left;
  return (*bytes_read > 0) ? G_IO_STATUS_NORMAL : G_IO_STATUS_EOF;
694
}
695

696 697 698 699 700 701 702 703 704 705 706 707 708 709 710 711 712 713 714 715 716 717 718 719 720 721 722 723 724 725 726 727 728 729 730 731 732 733 734 735 736 737 738 739 740 741 742 743 744 745 746 747 748 749 750 751 752 753 754 755 756 757 758 759 760 761 762 763 764 765 766 767

static GIOStatus
buffer_write (GIOWin32Channel *channel,
	      const guchar    *dest,
	      gsize            count,
	      gsize           *bytes_written,
	      GError         **err)
{
  guint nbytes;
  guint left = count;
  
  LOCK (channel->mutex);
  if (channel->debug)
    g_print ("buffer_write: writing to thread %#x %d bytes, rdp=%d, wrp=%d\n",
	     channel->thread_id, count, channel->rdp, channel->wrp);
  
  if ((channel->wrp + 1) % BUFFER_SIZE == channel->rdp)
    {
      /* Buffer is full */
      if (channel->debug)
	g_print ("buffer_write: tid %#x: resetting data_avail\n",
		 channel->thread_id);
      ResetEvent (channel->data_avail_event);
      if (channel->debug)
	g_print ("buffer_write: tid %#x: waiting for space\n",
		 channel->thread_id);
      UNLOCK (channel->mutex);
      WaitForSingleObject (channel->data_avail_event, INFINITE);
      LOCK (channel->mutex);
      if (channel->debug)
	g_print ("buffer_write: tid %#x: rdp=%d, wrp=%d\n",
		 channel->thread_id, channel->rdp, channel->wrp);
    }
   
  nbytes = MIN ((channel->rdp + BUFFER_SIZE - channel->wrp - 1) % BUFFER_SIZE,
		BUFFER_SIZE - channel->wrp);

  UNLOCK (channel->mutex);
  nbytes = MIN (left, nbytes);
  if (channel->debug)
    g_print ("buffer_write: tid %#x: writing %d bytes\n",
	     channel->thread_id, nbytes);
  memcpy (channel->buffer + channel->wrp, dest, nbytes);
  dest += nbytes;
  left -= nbytes;
  LOCK (channel->mutex);

  channel->wrp = (channel->wrp + nbytes) % BUFFER_SIZE;
  if (channel->debug)
    g_print ("buffer_write: tid %#x: rdp=%d, wrp=%d, setting space_avail\n",
	     channel->thread_id, channel->rdp, channel->wrp);
  SetEvent (channel->space_avail_event);

  if ((channel->wrp + 1) % BUFFER_SIZE == channel->rdp)
    {
      /* Buffer is full */
      if (channel->debug)
	g_print ("buffer_write: tid %#x: resetting data_avail\n",
		 channel->thread_id);
      ResetEvent (channel->data_avail_event);
    }

  UNLOCK (channel->mutex);
  
  /* We have no way to indicate any errors form the actual
   * write() call in the writer thread. Should we have?
   */
  *bytes_written = count - left;
  return (*bytes_written > 0) ? G_IO_STATUS_NORMAL : G_IO_STATUS_EOF;
}


768
static gboolean
769 770
g_io_win32_prepare (GSource *source,
		    gint    *timeout)
771
{
772
  GIOWin32Watch *watch = (GIOWin32Watch *)source;
773
  GIOCondition buffer_condition = g_io_channel_get_buffer_condition (watch->channel);
774
  GIOWin32Channel *channel = (GIOWin32Channel *)watch->channel;
775
  int event_mask;
776
  
777
  *timeout = -1;
778
  
779 780 781
  if (channel->debug)
    g_print ("g_io_win32_prepare: source=%p channel=%p", source, channel);

782
  switch (channel->type)
783
    {
784
    case G_IO_WIN32_WINDOWS_MESSAGES:
785 786 787 788
      if (channel->debug)
	g_print (" MSG");
      break;

789
    case G_IO_WIN32_CONSOLE:
790 791
      if (channel->debug)
	g_print (" CON");
792 793 794 795
      break;

    case G_IO_WIN32_FILE_DESC:
      if (channel->debug)
796 797
	g_print (" FD thread=%#x buffer_condition:{%s}"
		 "\n  watch->pollfd.events:{%s} watch->pollfd.revents:{%s} channel->revents:{%s}",
798 799 800 801 802
		 channel->thread_id, condition_to_string (buffer_condition),
		 condition_to_string (watch->pollfd.events),
		 condition_to_string (watch->pollfd.revents),
		 condition_to_string (channel->revents));
      
803
      LOCK (channel->mutex);
804
      if (channel->running)
805
	{
806 807 808
	  if (channel->direction == 0 && channel->wrp == channel->rdp)
	    {
	      if (channel->debug)
809
		g_print ("\n  setting revents=0");
810 811
	      channel->revents = 0;
	    }
812
	}
813 814 815 816 817 818
      else
	{
	  if (channel->direction == 1
	      && (channel->wrp + 1) % BUFFER_SIZE == channel->rdp)
	    {
	      if (channel->debug)
819
		g_print ("\n setting revents=0");
820 821 822
	      channel->revents = 0;
	    }
	}	  
823
      UNLOCK (channel->mutex);
824
      break;
825

826
    case G_IO_WIN32_SOCKET:
827 828
      if (channel->debug)
	g_print (" SOCK");
829 830 831 832 833
      event_mask = 0;
      if (watch->condition & G_IO_IN)
	event_mask |= (FD_READ | FD_ACCEPT);
      if (watch->condition & G_IO_OUT)
	event_mask |= (FD_WRITE | FD_CONNECT);
834
      event_mask |= FD_CLOSE;
835 836 837 838

      if (channel->event_mask != event_mask /* || channel->event != watch->pollfd.fd*/)
	{
	  if (channel->debug)
839
	    g_print ("\n  WSAEventSelect(%d,%#x,{%s})",
840 841 842 843 844 845 846 847 848
		     channel->fd, watch->pollfd.fd,
		     event_mask_to_string (event_mask));
	  if (WSAEventSelect (channel->fd, (HANDLE) watch->pollfd.fd,
			      event_mask) == SOCKET_ERROR)
	    ;			/* What? */
	  channel->event_mask = event_mask;
#if 0
	  channel->event = watch->pollfd.fd;
#endif
849 850
	  if (channel->debug)
	    g_print ("\n  setting last_events=0");
851 852 853 854 855 856 857 858
	  channel->last_events = 0;
	}
      break;

    default:
      g_assert_not_reached ();
      abort ();
    }
859 860 861
  if (channel->debug)
    g_print ("\n");

862
  return ((watch->condition & buffer_condition) == watch->condition);
863 864
}

865
static gboolean
866
g_io_win32_check (GSource *source)
867
{
868
  MSG msg;
869
  GIOWin32Watch *watch = (GIOWin32Watch *)source;
870
  GIOWin32Channel *channel = (GIOWin32Channel *)watch->channel;
871
  GIOCondition buffer_condition = g_io_channel_get_buffer_condition (watch->channel);
872
  WSANETWORKEVENTS events;
873

874 875 876
  if (channel->debug)
    g_print ("g_io_win32_check: source=%p channel=%p", source, channel);

877
  switch (channel->type)
878
    {
879
    case G_IO_WIN32_WINDOWS_MESSAGES:
880 881
      if (channel->debug)
	g_print (" MSG\n");
882
      return (PeekMessage (&msg, channel->hwnd, 0, 0, PM_NOREMOVE));
883 884

    case G_IO_WIN32_FILE_DESC:
885
      if (channel->debug)
886
	g_print (" FD thread=%#x buffer_condition=%s\n"
887 888 889 890 891 892 893 894 895 896
		 "  watch->pollfd.events={%s} watch->pollfd.revents={%s} channel->revents={%s}\n",
		 channel->thread_id, condition_to_string (buffer_condition),
		 condition_to_string (watch->pollfd.events),
		 condition_to_string (watch->pollfd.revents),
		 condition_to_string (channel->revents));
      
      watch->pollfd.revents = (watch->pollfd.events & channel->revents);

      return ((watch->pollfd.revents | buffer_condition) & watch->condition);

897
    case G_IO_WIN32_CONSOLE:
898 899
      if (channel->debug)
	g_print (" CON\n");
900 901 902 903 904 905 906 907 908 909 910 911 912 913 914 915 916 917 918 919 920 921
      if (watch->channel->is_writeable)
	return TRUE;
      else if (watch->channel->is_readable)
        {
	  INPUT_RECORD buffer;
	  DWORD n;
	  if (PeekConsoleInput ((HANDLE) watch->pollfd.fd, &buffer, 1, &n) &&
	      n == 1)
	    {
	      /* _kbhit() does quite complex processing to find out
	       * whether at least one of the key events pending corresponds
	       * to a "real" character that can be read.
	       */
	      if (_kbhit ())
		return TRUE;
	      
	      /* Discard all other kinds of events */
	      ReadConsoleInput ((HANDLE) watch->pollfd.fd, &buffer, 1, &n);
	    }
        }
      return FALSE;

922
    case G_IO_WIN32_SOCKET:
923 924
      if (channel->debug)
	g_print (" SOCK");
925 926 927
      if (channel->last_events & FD_WRITE)
	{
	  if (channel->debug)
928
	    g_print (" sock=%d event=%#x last_events has FD_WRITE",
929 930 931 932 933 934 935
		     channel->fd, watch->pollfd.fd);
	}
      else
	{
	  WSAEnumNetworkEvents (channel->fd, 0, &events);

	  if (channel->debug)
936 937
	    g_print ("\n  revents={%s} condition={%s}"
		     "\n  WSAEnumNetworkEvents(%d,0) sets events={%s}",
938 939
		     condition_to_string (watch->pollfd.revents),
		     condition_to_string (watch->condition),
940
		     channel->fd, 
941 942 943 944 945 946 947 948
		     event_mask_to_string (events.lNetworkEvents));
	  
	  if (watch->pollfd.revents != 0 &&
	      events.lNetworkEvents == 0 &&
	      !(channel->event_mask & FD_WRITE))
	    {
	      channel->event_mask = 0;
	      if (channel->debug)
949
		g_print ("\n  WSAEventSelect(%d,%#x,{})",
950 951 952
			 channel->fd, watch->pollfd.fd);
	      WSAEventSelect (channel->fd, (HANDLE) watch->pollfd.fd, 0);
	      if (channel->debug)
953
		g_print ("  ResetEvent(%#x)",
954 955 956 957 958 959 960 961
			 watch->pollfd.fd);
	      ResetEvent ((HANDLE) watch->pollfd.fd);
	    }
	  channel->last_events = events.lNetworkEvents;
	}
      watch->pollfd.revents = 0;
      if (channel->last_events & (FD_READ | FD_ACCEPT))
	watch->pollfd.revents |= G_IO_IN;
962
      if (channel->last_events & FD_WRITE)
963
	watch->pollfd.revents |= G_IO_OUT;
964 965 966 967 968 969 970 971 972 973 974 975 976 977 978
      else
	{
	  /* We have called WSAEnumNetworkEvents() above but it didn't
	   * set FD_WRITE.
	   */
	  if (events.lNetworkEvents & FD_CONNECT)
	    {
	      if (events.iErrorCode[FD_CONNECT_BIT] == 0)
		watch->pollfd.revents |= G_IO_OUT;
	      else
		watch->pollfd.revents |= (G_IO_HUP | G_IO_ERR);
	    }
	  if (watch->pollfd.revents == 0 && (channel->last_events & (FD_CLOSE)))
	    watch->pollfd.revents |= G_IO_HUP;
	}
979

980 981 982 983 984 985 986 987
      /* Regardless of WSAEnumNetworkEvents() result, if watching for
       * writability, unless last write would have blocked set
       * G_IO_OUT. But never set both G_IO_OUT and G_IO_HUP.
       */
      if (!(watch->pollfd.revents & G_IO_HUP) &&
	  !channel->write_would_have_blocked &&
	  (channel->event_mask & FD_WRITE))
	watch->pollfd.revents |= G_IO_OUT;
988

989 990 991 992 993
      if (channel->debug)
	g_print ("\n  revents={%s} retval={%s}\n",
		 condition_to_string (watch->pollfd.revents),
		 condition_to_string ((watch->pollfd.revents | buffer_condition) & watch->condition));

994 995 996 997 998 999
      return ((watch->pollfd.revents | buffer_condition) & watch->condition);

    default:
      g_assert_not_reached ();
      abort ();
    }
1000 1001 1002
}

static gboolean
1003 1004 1005
g_io_win32_dispatch (GSource     *source,
		     GSourceFunc  callback,
		     gpointer     user_data)
1006
{
1007
  GIOFunc func = (GIOFunc)callback;
1008
  GIOWin32Watch *watch = (GIOWin32Watch *)source;
1009
  GIOWin32Channel *channel = (GIOWin32Channel *)watch->channel;
1010
  GIOCondition buffer_condition = g_io_channel_get_buffer_condition (watch->channel);
1011
  
1012 1013
  if (!func)
    {
1014
      g_warning (G_STRLOC ": GIOWin32Watch dispatched without callback\n"
1015 1016 1017 1018
		 "You must call g_source_connect().");
      return FALSE;
    }
  
1019 1020 1021 1022 1023 1024
  if (channel->debug)
    g_print ("g_io_win32_dispatch: pollfd.revents=%s condition=%s result=%s\n",
	     condition_to_string (watch->pollfd.revents),
	     condition_to_string (watch->condition),
	     condition_to_string ((watch->pollfd.revents | buffer_condition) & watch->condition));

1025
  return (*func) (watch->channel,
1026
		  (watch->pollfd.revents | buffer_condition) & watch->condition,
1027
		  user_data);
1028 1029 1030
}

static void
1031
g_io_win32_finalize (GSource *source)
1032
{
1033
  GIOWin32Watch *watch = (GIOWin32Watch *)source;
1034
  GIOWin32Channel *channel = (GIOWin32Channel *)watch->channel;
1035
  
1036 1037 1038
  if (channel->debug)
    g_print ("g_io_win32_finalize: source=%p channel=%p", source, channel);

1039 1040 1041
  switch (channel->type)
    {
    case G_IO_WIN32_WINDOWS_MESSAGES:
1042 1043 1044 1045
      if (channel->debug)
	g_print (" MSG");
      break;

1046
    case G_IO_WIN32_CONSOLE:
1047 1048
      if (channel->debug)
	g_print (" CON");
1049
      break;
1050

1051 1052
    case G_IO_WIN32_FILE_DESC:
      if (channel->debug)
1053
	g_print (" FD thread=%#x", channel->thread_id);
1054 1055 1056 1057
      break;

    case G_IO_WIN32_SOCKET:
      if (channel->debug)
1058
	g_print (" SOCK sock=%d", channel->fd);
1059 1060 1061 1062 1063 1064 1065 1066 1067 1068 1069
#if 0
      CloseHandle ((HANDLE) watch->pollfd.fd);
      channel->event = 0;
      channel->event_mask = 0;
#endif
      break;

    default:
      g_assert_not_reached ();
      abort ();
    }
1070 1071
  if (channel->debug)
    g_print ("\n");
1072
  g_io_channel_unref (watch->channel);
1073 1074
}

1075
GSourceFuncs g_io_watch_funcs = {
1076 1077 1078
  g_io_win32_prepare,
  g_io_win32_check,
  g_io_win32_dispatch,
1079
  g_io_win32_finalize
1080
};
1081

1082
static GIOStatus
1083 1084
g_io_win32_msg_read (GIOChannel *channel,
		     gchar      *buf,
1085 1086 1087
		     gsize       count,
		     gsize      *bytes_read,
		     GError    **err)
1088
{
1089
  GIOWin32Channel *win32_channel = (GIOWin32Channel *)channel;
1090 1091
  MSG msg;               /* In case of alignment problems */
  
1092
  if (count < sizeof (MSG))
1093
    {
1094 1095
      g_set_error (err, G_IO_CHANNEL_ERROR, G_IO_CHANNEL_ERROR_INVAL,
		   "Incorrect message size"); /* Informative enough error message? */
1096 1097
      return G_IO_STATUS_ERROR;
    }
1098
  
1099
  if (win32_channel->debug)
1100 1101
    g_print ("g_io_win32_msg_read: channel=%p hwnd=%#x\n",
	     channel, (guint) win32_channel->hwnd);
1102
  if (!PeekMessage (&msg, win32_channel->hwnd, 0, 0, PM_REMOVE))
1103
    return G_IO_STATUS_AGAIN;
1104

1105 1106
  memmove (buf, &msg, sizeof (MSG));
  *bytes_read = sizeof (MSG);
1107

1108
  return G_IO_STATUS_NORMAL;
1109
}
1110

1111 1112 1113 1114 1115 1116
static GIOStatus
g_io_win32_msg_write (GIOChannel  *channel,
		      const gchar *buf,
		      gsize        count,
		      gsize       *bytes_written,
		      GError     **err)
1117
{
1118
  GIOWin32Channel *win32_channel = (GIOWin32Channel *)channel;
1119
  MSG msg;
1120
  
1121
  if (count != sizeof (MSG))
1122
    {
1123 1124
      g_set_error (err, G_IO_CHANNEL_ERROR, G_IO_CHANNEL_ERROR_INVAL,
		   "Incorrect message size"); /* Informative enough error message? */
1125 1126
      return G_IO_STATUS_ERROR;
    }
1127
  
1128 1129 1130
  /* In case of alignment problems */
  memmove (&msg, buf, sizeof (MSG));
  if (!PostMessage (win32_channel->hwnd, msg.message, msg.wParam, msg.lParam))
1131
    {
1132
      gchar *emsg = g_win32_error_message (GetLastError ());
1133
      g_set_error (err, G_IO_CHANNEL_ERROR, G_IO_CHANNEL_ERROR_FAILED, emsg);
1134
      g_free (emsg);
1135 1136 1137
      return G_IO_STATUS_ERROR;
    }

1138
  *bytes_written = sizeof (MSG);
1139 1140

  return G_IO_STATUS_NORMAL;
1141 1142
}

1143 1144 1145
static GIOStatus
g_io_win32_msg_close (GIOChannel *channel,
		      GError    **err)
1146 1147
{
  /* Nothing to be done. Or should we set hwnd to some invalid value? */
1148 1149

  return G_IO_STATUS_NORMAL;
1150 1151
}

1152
static void
1153 1154
g_io_win32_free (GIOChannel *channel)
{
1155
  GIOWin32Channel *win32_channel = (GIOWin32Channel *)channel;
1156
  
1157
  if (win32_channel->debug)
1158
    g_print ("g_io_win32_free channel=%p fd=%d\n", channel, win32_channel->fd);
1159

1160 1161 1162 1163
  if (win32_channel->data_avail_event)
    CloseHandle (win32_channel->data_avail_event);
  if (win32_channel->space_avail_event)
    CloseHandle (win32_channel->space_avail_event);
1164 1165
  if (win32_channel->type == G_IO_WIN32_SOCKET)
    WSAEventSelect (win32_channel->fd, NULL, 0);
1166
  DeleteCriticalSection (&win32_channel->mutex);
1167

1168
  g_free (win32_channel->buffer);
1169 1170 1171
  g_free (win32_channel);
}

1172
static GSource *
1173 1174
g_io_win32_msg_create_watch (GIOChannel   *channel,
			     GIOCondition  condition)
1175
{
1176 1177 1178
  GIOWin32Watch *watch;
  GSource *source;

1179
  source = g_source_new (&g_io_watch_funcs, sizeof (GIOWin32Watch));
1180
  watch = (GIOWin32Watch *)source;
1181 1182 1183
  
  watch->channel = channel;
  g_io_channel_ref (channel);
1184
  
1185
  watch->condition = condition;
1186
  
1187 1188
  watch->pollfd.fd = G_WIN32_MSG_HANDLE;
  watch->pollfd.events = condition;
1189
  
1190
  g_source_add_poll (source, &watch->pollfd);
1191
  
1192
  return source;
1193 1194
}

1195
static GIOStatus
1196 1197 1198 1199 1200
g_io_win32_fd_and_console_read (GIOChannel *channel,
				gchar      *buf,
				gsize       count,
				gsize      *bytes_read,
				GError    **err)
1201
{
1202
  GIOWin32Channel *win32_channel = (GIOWin32Channel *)channel;
1203
  gint result;
1204
  
1205
  if (win32_channel->debug)
1206
    g_print ("g_io_win32_fd_read: fd=%d count=%d\n",
1207 1208
	     win32_channel->fd, count);
  
1209 1210
  if (win32_channel->thread_id)
    {
1211
      return buffer_read (win32_channel, buf, count, bytes_read, err);
1212
    }
1213 1214

  result = read (win32_channel->fd, buf, count);
1215

1216
  if (win32_channel->debug)
1217
    g_print ("g_io_win32_fd_read: read() => %d\n", result);
1218

1219 1220 1221
  if (result < 0)
    {
      *bytes_read = 0;
1222

1223
      switch (errno)
1224 1225
        {
#ifdef EAGAIN
1226 1227
	case EAGAIN:
	  return G_IO_STATUS_AGAIN;
1228
#endif
1229 1230 1231 1232 1233
	default:
	  g_set_error (err, G_IO_CHANNEL_ERROR,
		       g_io_channel_error_from_errno (errno),
		       g_strerror (errno));
	  return G_IO_STATUS_ERROR;
1234
        }
1235
    }
1236 1237 1238 1239

  *bytes_read = result;

  return (result > 0) ? G_IO_STATUS_NORMAL : G_IO_STATUS_EOF;
1240
}
1241

1242
static GIOStatus