Implemented Generic select hook for ZOOM (bug #803). This is achieved with

[yaz-moved-to-github.git] / src / zoom-socket.c

diff --git a/src/zoom-socket.c b/src/zoom-socket.c
new file mode 100644 (file)
index 0000000..a3adebf
--- /dev/null
+++ b/src/zoom-socket.c
@@ -0,0 +1,219 @@
+/*
+ * Copyright (C) 1995-2007, Index Data ApS
+ * See the file LICENSE for details.
+ *
+ * $Id: zoom-socket.c,v 1.1 2007-01-09 13:56:48 adam Exp $
+ */
+/**
+ * \file zoom-socket.c
+ * \brief Implements ZOOM C socket interface.
+ */
+
+#include <assert.h>
+#include <string.h>
+#include <errno.h>
+#include <yaz/zoom.h>
+
+#include <yaz/log.h>
+
+#if HAVE_SYS_TYPES_H
+#include <sys/types.h>
+#endif
+#if HAVE_SYS_TIME_H
+#include <sys/time.h>
+#endif
+#if HAVE_SYS_POLL_H
+#include <sys/poll.h>
+#endif
+#if HAVE_SYS_SELECT_H
+#include <sys/select.h>
+#endif
+#ifdef WIN32
+#if FD_SETSIZE < 512
+#define FD_SETSIZE 512
+#endif
+#include <winsock.h>
+#endif
+
+
+ZOOM_API(int)
+    ZOOM_event_sys_select(int no, ZOOM_connection *cs)
+{
+    struct timeval tv;
+    fd_set input, output, except;
+    int i, r;
+    int max_fd = 0;
+    int timeout = 30;
+    int nfds = 0;
+
+    FD_ZERO(&input);
+    FD_ZERO(&output);
+    FD_ZERO(&except);
+
+    for (i = 0; i<no; i++)
+    {
+        ZOOM_connection c = cs[i];
+        int fd, mask;
+        
+        if (!c)
+            continue;
+        fd = ZOOM_connection_get_socket(c);
+        mask = ZOOM_connection_get_mask(c);
+        timeout = ZOOM_connection_get_timeout(c);
+
+        if (fd == -1)
+            continue;
+        if (max_fd < fd)
+            max_fd = fd;
+        
+        if (mask & ZOOM_SELECT_READ)
+            FD_SET(fd, &input);
+        if (mask & ZOOM_SELECT_WRITE)
+            FD_SET(fd, &output);
+        if (mask & ZOOM_SELECT_EXCEPT)
+            FD_SET(fd, &except);
+        if (mask)
+            nfds++;
+    }
+    if (nfds == 0)
+        return 0;
+
+    tv.tv_sec = timeout;
+    tv.tv_usec = 0;
+
+    while ((r = select(max_fd+1, &input, &output, &except,
+                       (timeout == -1 ? 0 : &tv))) < 0 && errno == EINTR)
+    {
+        ;
+    }
+    if (r < 0)
+    {
+        yaz_log(YLOG_WARN|YLOG_ERRNO, "ZOOM_event_sys_select");
+        return r;
+    }
+
+    for (i = 0; i<no; i++)
+    {
+        ZOOM_connection c = cs[i];
+        int fd, mask;
+
+        if (!c)
+            continue;
+        fd = ZOOM_connection_get_socket(c);
+        mask = 0;
+        if (r)
+        {
+            /* no timeout and real socket */
+            if (FD_ISSET(fd, &input))
+                mask += ZOOM_SELECT_READ;
+            if (FD_ISSET(fd, &output))
+                mask += ZOOM_SELECT_WRITE;
+            if (FD_ISSET(fd, &except))
+                mask += ZOOM_SELECT_EXCEPT;
+            if (mask)
+                ZOOM_connection_fire_event_socket(c, mask);
+        }
+        else
+            ZOOM_connection_fire_event_timeout(c);
+    }
+    return r;
+}
+
+#if HAVE_SYS_POLL_H
+ZOOM_API(int)
+    ZOOM_event_sys_poll(int no, ZOOM_connection *cs)
+{
+    struct pollfd pollfds[1024];
+    ZOOM_connection poll_cs[1024];
+    int i, r;
+    int nfds = 0;
+    int timeout = 30;
+
+    for (i = 0; i<no; i++)
+    {
+        ZOOM_connection c = cs[i];
+        int fd, mask;
+        
+        if (!c)
+            continue;
+        fd = ZOOM_connection_get_socket(c);
+        mask = ZOOM_connection_get_mask(c);
+        timeout = ZOOM_connection_get_timeout(c);
+
+        if (fd == -1)
+            continue;
+        if (mask)
+        {
+            short poll_events = 0;
+
+            if (mask & ZOOM_SELECT_READ)
+                poll_events += POLLIN;
+            if (mask & ZOOM_SELECT_WRITE)
+                poll_events += POLLOUT;
+            if (mask & ZOOM_SELECT_EXCEPT)
+                poll_events += POLLERR;
+            pollfds[nfds].fd = fd;
+            pollfds[nfds].events = poll_events;
+            pollfds[nfds].revents = 0;
+            poll_cs[nfds] = c;
+            nfds++;
+        }
+    }
+    if (nfds == 0)
+        return 0;
+    while ((r = poll(pollfds, nfds,
+         (timeout == -1 ? -1 : timeout * 1000))) < 0
+          && errno == EINTR)
+    {
+        ;
+    }
+    if (r < 0)
+    {
+        yaz_log(YLOG_WARN|YLOG_ERRNO, "ZOOM_event_sys_poll");
+        return r;
+    }
+    for (i = 0; i<nfds; i++)
+    {
+        ZOOM_connection c = poll_cs[i];
+        if (r)
+        {
+            int mask = 0;
+            if (pollfds[i].revents & POLLIN)
+                mask += ZOOM_SELECT_READ;
+            if (pollfds[i].revents & POLLOUT)
+                mask += ZOOM_SELECT_WRITE;
+            if (pollfds[i].revents & POLLERR)
+                mask += ZOOM_SELECT_EXCEPT;
+            ZOOM_connection_fire_event_socket(c, mask);
+        }
+        else
+            ZOOM_connection_fire_event_timeout(c);
+    }
+    return r;
+}
+#endif
+
+ZOOM_API(int)
+    ZOOM_event(int no, ZOOM_connection *cs)
+{
+    int r;
+
+    r = ZOOM_process_event(no, cs);
+    if (r)
+        return r;
+#if HAVE_SYS_POLL_H
+    ZOOM_event_sys_poll(no, cs);
+#else
+    ZOOM_event_sys_select(no, cs);
+#endif
+    return ZOOM_process_event(no, cs);
+}
+
+/*
+ * Local variables:
+ * c-basic-offset: 4
+ * indent-tabs-mode: nil
+ * End:
+ * vim: shiftwidth=4 tabstop=8 expandtab
+ */
+