evolution/wvglibstreamclone.cc

   1 #include "wvglibstreamclone.h"
   2 #include <sys/poll.h>
   3
   4 WvGlibStreamClone *WvGlibStreamClone::singleton;
   5 GPollFunc WvGlibStreamClone::oldpoll;
   6
   7
   8 WvGlibStreamClone::WvGlibStreamClone(GMainContext *_context, IWvStream *_clone)
   9      : WvStreamClone(_clone)
  10 {
  11     context = _context;
  12     assert(!singleton);
  13     singleton = this;
  14     oldpoll = g_main_context_get_poll_func(context);
  15     g_main_context_set_poll_func(context, mypoll);
  16 }
  17
  18
  19 WvGlibStreamClone::~WvGlibStreamClone()
  20 {
  21     g_main_context_set_poll_func(context, oldpoll);
  22
  23     assert(singleton == this);
  24     singleton = NULL;
  25 }
  26
  27
  28 gint WvGlibStreamClone::mypoll(GPollFD *ufds, guint nfds, gint timeout)
  29 {
  30     SelectInfo si;
  31     bool sure = singleton->_build_selectinfo(si, timeout,
  32                                              false, false, false, true);
  33     if (sure)
  34         si.msec_timeout = 0;
  35
  36     GPollFD *newfds = new GPollFD[nfds + si.max_fd + 1], *fp;
  37     int newnfds = nfds;
  38     memcpy(newfds, ufds, nfds * sizeof(GPollFD));
  39     fp = newfds + nfds;
  40
  41     // add any wvstream-related file descriptors
  42     for (int fd = 0; fd <= si.max_fd; fd++)
  43     {
  44         bool readme   = FD_ISSET(fd, &si.read);
  45         bool writeme  = FD_ISSET(fd, &si.write);
  46         bool exceptme = FD_ISSET(fd, &si.except);
  47
  48         if (readme || writeme || exceptme)
  49         {
  50             fp->fd = fd;
  51             fp->events = readme*(POLLIN|POLLPRI)
  52                            | writeme*POLLOUT | exceptme*POLLERR;
  53             fp->revents = 0;
  54
  55             fp++;
  56             newnfds++;
  57         }
  58     }
  59
  60     int retval = oldpoll(newfds, newnfds, si.msec_timeout);
  61
  62     FD_ZERO(&si.read);
  63     FD_ZERO(&si.write);
  64     FD_ZERO(&si.except);
  65
  66     if (retval > 0)
  67     {
  68         fp = newfds + nfds;
  69         for (int i = nfds; i < newnfds; i++)
  70         {
  71             fp = newfds + i;
  72             if (fp->revents)
  73             {
  74                 if (fp->revents & (POLLIN|POLLPRI))
  75                     FD_SET(fp->fd, &si.read);
  76                 if (fp->revents & POLLOUT)
  77                     FD_SET(fp->fd, &si.write);
  78                 if (fp->revents & POLLERR)
  79                     FD_SET(fp->fd, &si.except);
  80                 retval--;
  81             }
  82         }
  83     }
  84
  85     memcpy(ufds, newfds, nfds * sizeof(GPollFD));
  86     delete[] newfds;
  87
  88     sure = sure || singleton->_process_selectinfo(si, true);
  89     if (sure)
  90         singleton->callback();
  91
  92     return retval;
  93 }
  94
  95