src/yaz-socket-manager.cpp

   1 /*
   2  * Copyright (c) 1998-2005, Index Data.
   3  * See the file LICENSE for details.
   4  *
   5  * $Id: yaz-socket-manager.cpp,v 1.38 2007-11-09 22:10:10 adam Exp $
   6  */
   7
   8 #if HAVE_SYS_TIME_H
   9 #include <sys/time.h>
  10 #endif
  11 #if HAVE_SYS_TYPES_H
  12 #include <sys/types.h>
  13 #endif
  14 #if HAVE_UNISTD_H
  15 #include <unistd.h>
  16 #endif
  17
  18 #include <errno.h>
  19 #include <string.h>
  20 #include <assert.h>
  21 #include <stdlib.h>
  22
  23 #include <yaz/log.h>
  24
  25 #include <yazpp/socket-manager.h>
  26 #include <yaz/poll.h>
  27
  28 using namespace yazpp_1;
  29
  30 SocketManager::SocketEntry **SocketManager::lookupObserver(
  31     ISocketObserver *observer)
  32 {
  33     SocketEntry **se;
  34
  35     for (se = &m_observers; *se; se = &(*se)->next)
  36         if ((*se)->observer == observer)
  37             break;
  38     return se;
  39 }
  40
  41 void SocketManager::addObserver(int fd, ISocketObserver *observer)
  42 {
  43     SocketEntry *se;
  44
  45     se = *lookupObserver(observer);
  46     if (!se)
  47     {
  48         se = new SocketEntry;
  49         se->next= m_observers;
  50         m_observers = se;
  51         se->observer = observer;
  52     }
  53     se->fd = fd;
  54     se->mask = 0;
  55     se->last_activity = 0;
  56     se->timeout = -1;
  57 }
  58
  59 void SocketManager::deleteObserver(ISocketObserver *observer)
  60 {
  61     SocketEntry **se = lookupObserver(observer);
  62     if (*se)
  63     {
  64         removeEvent (observer);
  65         SocketEntry *se_tmp = *se;
  66         *se = (*se)->next;
  67         delete se_tmp;
  68     }
  69 }
  70
  71 void SocketManager::deleteObservers()
  72 {
  73     SocketEntry *se = m_observers;
  74
  75     while (se)
  76     {
  77         SocketEntry *se_next = se->next;
  78         delete se;
  79         se = se_next;
  80     }
  81     m_observers = 0;
  82 }
  83
  84 void SocketManager::maskObserver(ISocketObserver *observer, int mask)
  85 {
  86     SocketEntry *se;
  87
  88     yaz_log(m_log, "obs=%p read=%d write=%d except=%d", observer,
  89                     mask & SOCKET_OBSERVE_READ,
  90                     mask & SOCKET_OBSERVE_WRITE,
  91                     mask & SOCKET_OBSERVE_EXCEPT);
  92
  93     se = *lookupObserver(observer);
  94     if (se)
  95         se->mask = mask;
  96 }
  97
  98 void SocketManager::timeoutObserver(ISocketObserver *observer,
  99                                         int timeout)
 100 {
 101     SocketEntry *se;
 102
 103     se = *lookupObserver(observer);
 104     if (se)
 105         se->timeout = timeout;
 106 }
 107
 108
 109 void SocketManager::inspect_poll_result(int res, struct yaz_poll_fd *fds,
 110                                         int no_fds, int timeout)
 111
 112 {
 113     yaz_log(m_log, "yaz_poll returned res=%d", res);
 114     time_t now = time(0);
 115     int i;
 116     int no_put_events = 0;
 117     SocketEntry *p;
 118
 119     for (i = 0, p = m_observers; p; p = p->next, i++)
 120     {
 121         enum yaz_poll_mask output_mask = fds[i].output_mask;
 122
 123         int mask = 0;
 124         if (output_mask & yaz_poll_read)
 125             mask |= SOCKET_OBSERVE_READ;
 126
 127         if (output_mask & yaz_poll_write)
 128             mask |= SOCKET_OBSERVE_WRITE;
 129
 130         if (output_mask & yaz_poll_except)
 131             mask |= SOCKET_OBSERVE_EXCEPT;
 132
 133         if (mask)
 134         {
 135             SocketEvent *event = new SocketEvent;
 136             p->last_activity = now;
 137             event->observer = p->observer;
 138             event->event = mask;
 139             putEvent (event);
 140             no_put_events++;
 141             yaz_log (m_log, "putEvent I/O mask=%d", mask);
 142         }
 143         else if (res == 0 && p->timeout_this == timeout)
 144         {
 145             SocketEvent *event = new SocketEvent;
 146             assert (p->last_activity);
 147             yaz_log (m_log, "putEvent timeout fd=%d, now = %ld last_activity=%ld timeout=%d",
 148                      p->fd, now, p->last_activity, p->timeout);
 149             p->last_activity = now;
 150             event->observer = p->observer;
 151             event->event = SOCKET_OBSERVE_TIMEOUT;
 152             putEvent (event);
 153             no_put_events++;
 154
 155         }
 156     }
 157     SocketEvent *event = getEvent();
 158     if (event)
 159     {
 160         event->observer->socketNotify(event->event);
 161         delete event;
 162     }
 163     else
 164     {
 165         // bug #2035
 166
 167         yaz_log(YLOG_WARN, "unhandled socket event. yaz_poll returned %d", res);
 168         yaz_log(YLOG_WARN, "no_put_events=%d no_fds=%d i=%d timeout=%d",
 169                 no_put_events, no_fds, i, timeout);
 170         abort();
 171     }
 172 }
 173
 174 int SocketManager::processEvent()
 175 {
 176     SocketEntry *p;
 177     SocketEvent *event = getEvent();
 178     int timeout = -1;
 179     yaz_log (m_log, "SocketManager::processEvent manager=%p", this);
 180     if (event)
 181     {
 182         event->observer->socketNotify(event->event);
 183         delete event;
 184         return 1;
 185     }
 186
 187     int res;
 188     time_t now = time(0);
 189     int i;
 190     int no_fds = 0;
 191     for (p = m_observers; p; p = p->next)
 192         no_fds++;
 193
 194     if (!no_fds)
 195         return 0;
 196     struct yaz_poll_fd *fds = new yaz_poll_fd [no_fds];
 197     for (i = 0, p = m_observers; p; p = p->next, i++)
 198     {
 199         fds[i].fd = p->fd;
 200         int input_mask = 0;
 201         if (p->mask & SOCKET_OBSERVE_READ)
 202             input_mask += yaz_poll_read;
 203         if (p->mask & SOCKET_OBSERVE_WRITE)
 204             input_mask += yaz_poll_write;
 205         if (p->mask & SOCKET_OBSERVE_EXCEPT)
 206             input_mask += yaz_poll_except;
 207         if (p->timeout > 0 ||
 208             (p->timeout == 0 && (p->mask & SOCKET_OBSERVE_WRITE) == 0))
 209         {
 210             int timeout_this;
 211             timeout_this = p->timeout;
 212             if (p->last_activity)
 213                 timeout_this -= now - p->last_activity;
 214             else
 215                 p->last_activity = now;
 216             if (timeout_this < 0 || timeout_this > 2147483646)
 217                 timeout_this = 0;
 218             if (timeout == -1 || timeout_this < timeout)
 219                 timeout = timeout_this;
 220             p->timeout_this = timeout_this;
 221             yaz_log (m_log, "SocketManager::select timeout_this=%d",
 222                      p->timeout_this);
 223         }
 224         else
 225             p->timeout_this = -1;
 226         fds[i].input_mask = (enum yaz_poll_mask) input_mask;
 227     }
 228
 229     int pass = 0;
 230     while ((res = yaz_poll(fds, no_fds, timeout, 0)) < 0)
 231     {
 232         if (errno != EINTR)
 233         {
 234             yaz_log(YLOG_ERRNO|YLOG_WARN, "yaz_poll");
 235             yaz_log(YLOG_WARN, "errno=%d timeout=%d", errno, timeout);
 236             if (++pass > 10)
 237                 return -1;
 238         }
 239     }
 240
 241     inspect_poll_result(res, fds, no_fds, timeout);
 242
 243     delete [] fds;
 244     return 1;
 245 }
 246
 247
 248 //    n p    n p  ......   n p    n p
 249 //   front                        back
 250
 251 void SocketManager::putEvent(SocketEvent *event)
 252 {
 253     // put in back of queue
 254     if (m_queue_back)
 255     {
 256         m_queue_back->prev = event;
 257         assert (m_queue_front);
 258     }
 259     else
 260     {
 261         assert (!m_queue_front);
 262         m_queue_front = event;
 263     }
 264     event->next = m_queue_back;
 265     event->prev = 0;
 266     m_queue_back = event;
 267 }
 268
 269 SocketManager::SocketEvent *SocketManager::getEvent()
 270 {
 271     // get from front of queue
 272     SocketEvent *event = m_queue_front;
 273     if (!event)
 274         return 0;
 275     assert (m_queue_back);
 276     m_queue_front = event->prev;
 277     if (m_queue_front)
 278     {
 279         assert (m_queue_back);
 280         m_queue_front->next = 0;
 281     }
 282     else
 283         m_queue_back = 0;
 284     return event;
 285 }
 286
 287 void SocketManager::removeEvent(ISocketObserver *observer)
 288 {
 289     SocketEvent *ev = m_queue_back;
 290     while (ev)
 291     {
 292         SocketEvent *ev_next = ev->next;
 293         if (observer == ev->observer)
 294         {
 295             if (ev->prev)
 296                 ev->prev->next = ev->next;
 297             else
 298                 m_queue_back = ev->next;
 299             if (ev->next)
 300                 ev->next->prev = ev->prev;
 301             else
 302                 m_queue_front = ev->prev;
 303             delete ev;
 304         }
 305         ev = ev_next;
 306     }
 307 }
 308
 309 SocketManager::SocketManager()
 310 {
 311     m_observers = 0;
 312     m_queue_front = 0;
 313     m_queue_back = 0;
 314     m_log = YLOG_DEBUG;
 315 }
 316
 317 SocketManager::~SocketManager()
 318 {
 319     deleteObservers();
 320 }
 321 /*
 322  * Local variables:
 323  * c-basic-offset: 4
 324  * indent-tabs-mode: nil
 325  * End:
 326  * vim: shiftwidth=4 tabstop=8 expandtab
 327  */
 328