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