2007-07-13 Marcus Brinkmann <marcus@g10code.de>
[gpgme.git] / gpgme / w32-glib-io.c
index 8e7abea..8f56eee 100644 (file)
@@ -32,6 +32,7 @@
 #include <unistd.h>
 #include <sys/time.h>
 #include <sys/types.h>
+#include <glib.h>
 #include <windows.h>
 #include <io.h>
 
 #include "sema.h"
 #include "debug.h"
 
-#include <glib.h>
+#ifndef O_BINARY
+#ifdef _O_BINARY
+#define O_BINARY       _O_BINARY
+#else
+#define O_BINARY       0
+#endif
+#endif
 
 \f
 /* This file is an ugly hack to get GPGME working with glib on Windows
    whole thing is also interconnected with the creation of pipes and
    child processes.
 
-   The following rules apply only to this I/O backend:
-
-   * All "file descriptors" that GPGME gives to the application are
-   not system file descriptors, but some internal number maintained by
-   GPGME.  I call them "Something like a file descriptor" (SLAFD).
-   It's an ugly name for an ugly thing.
-
-   * The application can use this "file descriptor" for exactly one
-   thing: To call gpgme_get_giochannel on it.  This returns the
-   GIOChannel that the application can actually use.  The channel can
-   then be integrated in the event loop.
+   The following rule applies only to this I/O backend:
 
    * ALL operations must use the user defined event loop.  GPGME can
    not anymore provide its own event loop.  This is mostly a sanity
    a per-context level.  */
 
 \f
-/* Something like a file descriptor.  We can not use "real" file
-   descriptors, because for some reason we can't create them from
-   osfhandles to be inherited.  Argh!  */
-static struct
-{
-  /* This is non-null if the entry is used.  */
-  HANDLE osfhandle;
+#define MAX_SLAFD 256
 
-  /* This is non-null if there is a GIOChannel for this handle.  Only
-     for our end of the pipe.  */
-  GIOChannel *channel;
-} slafd_table[256];
-
-#define MAX_SLAFD ((int) DIM (slafd_table))
-
-static int
-create_slafd (HANDLE handle, int create_channel)
-{
-  int slafd;
-
-  for (slafd = 0; slafd < MAX_SLAFD; slafd++)
-    if (slafd_table[slafd].osfhandle == NULL)
-      break;
-
-  if (slafd == MAX_SLAFD)
-    return -1;
-
-  if (create_channel)
-    {
-      /* FIXME: Do we need to specify the direction, too?  */
-      //      int fd = _open_osfhandle ((long) handle, 0);
-      //      DEBUG2("opened handle %p to %i\n", handle, fd);
-      slafd_table[slafd].channel = g_io_channel_unix_new ((int)handle);
-      if (!slafd_table[slafd].channel)
-       {
-         errno = EIO;  /* XXX */
-         return -1;
-       }
-    }
-  else
-    slafd_table[slafd].channel = NULL;
-  
-  slafd_table[slafd].osfhandle = handle;
-  return slafd;
-}
+GIOChannel *giochannel_table[MAX_SLAFD];
 
 
 static GIOChannel *
-find_channel (int fd)
+find_channel (int fd, int create)
 {
   if (fd < 0 || fd >= MAX_SLAFD)
     return NULL;
 
-  return slafd_table[fd].channel;
-}
-
-
-static HANDLE
-find_handle (int fd)
-{
-  if (fd < 0 || fd >= MAX_SLAFD)
-    return NULL;
+  if (create && !giochannel_table[fd])
+    {
+      giochannel_table[fd] = g_io_channel_win32_new_fd (fd);
+      g_io_channel_set_encoding (giochannel_table[fd], NULL, NULL);
+      g_io_channel_set_buffered (giochannel_table[fd], FALSE);
+    }
 
-  return slafd_table[fd].osfhandle;
+  return giochannel_table[fd];
 }
 
-
 /* Look up the giochannel for "file descriptor" FD.  */
 GIOChannel *
 gpgme_get_giochannel (int fd)
 {
-  return find_channel (fd);
+  return find_channel (fd, 0);
+}
+
+
+/* Write the printable version of FD to the buffer BUF of length
+   BUFLEN.  The printable version is the representation on the command
+   line that the child process expects.  */
+int
+_gpgme_io_fd2str (char *buf, int buflen, int fd)
+{
+  return snprintf (buf, buflen, "%ld", (long) _get_osfhandle (fd));
 }
 
 \f
@@ -162,10 +123,11 @@ _gpgme_io_subsystem_init (void)
 \f
 static struct
 {
-  void (*handler) (int,void*);
+  _gpgme_close_notify_handler_t handler;
   void *value;
 } notify_table[MAX_SLAFD];
 
+
 int
 _gpgme_io_read (int fd, void *buffer, size_t count)
 {
@@ -176,7 +138,7 @@ _gpgme_io_read (int fd, void *buffer, size_t count)
 
   DEBUG2 ("fd %d: about to read %d bytes\n", fd, (int) count);
 
-  chan = find_channel (fd);
+  chan = find_channel (fd, 0);
   if (!chan)
     {
       DEBUG1 ("fd %d: no channel registered\n", fd);
@@ -225,7 +187,7 @@ _gpgme_io_write (int fd, const void *buffer, size_t count)
   DEBUG2 ("fd %d: about to write %d bytes\n", fd, (int) count);
   _gpgme_debug (2, "fd %d: write `%.*s'\n", fd, (int) count, buffer);
 
-  chan = find_channel (fd);
+  chan = find_channel (fd, 0);
   if (!chan)
     {
       DEBUG1 ("fd %d: no channel registered\n", fd);
@@ -247,69 +209,61 @@ _gpgme_io_write (int fd, const void *buffer, size_t count)
 
 
 int
-_gpgme_io_pipe ( int filedes[2], int inherit_idx )
+_gpgme_io_pipe (int filedes[2], int inherit_idx)
 {
-    HANDLE r, w;
-    SECURITY_ATTRIBUTES sec_attr;
-
-    memset (&sec_attr, 0, sizeof sec_attr );
-    sec_attr.nLength = sizeof sec_attr;
-    sec_attr.bInheritHandle = FALSE;
+  GIOChannel *chan;
 
-    DEBUG1("INHERIT: %i\n", inherit_idx);
-    
 #define PIPEBUF_SIZE  4096
-    if (!CreatePipe ( &r, &w, &sec_attr, PIPEBUF_SIZE))
-        return -1;
-    /* Make one end inheritable. */
-    if ( inherit_idx == 0 ) {
-        HANDLE h;
-        if (!DuplicateHandle( GetCurrentProcess(), r,
-                              GetCurrentProcess(), &h, 0,
-                              TRUE, DUPLICATE_SAME_ACCESS ) ) {
-            DEBUG1 ("DuplicateHandle failed: ec=%d\n", (int)GetLastError());
-            CloseHandle (r);
-            CloseHandle (w);
-            return -1;
-        }
-        CloseHandle (r);
-        r = h;
+  if (_pipe (filedes, PIPEBUF_SIZE, O_NOINHERIT | O_BINARY) == -1)
+    return -1;
+
+  /* Make one end inheritable. */
+  if (inherit_idx == 0)
+    {
+      int new_read;
+
+      new_read = _dup (filedes[0]);
+      _close (filedes[0]);
+      filedes[0] = new_read;
+
+      if (new_read < 0)
+       {
+         _close (filedes[1]);
+         return -1;
+       }
     }
-    else if ( inherit_idx == 1 ) {
-        HANDLE h;
-        if (!DuplicateHandle( GetCurrentProcess(), w,
-                              GetCurrentProcess(), &h, 0,
-                              TRUE, DUPLICATE_SAME_ACCESS ) ) {
-            DEBUG1 ("DuplicateHandle failed: ec=%d\n", (int)GetLastError());
-            CloseHandle (r);
-            CloseHandle (w);
-            return -1;
-        }
-        CloseHandle (w);
-        w = h;
+  else if (inherit_idx == 1)
+    {
+      int new_write;
+
+      new_write = _dup (filedes[1]);
+      _close (filedes[1]);
+      filedes[1] = new_write;
+
+      if (new_write < 0)
+       {
+         _close (filedes[0]);
+         return -1;
+       }
     }
-    filedes[0] = create_slafd (r, inherit_idx == 1);
-    if (filedes[0] == -1)
-      {
-       DEBUG1 ("create_slafd failed: ec=%d\n", errno);
-       CloseHandle (r);
-       CloseHandle (w);
-       return -1;
-      }
 
-    filedes[1] = create_slafd (w, inherit_idx == 0);
-    if (filedes[1] == -1)
-      {
-       DEBUG1 ("create_slafd failed: ec=%d\n", errno);
-       _gpgme_io_close (filedes[0]);
-       CloseHandle (r);
-       CloseHandle (w);
-       return -1;
-      }
+  /* Now we have a pipe with the right end inheritable.  The other end
+     should have a giochannel.  */
+  chan = find_channel (filedes[1 - inherit_idx], 1);
+  if (!chan)
+    {
+      DEBUG2 ("channel creation for %d failed: ec=%d\n",
+             filedes[1 - inherit_idx], errno);
+      _close (filedes[0]);
+      _close (filedes[1]);
+      return -1;
+    }
 
-    DEBUG5 ("CreatePipe %p %p %d %d inherit=%d\n", r, w,
-                   filedes[0], filedes[1], inherit_idx );
-    return 0;
+  DEBUG5 ("CreatePipe %d (%p) %d (%p) inherit=%p\n",
+         filedes[0], (HANDLE) _get_osfhandle (filedes[0]),
+         filedes[1], (HANDLE) _get_osfhandle (filedes[1]),
+         chan);
+  return 0;
 }
 
 
@@ -334,27 +288,23 @@ _gpgme_io_close (int fd)
     }
 
   /* Then do the close.  */    
-  chan = slafd_table[fd].channel;
+  chan = giochannel_table[fd];
   if (chan)
     {
       g_io_channel_shutdown (chan, 1, NULL);
       g_io_channel_unref (chan);
+      giochannel_table[fd] = NULL;
     }
-  
-  if (!CloseHandle (slafd_table[fd].osfhandle))
-    { 
-      DEBUG2 ("CloseHandle for fd %d failed: ec=%d\n",
-             fd, (int)GetLastError ());
-    }
-
-  slafd_table[fd].osfhandle = NULL;
+  else
+    _close (fd);
 
   return 0;
 }
 
 
 int
-_gpgme_io_set_close_notify (int fd, void (*handler)(int, void*), void *value)
+_gpgme_io_set_close_notify (int fd, _gpgme_close_notify_handler_t handler,
+                           void *value)
 {
   assert (fd != -1);
 
@@ -372,21 +322,25 @@ _gpgme_io_set_nonblocking (int fd)
 {
   GIOChannel *chan;
   GIOStatus status;
-
-  chan = find_channel (fd);
+  chan = find_channel (fd, 0);
   if (!chan)
     {
+      DEBUG1 ("set nonblocking for fd %d failed: channel not found", fd);
       errno = EIO;
       return -1;
     }
 
-  status = g_io_channel_set_flags (chan,
+   status = g_io_channel_set_flags (chan,
                                   g_io_channel_get_flags (chan) |
                                   G_IO_FLAG_NONBLOCK, NULL);
   if (status != G_IO_STATUS_NORMAL)
     {
-      errno = EIO;
-      return -1;
+      /* glib 1.9.2 does not implement set_flags and returns an error. */
+      DEBUG2 ("set nonblocking for fd %d failed: status=%d - ignored",
+              fd, status);
+/*       errno = EIO; */
+/*       return -1; */
     }
 
   return 0;
@@ -394,37 +348,57 @@ _gpgme_io_set_nonblocking (int fd)
 
 
 static char *
-build_commandline ( char **argv )
+build_commandline (char **argv)
 {
-  int i, n = 0;
-  char *buf, *p;
+  int i;
+  int j;
+  int n = 0;
+  char *buf;
+  char *p;
   
-  /* FIXME: we have to quote some things because under Windows the
-   * program parses the commandline and does some unquoting.  For now
-   * we only do very basic quoting to the first argument because this
-   * one often contains a space (e.g. C:\\Program Files\GNU\GnuPG\gpg.exe) 
-   * and we would produce an invalid line in that case.  */
-  for (i=0; argv[i]; i++)
-    n += strlen (argv[i]) + 2 + 1; /* 2 extra bytes for possible quoting */
+  /* We have to quote some things because under Windows the program
+     parses the commandline and does some unquoting.  We enclose the
+     whole argument in double-quotes, and escape literal double-quotes
+     as well as backslashes with a backslash.  We end up with a
+     trailing space at the end of the line, but that is harmless.  */
+  for (i = 0; argv[i]; i++)
+    {
+      p = argv[i];
+      /* The leading double-quote.  */
+      n++;
+      while (*p)
+       {
+         /* An extra one for each literal that must be escaped.  */
+         if (*p == '\\' || *p == '"')
+           n++;
+         n++;
+         p++;
+       }
+      /* The trailing double-quote and the delimiter.  */
+      n += 2;
+    }
+  /* And a trailing zero.  */
+  n++;
+
   buf = p = malloc (n);
-  if ( !buf )
+  if (!buf)
     return NULL;
-  *buf = 0;
-  if ( argv[0] )
+  for (i = 0; argv[i]; i++)
     {
-      if (strpbrk (argv[0], " \t"))
-        p = stpcpy (stpcpy (stpcpy (p, "\""), argv[0]), "\"");
-      else
-        p = stpcpy (p, argv[0]);
-      for (i = 1; argv[i]; i++)
-        {
-          if (!*argv[i])
-            p = stpcpy (p, " \"\"");
-          else
-            p = stpcpy (stpcpy (p, " "), argv[i]);
-        }
+      char *argvp = argv[i];
+
+      *(p++) = '"';
+      while (*argvp)
+       {
+         if (*argvp == '\\' || *argvp == '"')
+           *(p++) = '\\';
+         *(p++) = *(argvp++);
+       }
+      *(p++) = '"';
+      *(p++) = ' ';
     }
-  
+  *(p++) = 0;
+
   return buf;
 }
 
@@ -453,11 +427,11 @@ _gpgme_io_spawn ( const char *path, char **argv,
     /* FIXME.  */
     int debug_me = 0;
 
-    memset (&sec_attr, 0, sizeof sec_attr );
+    memset (&sec_attr, 0, sizeof sec_attr);
     sec_attr.nLength = sizeof sec_attr;
     sec_attr.bInheritHandle = FALSE;
 
-    arg_string = build_commandline ( argv );
+    arg_string = build_commandline (argv);
     if (!arg_string )
         return -1; 
 
@@ -471,19 +445,20 @@ _gpgme_io_spawn ( const char *path, char **argv,
 
     for (i=0; fd_child_list[i].fd != -1; i++ ) {
         if (fd_child_list[i].dup_to == 0 ) {
-            si.hStdInput = find_handle (fd_child_list[i].fd);
+            si.hStdInput = (HANDLE) _get_osfhandle (fd_child_list[i].fd);
             DEBUG2 ("using %d (%p) for stdin", fd_child_list[i].fd,
-                   find_handle (fd_child_list[i].fd));
+                   _get_osfhandle (fd_child_list[i].fd));
             duped_stdin=1;
         }
         else if (fd_child_list[i].dup_to == 1 ) {
-            si.hStdOutput = find_handle (fd_child_list[i].fd);
+            si.hStdOutput = (HANDLE) _get_osfhandle (fd_child_list[i].fd);
             DEBUG2 ("using %d (%p) for stdout", fd_child_list[i].fd,
-                   find_handle (fd_child_list[i].fd));
+                   _get_osfhandle (fd_child_list[i].fd));
         }
         else if (fd_child_list[i].dup_to == 2 ) {
-            si.hStdError = find_handle (fd_child_list[i].fd);
-            DEBUG1 ("using %d for stderr", fd_child_list[i].fd );
+            si.hStdError = (HANDLE) _get_osfhandle (fd_child_list[i].fd);
+            DEBUG2 ("using %d (%p) for stderr", fd_child_list[i].fd,
+                   _get_osfhandle (fd_child_list[i].fd));
             duped_stderr = 1;
         }
     }
@@ -574,7 +549,115 @@ _gpgme_io_spawn ( const char *path, char **argv,
 int
 _gpgme_io_select (struct io_select_fd_s *fds, size_t nfds, int nonblock)
 {
-  assert (!"ARGH!  The user of this library MUST define io callbacks!");
-  errno = EINVAL;
-  return -1;
+  int     npollfds;
+  GPollFD *pollfds;
+  int     *pollfds_map; 
+  int i, j;
+  int any, n, count;
+  int timeout = 1000;  /* Use a 1s timeout.  */
+  void *dbg_help = NULL;
+
+  if (nonblock)
+    timeout = 0;
+
+  pollfds = calloc (nfds, sizeof *pollfds);
+  if (!pollfds)
+    return -1;
+  pollfds_map = calloc (nfds, sizeof *pollfds_map);
+  if (!pollfds_map)
+    {
+      free (pollfds);
+      return -1;
+    }
+  npollfds = 0;
+
+  DEBUG_BEGIN (dbg_help, 3, "gpgme:select on [ ");
+  any = 0;
+  for (i = 0; i < nfds; i++)
+    {
+      if (fds[i].fd == -1) 
+       continue;
+      if (fds[i].frozen)
+       DEBUG_ADD1 (dbg_help, "f%d ", fds[i].fd);
+      else if (fds[i].for_read )
+       {
+          GIOChannel *chan = find_channel (fds[i].fd, 0);
+          assert (chan);
+          g_io_channel_win32_make_pollfd (chan, G_IO_IN, pollfds + npollfds);
+          pollfds_map[npollfds] = i;
+         DEBUG_ADD2 (dbg_help, "r%d<%d> ", fds[i].fd, pollfds[npollfds].fd);
+          npollfds++;
+         any = 1;
+        }
+      else if (fds[i].for_write)
+       {
+          GIOChannel *chan = find_channel (fds[i].fd, 0);
+          assert (chan);
+          g_io_channel_win32_make_pollfd (chan, G_IO_OUT, pollfds + npollfds);
+          pollfds_map[npollfds] = i;
+         DEBUG_ADD2 (dbg_help, "w%d<%d> ", fds[i].fd, pollfds[npollfds].fd);
+          npollfds++;
+         any = 1;
+        }
+      fds[i].signaled = 0;
+    }
+  DEBUG_END (dbg_help, "]"); 
+  if (!any)
+    {
+      count = 0;
+      goto leave;
+    }
+
+
+  count = g_io_channel_win32_poll (pollfds, npollfds, timeout);
+  if (count < 0)
+    {
+      int saved_errno = errno;
+      DEBUG1 ("_gpgme_io_select failed: %s\n", strerror (errno));
+      errno = saved_errno;
+      goto leave;
+    }
+
+  DEBUG_BEGIN (dbg_help, 3, "select OK [ ");
+  if (DEBUG_ENABLED (dbg_help))
+    {
+      for (i = 0; i < npollfds; i++)
+       {
+         if ((pollfds[i].revents & G_IO_IN))
+           DEBUG_ADD1 (dbg_help, "r%d ", fds[pollfds_map[i]].fd);
+          if ((pollfds[i].revents & G_IO_OUT))
+            DEBUG_ADD1 (dbg_help, "w%d ", fds[pollfds_map[i]].fd);
+        }
+      DEBUG_END (dbg_help, "]");
+    }
+    
+  /* COUNT is used to stop the lop as soon as possible.  */
+  for (n = count, i = 0; i < npollfds && n; i++)
+    {
+      j = pollfds_map[i];
+      assert (j >= 0 && j < nfds);
+      if (fds[j].fd == -1)
+       ;
+      else if (fds[j].for_read)
+       {
+         if ((pollfds[i].revents & G_IO_IN))
+           {
+             fds[j].signaled = 1;
+             n--;
+            }
+        }
+      else if (fds[j].for_write)
+       {
+         if ((pollfds[i].revents & G_IO_OUT))
+           {
+             fds[j].signaled = 1;
+             n--;
+            }
+        }
+    }
+
+leave:
+  free (pollfds);
+  free (pollfds_map);
+  return count;
 }