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 libc_hidden_nolink_sunrpc (svc_exit, GLIBC_2_0)
  50
  51 void
  52 svc_run (void)
  53 {
  54   int i;
  55   struct pollfd *my_pollfd = NULL;
  56   int last_max_pollfd = 0;
  57
  58   for (;;)
  59     {
  60       int max_pollfd = svc_max_pollfd;
  61       if (max_pollfd == 0 && svc_pollfd == NULL)
  62         break;
  63
  64       if (last_max_pollfd != max_pollfd)
  65         {
  66           struct pollfd *new_pollfd
  67             = realloc (my_pollfd, sizeof (struct pollfd) * max_pollfd);
  68
  69           if (new_pollfd == NULL)
  70             {
  71               perror (_("svc_run: - out of memory"));
  72               break;
  73             }
  74
  75           my_pollfd = new_pollfd;
  76           last_max_pollfd = max_pollfd;
  77         }
  78
  79       for (i = 0; i < max_pollfd; ++i)
  80         {
  81           my_pollfd[i].fd = svc_pollfd[i].fd;
  82           my_pollfd[i].events = svc_pollfd[i].events;
  83           my_pollfd[i].revents = 0;
  84         }
  85
  86       switch (i = __poll (my_pollfd, max_pollfd, -1))
  87         {
  88         case -1:
  89           if (errno == EINTR)
  90             continue;
  91           perror (_("svc_run: - poll failed"));
  92           break;
  93         case 0:
  94           continue;
  95         default:
  96           svc_getreq_poll (my_pollfd, i);
  97           continue;
  98         }
  99       break;
 100     }
 101
 102   free (my_pollfd);
 103 }
 104 #ifdef EXPORT_RPC_SYMBOLS
 105 libc_hidden_def (svc_run)
 106 #else
 107 libc_hidden_nolink_sunrpc (svc_run, GLIBC_2_0)
 108 #endif