src/gras/Transport/rl_transport.c

   1 /* $Id$ */
   2
   3 /* rl_transport - RL specific functions for transport                       */
   4
   5 /* Copyright (c) 2004 Martin Quinson. All rights reserved.                  */
   6
   7 /* This program is free software; you can redistribute it and/or modify it
   8  * under the terms of the license (GNU LGPL) which comes with this package. */
   9
  10 #include "xbt/ex.h"
  11 #include "portable.h"
  12 #include "gras/Transport/transport_private.h"
  13 XBT_LOG_EXTERNAL_DEFAULT_CATEGORY(gras_trp);
  14
  15 /* check transport_private.h for an explanation of this variable */
  16 gras_socket_t _gras_lastly_selected_socket = NULL;
  17
  18 /**
  19  * gras_trp_select:
  20  *
  21  * Returns the next socket to service because it receives a message.
  22  *
  23  * if timeout<0, we ought to implement the adaptative timeout (FIXME)
  24  *
  25  * if timeout=0, do not wait for new message, only handle the ones already there.
  26  *
  27  * if timeout>0 and no message there, wait at most that amount of time before giving up.
  28  */
  29 gras_socket_t gras_trp_select(double timeout) {
  30   xbt_dynar_t sockets= ((gras_trp_procdata_t) gras_libdata_by_id(gras_trp_libdata_id))->sockets;
  31   int done = -1;
  32   double wakeup = gras_os_time() + timeout;
  33   double now = 0;
  34   /* nextToService used to make sure socket with high number do not starve */
  35   /*  static int nextToService = 0; */
  36   struct timeval tout, *p_tout;
  37
  38   int max_fds=0; /* first arg of select: number of existing sockets */
  39   /* but accept() of winsock returns sockets bigger than the limit, so don't bother
  40      with this tiny optimisation on BillWare */
  41   fd_set FDS;
  42   int ready; /* return of select: number of socket ready to be serviced */
  43   static int fd_setsize=-1; /* FD_SETSIZE not always defined. Get this portably */
  44
  45   gras_socket_t sock_iter; /* iterating over all sockets */
  46   int cursor;              /* iterating over all sockets */
  47
  48   /* Check whether there is more data to read from the socket we selected last time.
  49      This can happen with tcp buffered sockets since we try to get as much data as we can for them */
  50   if (_gras_lastly_selected_socket && _gras_lastly_selected_socket->moredata) {
  51      VERB0("Returning _gras_lastly_selected_socket since there is more data on it");
  52      return _gras_lastly_selected_socket;
  53   }
  54
  55   /* Compute FD_SETSIZE on need */
  56   if (fd_setsize < 0) {
  57 #ifdef HAVE_SYSCONF
  58      fd_setsize = sysconf( _SC_OPEN_MAX );
  59 #else
  60 #  ifdef HAVE_GETDTABLESIZE
  61      fd_setsize = getdtablesize();
  62 #  else
  63      fd_setsize = FD_SETSIZE;
  64 #  endif /* !USE_SYSCONF */
  65 #endif
  66   }
  67
  68   while (done == -1) {
  69     if (timeout > 0) { /* did we timeout already? */
  70       now = gras_os_time();
  71       DEBUG2("wakeup=%f now=%f",wakeup, now);
  72       if (now == -1 || now >= wakeup) {
  73         /* didn't find anything; no need to update _gras_lastly_selected_socket since its moredata is 0 (or we would have returned it directly) */
  74         THROW1(timeout_error,0,
  75                "Timeout (%f) elapsed with selecting for incomming connexions",
  76                timeout);
  77       }
  78     }
  79
  80     /* construct the set of socket to ear from */
  81     FD_ZERO(&FDS);
  82     max_fds = -1;
  83     xbt_dynar_foreach(sockets,cursor,sock_iter) {
  84       if (!sock_iter->valid)
  85          continue;
  86
  87       if (sock_iter->incoming) {
  88         DEBUG1("Considering socket %d for select",sock_iter->sd);
  89 #ifndef HAVE_WINSOCK_H
  90         if (max_fds < sock_iter->sd)
  91           max_fds = sock_iter->sd;
  92 #else
  93       max_fds = 0;
  94
  95 #endif
  96         FD_SET(sock_iter->sd, &FDS);
  97       } else {
  98         DEBUG1("Not considering socket %d for select",sock_iter->sd);
  99       }
 100     }
 101
 102     if (max_fds == -1) {
 103        if (timeout > 0) {
 104           DEBUG1("No socket to select onto. Sleep %f sec instead.",timeout);
 105           gras_os_sleep(timeout);
 106           THROW1(timeout_error,0,"No socket to select onto. Sleep %f sec instead",timeout);
 107        } else {
 108           DEBUG0("No socket to select onto. Return directly.");
 109           THROW0(timeout_error,0, "No socket to select onto. Return directly.");
 110        }
 111     }
 112
 113 #ifndef HAVE_WINSOCK_H
 114     /* we cannot have more than FD_SETSIZE sockets
 115        ... but with WINSOCK which returns sockets higher than the limit (killing this optim) */
 116     if (++max_fds > fd_setsize && fd_setsize > 0) {
 117       WARN1("too many open sockets (%d).",max_fds);
 118       done = 0;
 119       break;
 120     }
 121 #else
 122     max_fds = fd_setsize;
 123 #endif
 124
 125     if (timeout > 0) {
 126       /* set the timeout */
 127       tout.tv_sec = (unsigned long)(wakeup - now);
 128       tout.tv_usec = ((wakeup -now) - ((unsigned long)(wakeup - now))) * 1000000;
 129       p_tout = &tout;
 130     } else if (timeout == 0) {
 131       /* polling only */
 132       tout.tv_sec = 0;
 133       tout.tv_usec = 0;
 134       p_tout = &tout;
 135       /* we just do one loop around */
 136       done = 0;
 137     } else {
 138       /* no timeout: good luck! */
 139       p_tout = NULL;
 140     }
 141
 142     DEBUG2("Selecting over %d socket(s); timeout=%f", max_fds-1,timeout);
 143     ready = select(max_fds, &FDS, NULL, NULL, p_tout);
 144     DEBUG1("select returned %d", ready);
 145     if (ready == -1) {
 146       switch (errno) {
 147       case  EINTR: /* a signal we don't care about occured. we don't care */
 148         /* if we cared, we would have set an handler */
 149         continue;
 150       case EINVAL: /* invalid value */
 151         THROW3(system_error,EINVAL,"invalid select: nb fds: %d, timeout: %d.%d",
 152                max_fds, (int)tout.tv_sec,(int) tout.tv_usec);
 153       case ENOMEM:
 154         xbt_assert0(0,"Malloc error during the select");
 155       default:
 156         THROW2(system_error,errno,"Error during select: %s (%d)",
 157                strerror(errno),errno);
 158       }
 159       THROW_IMPOSSIBLE;
 160     } else if (ready == 0) {
 161       continue;  /* this was a timeout */
 162     }
 163
 164     xbt_dynar_foreach(sockets,cursor,sock_iter) {
 165        if(!FD_ISSET(sock_iter->sd, &FDS)) { /* this socket is not ready */
 166         continue;
 167        }
 168
 169        /* Got a socket to serve */
 170        ready--;
 171
 172        if (   sock_iter->accepting
 173            && sock_iter->plugin->socket_accept) {
 174          /* not a socket but an ear. accept on it and serve next socket */
 175          gras_socket_t accepted=NULL;
 176
 177          /* release mutex before accept */
 178          accepted = (sock_iter->plugin->socket_accept)(sock_iter);
 179
 180          DEBUG2("accepted=%p,&accepted=%p",accepted,&accepted);
 181          accepted->meas = sock_iter->meas;
 182
 183        } else if (sock_iter->recv_ok) {
 184          /* Make sure the socket is still alive by reading the first byte */
 185          char lookahead;
 186          int recvd;
 187
 188          recvd = recv(sock_iter->sd, &lookahead, 1, MSG_PEEK);
 189          if (recvd < 0) {
 190            WARN2("socket %d failed: %s", sock_iter->sd, strerror(errno));
 191            /* done with this socket */
 192            gras_socket_close(sock_iter);
 193            cursor--;
 194          } else if (recvd == 0) {
 195            /* Connection reset (=closed) by peer. */
 196            DEBUG1("Connection %d reset by peer", sock_iter->sd);
 197            sock_iter->valid=0; /* don't close it. User may keep references to it */
 198          } else {
 199            /* Got a suited socket ! */
 200            XBT_OUT;
 201            _gras_lastly_selected_socket = sock_iter;
 202                  /* break sync dynar iteration */
 203                  xbt_dynar_cursor_unlock(sockets);
 204            return sock_iter;
 205          }
 206
 207        } else {
 208          /* This is a file socket. Cannot recv() on it, but it must be alive */
 209            XBT_OUT;
 210            _gras_lastly_selected_socket = sock_iter;
 211                  xbt_dynar_cursor_unlock(sockets);
 212            return sock_iter;
 213        }
 214
 215
 216        /* if we're here, the socket we found wasn't really ready to be served */
 217        if (ready == 0) { /* exausted all sockets given by select. Request new ones */
 218
 219                                  xbt_dynar_cursor_unlock(sockets);
 220                                  break;
 221                          }
 222     }
 223
 224   }
 225
 226   /* No socket found. Maybe we had timeout=0 and nothing to do */
 227   DEBUG0("TIMEOUT");
 228   THROW0(timeout_error,0,"Timeout");
 229 }
 230
 231 void gras_trp_sg_setup(gras_trp_plugin_t plug) {
 232   THROW0(mismatch_error,0,"No SG transport on live platforms");
 233 }
 234