]> git.ipfire.org Git - thirdparty/cups.git/blobdiff - scheduler/select.c
Merge changes from CUPS 1.4svn-r7874.
[thirdparty/cups.git] / scheduler / select.c
index 960b0a37d1b809b0599bd437a450dbd97a80f396..20c9547fd60c1d05a0c4f9c30d1a069b29953ffe 100644 (file)
@@ -1,25 +1,16 @@
 /*
- * "$Id: select.c 6376 2007-03-21 06:39:10Z mike $"
+ * "$Id: select.c 7093 2007-11-30 19:09:36Z mike $"
  *
  *   Select abstraction functions for the Common UNIX Printing System (CUPS).
  *
+ *   Copyright 2007-2008 by Apple Inc.
  *   Copyright 2006-2007 by Easy Software Products.
  *
  *   These coded instructions, statements, and computer programs are the
- *   property of Easy Software Products and are protected by Federal
- *   copyright law.  Distribution and use rights are outlined in the file
- *   "LICENSE.txt" which should have been included with this file.  If this
- *   file is missing or damaged please contact Easy Software Products
- *   at:
- *
- *       Attn: CUPS Licensing Information
- *       Easy Software Products
- *       44141 Airport View Drive, Suite 204
- *       Hollywood, Maryland 20636 USA
- *
- *       Voice: (301) 373-9600
- *       EMail: cups-info@cups.org
- *         WWW: http://www.cups.org
+ *   property of Apple Inc. and are protected by Federal copyright
+ *   law.  Distribution and use rights are outlined in the file "LICENSE.txt"
+ *   which should have been included with this file.  If this file is
+ *   file is missing or damaged, see the license at "http://www.cups.org/".
  *
  * Contents:
  *
@@ -42,6 +33,7 @@
 
 #ifdef HAVE_EPOLL
 #  include <sys/epoll.h>
+#  include <sys/poll.h>
 #elif defined(HAVE_KQUEUE)
 #  include <sys/event.h>
 #  include <sys/time.h>
  * 
  *     0. Common Stuff
  *         a. CUPS array of file descriptor to callback functions
- *            and data.
- *         b. cupsdStartSelect() creates the array
- *         c. cupsdStopSelect() destroys the array and all elements.
+ *            and data + temporary array of removed fd's.
+ *         b. cupsdStartSelect() creates the arrays
+ *         c. cupsdStopSelect() destroys the arrays and all elements.
  *         d. cupsdAddSelect() adds to the array and allocates a
  *            new callback element.
- *         e. cupsdRemoveSelect() removes from the array and frees
- *            the callback element.
+ *         e. cupsdRemoveSelect() removes from the active array and
+ *            adds to the inactive array.
  *         f. _cupsd_fd_t provides a reference-counted structure for
  *            tracking file descriptors that are monitored.
- * 
+ *         g. cupsdDoSelect() frees all inactive FDs.
+ *
  *     1. select() O(n)
  *         a. Input/Output fd_set variables, copied to working
  *            copies and then used with select().
  *
  *   In tests using the "make test" target with option 0 (keep cupsd
  *   running) and the "testspeed" program with "-c 50 -r 1000", epoll()
- *   performed 5.5% slower select(), followed by kqueue() at 16% slower
- *   than select() and poll() at 18% slower than select().  Similar
+ *   performed 5.5% slower than select(), followed by kqueue() at 16%
+ *   slower than select() and poll() at 18% slower than select().  Similar
  *   results were seen with twice the number of client connections.
  *
  *   The epoll() and kqueue() performance is likely limited by the
@@ -217,11 +210,12 @@ typedef struct _cupsd_fd_s
  */
 
 static cups_array_t    *cupsd_fds = NULL;
+#if defined(HAVE_EPOLL) || defined(HAVE_KQUEUE)
+static cups_array_t    *cupsd_inactive_fds = NULL;
+static int             cupsd_in_select = 0;
+#endif /* HAVE_EPOLL || HAVE_KQUEUE */
 
-#ifdef HAVE_EPOLL
-static int             cupsd_epoll_fd = -1;
-static struct epoll_event *cupsd_epoll_events = NULL;
-#elif defined(HAVE_KQUEUE)
+#ifdef HAVE_KQUEUE
 static int             cupsd_kqueue_fd = -1,
                        cupsd_kqueue_changes = 0;
 static struct kevent   *cupsd_kqueue_events = NULL;
@@ -229,12 +223,16 @@ static struct kevent      *cupsd_kqueue_events = NULL;
 static int             cupsd_alloc_pollfds = 0,
                        cupsd_update_pollfds = 0;
 static struct pollfd   *cupsd_pollfds = NULL;
+#  ifdef HAVE_EPOLL
+static int             cupsd_epoll_fd = -1;
+static struct epoll_event *cupsd_epoll_events = NULL;
+#  endif /* HAVE_EPOLL */
 #else /* select() */
 static fd_set          cupsd_global_input,
                        cupsd_global_output,
                        cupsd_current_input,
                        cupsd_current_output;
-#endif /* HAVE_EPOLL */
+#endif /* HAVE_KQUEUE */
 
 
 /*
@@ -261,7 +259,9 @@ cupsdAddSelect(int             fd,  /* I - File descriptor */
               void            *data)   /* I - Data to pass to callback */
 {
   _cupsd_fd_t  *fdptr;                 /* File descriptor record */
+#ifdef HAVE_EPOLL
   int          added;                  /* 1 if added, 0 if modified */
+#endif /* HAVE_EPOLL */
 
 
  /*
@@ -298,31 +298,16 @@ cupsdAddSelect(int             fd,        /* I - File descriptor */
       return (0);
     }
 
+#ifdef HAVE_EPOLL
     added = 1;
   }
   else
     added = 0;
-
-#ifdef HAVE_EPOLL
-  {
-    struct epoll_event event;          /* Event data */
-
-
-    event.events = 0;
-
-    if (read_cb)
-      event.events |= EPOLLIN;
-
-    if (write_cb)
-      event.events |= EPOLLOUT;
-
-    event.data.ptr = fdptr;
-
-    epoll_ctl(cupsd_epoll_fd, added ? EPOLL_CTL_ADD : EPOLL_CTL_MOD, fd,
-              &event);
+#else
   }
+#endif /* HAVE_EPOLL */
 
-#elif defined(HAVE_KQUEUE)
+#ifdef HAVE_KQUEUE
   {
     struct kevent      event;          /* Event data */
     struct timespec    timeout;        /* Timeout value */
@@ -365,6 +350,33 @@ cupsdAddSelect(int             fd, /* I - File descriptor */
   }
 
 #elif defined(HAVE_POLL)
+#  ifdef HAVE_EPOLL
+  if (cupsd_epoll_fd >= 0)
+  {
+    struct epoll_event event;          /* Event data */
+
+
+    event.events = 0;
+
+    if (read_cb)
+      event.events |= EPOLLIN;
+
+    if (write_cb)
+      event.events |= EPOLLOUT;
+
+    event.data.ptr = fdptr;
+
+    if (epoll_ctl(cupsd_epoll_fd, added ? EPOLL_CTL_ADD : EPOLL_CTL_MOD, fd,
+                  &event))
+    {
+      close(cupsd_epoll_fd);
+      cupsd_epoll_fd       = -1;
+      cupsd_update_pollfds = 1;
+    }
+  }
+  else
+#  endif /* HAVE_EPOLL */
+
   cupsd_update_pollfds = 1;
 
 #else /* select() */
@@ -400,7 +412,7 @@ cupsdAddSelect(int             fd,  /* I - File descriptor */
     FD_CLR(fd, &cupsd_global_output);
     FD_CLR(fd, &cupsd_current_output);
   }
-#endif /* HAVE_EPOLL */
+#endif /* HAVE_KQUEUE */
 
  /*
   * Save the (new) read and write callbacks...
@@ -423,48 +435,7 @@ cupsdDoSelect(long timeout)                /* I - Timeout in seconds */
 {
   int                  nfds;           /* Number of file descriptors */
   _cupsd_fd_t          *fdptr;         /* Current file descriptor */
-#ifdef HAVE_EPOLL
-  int                  i;              /* Looping var */
-  struct epoll_event   *event;         /* Current event */
-
-
-  cupsdLogMessage(CUPSD_LOG_DEBUG2,
-                  "cupsdDoSelect: polling %d fds for %ld seconds...",
-                 cupsArrayCount(cupsd_fds), timeout);
-
-  if (timeout >= 0 && timeout < 86400)
-    nfds = epoll_wait(cupsd_epoll_fd, cupsd_epoll_events, MaxFDs,
-                      timeout * 1000);
-  else
-    nfds = epoll_wait(cupsd_epoll_fd, cupsd_epoll_events, MaxFDs, -1);
-
-  cupsdLogMessage(CUPSD_LOG_DEBUG2, "cupsdDoSelect: epoll() returned %d...",
-                  nfds);
-
-  for (i = nfds, event = cupsd_epoll_events; i > 0; i --, event ++)
-  {
-    fdptr = (_cupsd_fd_t *)event->data.ptr;
-
-    retain_fd(fdptr);
-
-    if (fdptr->read_cb && (event->events & (EPOLLIN | EPOLLERR | EPOLLHUP)))
-    {
-      cupsdLogMessage(CUPSD_LOG_DEBUG2, "cupsdDoSelect: Read on fd %d...",
-                     fdptr->fd);
-      (*(fdptr->read_cb))(fdptr->data);
-    }
-
-    if (fdptr->write_cb && (event->events & (EPOLLOUT | EPOLLERR | EPOLLHUP)))
-    {
-      cupsdLogMessage(CUPSD_LOG_DEBUG2, "cupsdDoSelect: Write on fd %d...",
-                     fdptr->fd);
-      (*(fdptr->write_cb))(fdptr->data);
-    }
-
-    release_fd(fdptr);
-  }
-
-#elif defined(HAVE_KQUEUE)
+#ifdef HAVE_KQUEUE
   int                  i;              /* Looping var */
   struct kevent                *event;         /* Current event */
   struct timespec      ktimeout;       /* kevent() timeout */
@@ -474,6 +445,8 @@ cupsdDoSelect(long timeout)         /* I - Timeout in seconds */
                   "cupsdDoSelect: polling %d fds for %ld seconds...",
                  cupsArrayCount(cupsd_fds), timeout);
 
+  cupsd_in_select = 1;
+
   if (timeout >= 0 && timeout < 86400)
   {
     ktimeout.tv_sec  = timeout;
@@ -495,6 +468,9 @@ cupsdDoSelect(long timeout)         /* I - Timeout in seconds */
   {
     fdptr = (_cupsd_fd_t *)event->udata;
 
+    if (cupsArrayFind(cupsd_inactive_fds, fdptr))
+      continue;
+
     cupsdLogMessage(CUPSD_LOG_DEBUG2, "event->filter=%d, event->ident=%d",
                     event->filter, (int)event->ident);
 
@@ -522,6 +498,66 @@ cupsdDoSelect(long timeout)                /* I - Timeout in seconds */
   int                  count;          /* Number of file descriptors */
 
 
+  cupsdLogMessage(CUPSD_LOG_DEBUG2,
+                  "cupsdDoSelect: polling %d fds for %ld seconds...",
+                 cupsArrayCount(cupsd_fds), timeout);
+
+#  ifdef HAVE_EPOLL
+  cupsd_in_select = 1;
+
+  if (cupsd_epoll_fd >= 0)
+  {
+    int                        i;              /* Looping var */
+    struct epoll_event *event;         /* Current event */
+
+
+    if (timeout >= 0 && timeout < 86400)
+      nfds = epoll_wait(cupsd_epoll_fd, cupsd_epoll_events, MaxFDs,
+                       timeout * 1000);
+    else
+      nfds = epoll_wait(cupsd_epoll_fd, cupsd_epoll_events, MaxFDs, -1);
+
+    cupsdLogMessage(CUPSD_LOG_DEBUG2, "cupsdDoSelect: epoll() returned %d...",
+                    nfds);
+
+    if (nfds < 0 && errno != EINTR)
+    {
+      close(cupsd_epoll_fd);
+      cupsd_epoll_fd = -1;
+    }
+    else
+    {
+      for (i = nfds, event = cupsd_epoll_events; i > 0; i --, event ++)
+      {
+       fdptr = (_cupsd_fd_t *)event->data.ptr;
+
+       if (cupsArrayFind(cupsd_inactive_fds, fdptr))
+         continue;
+
+       retain_fd(fdptr);
+
+       if (fdptr->read_cb && (event->events & (EPOLLIN | EPOLLERR | EPOLLHUP)))
+       {
+         cupsdLogMessage(CUPSD_LOG_DEBUG2, "cupsdDoSelect: Read on fd %d...",
+                         fdptr->fd);
+         (*(fdptr->read_cb))(fdptr->data);
+       }
+
+       if (fdptr->write_cb && (event->events & (EPOLLOUT | EPOLLERR | EPOLLHUP)))
+       {
+         cupsdLogMessage(CUPSD_LOG_DEBUG2, "cupsdDoSelect: Write on fd %d...",
+                         fdptr->fd);
+         (*(fdptr->write_cb))(fdptr->data);
+       }
+
+       release_fd(fdptr);
+      }
+
+      goto release_inactive;
+    }
+  }
+#  endif /* HAVE_EPOLL */
+
   count = cupsArrayCount(cupsd_fds);
 
   if (cupsd_update_pollfds)
@@ -700,7 +736,27 @@ cupsdDoSelect(long timeout)                /* I - Timeout in seconds */
     }
   }
 
-#endif /* HAVE_EPOLL */
+#endif /* HAVE_KQUEUE */
+
+#if defined(HAVE_EPOLL) || defined(HAVE_KQUEUE)
+ /*
+  * Release all inactive file descriptors...
+  */
+
+#  ifndef HAVE_KQUEUE
+  release_inactive:
+#  endif /* !HAVE_KQUEUE */
+
+  cupsd_in_select = 0;
+
+  for (fdptr = (_cupsd_fd_t *)cupsArrayFirst(cupsd_inactive_fds);
+       fdptr;
+       fdptr = (_cupsd_fd_t *)cupsArrayNext(cupsd_inactive_fds))
+  {
+    cupsArrayRemove(cupsd_inactive_fds, fdptr);
+    release_fd(fdptr);
+  }
+#endif /* HAVE_EPOLL || HAVE_KQUEUE */
 
  /*
   * Return the number of file descriptors handled...
@@ -759,7 +815,12 @@ cupsdRemoveSelect(int fd)          /* I - File descriptor */
     return;
 
 #ifdef HAVE_EPOLL
-  epoll_ctl(cupsd_epoll_fd, EPOLL_CTL_DEL, fd, &event);
+  if (epoll_ctl(cupsd_epoll_fd, EPOLL_CTL_DEL, fd, &event))
+  {
+    close(cupsd_epoll_fd);
+    cupsd_epoll_fd       = -1;
+    cupsd_update_pollfds = 1;
+  }
 
 #elif defined(HAVE_KQUEUE)
   timeout.tv_sec  = 0;
@@ -810,10 +871,18 @@ cupsdRemoveSelect(int fd)         /* I - File descriptor */
 #endif /* HAVE_EPOLL */
 
  /*
-  * Remove the file descriptor for from the FD array...
+  * Remove the file descriptor from the active array and add to the
+  * inactive array (or release, if we don't need the inactive array...)
   */
 
   cupsArrayRemove(cupsd_fds, fdptr);
+
+#if defined(HAVE_EPOLL) || defined(HAVE_KQUEUE)
+  if (cupsd_in_select)
+    cupsArrayAdd(cupsd_inactive_fds, fdptr);
+  else
+#endif /* HAVE_EPOLL || HAVE_KQUEUE */
+
   release_fd(fdptr);
 }
 
@@ -827,9 +896,14 @@ cupsdStartSelect(void)
 {
   cupsd_fds = cupsArrayNew((cups_array_func_t)compare_fds, NULL);
 
+#if defined(HAVE_EPOLL) || defined(HAVE_KQUEUE)
+  cupsd_inactive_fds = cupsArrayNew((cups_array_func_t)compare_fds, NULL);
+#endif /* HAVE_EPOLL || HAVE_KQUEUE */
+
 #ifdef HAVE_EPOLL
-  cupsd_epoll_fd     = epoll_create(MaxFDs);
-  cupsd_epoll_events = calloc(MaxFDs, sizeof(struct epoll_event));
+  cupsd_epoll_fd       = epoll_create(MaxFDs);
+  cupsd_epoll_events   = calloc(MaxFDs, sizeof(struct epoll_event));
+  cupsd_update_pollfds = 0;
 
 #elif defined(HAVE_KQUEUE)
   cupsd_kqueue_fd      = kqueue();
@@ -864,20 +938,12 @@ cupsdStopSelect(void)
   cupsArrayDelete(cupsd_fds);
   cupsd_fds = NULL;
 
-#ifdef HAVE_EPOLL
-  if (cupsd_epoll_events)
-  {
-    free(cupsd_epoll_events);
-    cupsd_epoll_events = NULL;
-  }
+#if defined(HAVE_EPOLL) || defined(HAVE_KQUEUE)
+  cupsArrayDelete(cupsd_inactive_fds);
+  cupsd_inactive_fds = NULL;
+#endif /* HAVE_EPOLL || HAVE_KQUEUE */
 
-  if (cupsd_epoll_fd >= 0)
-  {
-    close(cupsd_epoll_fd);
-    cupsd_epoll_fd = -1;
-  }
-
-#elif defined(HAVE_KQUEUE)
+#ifdef HAVE_KQUEUE
   if (cupsd_kqueue_events)
   {
     free(cupsd_kqueue_events);
@@ -893,6 +959,20 @@ cupsdStopSelect(void)
   cupsd_kqueue_changes = 0;
 
 #elif defined(HAVE_POLL)
+#  ifdef HAVE_EPOLL
+  if (cupsd_epoll_events)
+  {
+    free(cupsd_epoll_events);
+    cupsd_epoll_events = NULL;
+  }
+
+  if (cupsd_epoll_fd >= 0)
+  {
+    close(cupsd_epoll_fd);
+    cupsd_epoll_fd = -1;
+  }
+#  endif /* HAVE_EPOLL */
+
   if (cupsd_pollfds)
   {
     free(cupsd_pollfds);
@@ -944,5 +1024,5 @@ find_fd(int fd)                            /* I - File descriptor */
 
 
 /*
- * End of "$Id: select.c 6376 2007-03-21 06:39:10Z mike $".
+ * End of "$Id: select.c 7093 2007-11-30 19:09:36Z mike $".
  */