src/yaz-socket-manager.cpp

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