src/yaz-socket-manager.cpp

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