sunrpc/svc_run.c

   1 /*
   2  * Copyright (c) 2010, Oracle America, Inc.
   3  * Redistribution and use in source and binary forms, with or without
   4  * modification, are permitted provided that the following conditions are
   5  * met:
   6  *
   7  *     * Redistributions of source code must retain the above copyright
   8  *       notice, this list of conditions and the following disclaimer.
   9  *     * Redistributions in binary form must reproduce the above
  10  *       copyright notice, this list of conditions and the following
  11  *       disclaimer in the documentation and/or other materials
  12  *       provided with the distribution.
  13  *     * Neither the name of the "Oracle America, Inc." nor the names of its
  14  *       contributors may be used to endorse or promote products derived
  15  *       from this software without specific prior written permission.
  16  *
  17  *   THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
  18  *   "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
  19  *   LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
  20  *   FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
  21  *   COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT,
  22  *   INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  23  *   DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE
  24  *   GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
  25  *   INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
  26  *   WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
  27  *   NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
  28  *   OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  29  *
  30  * This is the rpc server side idle loop
  31  * Wait for input, call server program.
  32  */
  33
  34 #include <errno.h>
  35 #include <unistd.h>
  36 #include <libintl.h>
  37 #include <sys/poll.h>
  38 #include <rpc/rpc.h>
  39
  40 /* This function can be used as a signal handler to terminate the
  41    server loop.  */
  42 void
  43 svc_exit (void)
  44 {
  45   free (svc_pollfd);
  46   svc_pollfd = NULL;
  47   svc_max_pollfd = 0;
  48 }
  49
  50 void
  51 svc_run (void)
  52 {
  53   int i;
  54   struct pollfd *my_pollfd = NULL;
  55   int last_max_pollfd = 0;
  56
  57   for (;;)
  58     {
  59       int max_pollfd = svc_max_pollfd;
  60       if (max_pollfd == 0 && svc_pollfd == NULL)
  61         break;
  62
  63       if (last_max_pollfd != max_pollfd)
  64         {
  65           struct pollfd *new_pollfd
  66             = realloc (my_pollfd, sizeof (struct pollfd) * max_pollfd);
  67
  68           if (new_pollfd == NULL)
  69             {
  70               perror (_("svc_run: - out of memory"));
  71               break;
  72             }
  73
  74           my_pollfd = new_pollfd;
  75           last_max_pollfd = max_pollfd;
  76         }
  77
  78       for (i = 0; i < max_pollfd; ++i)
  79         {
  80           my_pollfd[i].fd = svc_pollfd[i].fd;
  81           my_pollfd[i].events = svc_pollfd[i].events;
  82           my_pollfd[i].revents = 0;
  83         }
  84
  85       switch (i = __poll (my_pollfd, max_pollfd, -1))
  86         {
  87         case -1:
  88           if (errno == EINTR)
  89             continue;
  90           perror (_("svc_run: - poll failed"));
  91           break;
  92         case 0:
  93           continue;
  94         default:
  95           INTUSE(svc_getreq_poll) (my_pollfd, i);
  96           continue;
  97         }
  98       break;
  99     }
 100
 101   free (my_pollfd);
 102 }