src/gras/Transport/sg_transport.c

   1 /* sg_transport - SG specific functions for transport                       */
   2
   3 /* Copyright (c) 2004, 2005, 2006, 2007, 2009, 2010. The SimGrid Team.
   4  * All rights reserved.                                                     */
   5
   6 /* This program is free software; you can redistribute it and/or modify it
   7  * under the terms of the license (GNU LGPL) which comes with this package. */
   8
   9 #include "xbt/ex.h"
  10 #include "gras/Transport/transport_private.h"
  11 #include "gras/Virtu/virtu_sg.h"
  12
  13 XBT_LOG_EXTERNAL_DEFAULT_CATEGORY(gras_trp);
  14
  15 /* check transport_private.h for an explanation of this variable; this just need to be defined to NULL in SG */
  16 gras_socket_t _gras_lastly_selected_socket = NULL;
  17
  18 /**
  19  * gras_trp_select:
  20  *
  21  * Returns the next socket to service having a message awaiting.
  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 {
  31   gras_socket_t res;
  32   gras_trp_procdata_t pd =
  33     (gras_trp_procdata_t) gras_libdata_by_id(gras_trp_libdata_id);
  34   gras_trp_sg_sock_data_t *sockdata;
  35   gras_trp_plugin_t trp;
  36   gras_socket_t active_socket = NULL;
  37   gras_trp_sg_sock_data_t *active_socket_data;
  38   gras_socket_t sock_iter;      /* iterating over all sockets */
  39   unsigned int cursor;
  40
  41   DEBUG3("select on %s@%s with timeout=%f",
  42          SIMIX_process_get_name(SIMIX_process_self()),
  43          SIMIX_host_get_name(SIMIX_host_self()), timeout);
  44   if (timeout >= 0) {
  45     xbt_queue_shift_timed(pd->msg_selectable_sockets,
  46                           &active_socket, timeout);
  47   } else {
  48     xbt_queue_shift(pd->msg_selectable_sockets, &active_socket);
  49   }
  50
  51   if (active_socket == NULL) {
  52     DEBUG0("TIMEOUT");
  53     THROW0(timeout_error, 0, "Timeout");
  54   }
  55   active_socket_data = (gras_trp_sg_sock_data_t *) active_socket->data;
  56
  57   /* Ok, got something. Open a socket back to the expeditor */
  58
  59   /* Try to reuse an already openned socket to that expeditor */
  60   DEBUG1("Open sockets size %lu", xbt_dynar_length(pd->sockets));
  61   xbt_dynar_foreach(pd->sockets, cursor, sock_iter) {
  62     gras_trp_sg_sock_data_t *sock_data;
  63     DEBUG1("Consider %p as outgoing socket to expeditor", sock_iter);
  64
  65     if (sock_iter->meas || !sock_iter->outgoing)
  66       continue;
  67     sock_data = ((gras_trp_sg_sock_data_t *) sock_iter->data);
  68
  69     if ((sock_data->to_socket == active_socket) &&
  70         (sock_data->to_host ==
  71          SIMIX_process_get_host(active_socket_data->from_process))) {
  72       xbt_dynar_cursor_unlock(pd->sockets);
  73       return sock_iter;
  74     }
  75   }
  76
  77   /* Socket to expeditor not created yet */
  78   DEBUG0("Create a socket to the expeditor");
  79
  80   trp = gras_trp_plugin_get_by_name("sg");
  81
  82   gras_trp_socket_new(1, &res);
  83   res->plugin = trp;
  84
  85   res->incoming = 1;
  86   res->outgoing = 1;
  87   res->accepting = 0;
  88   res->sd = -1;
  89
  90   res->port = -1;
  91
  92   /* initialize the ports */
  93   //res->peer_port = active_socket->port;
  94   res->port = active_socket->peer_port;
  95
  96   /* create sockdata */
  97   sockdata = xbt_new(gras_trp_sg_sock_data_t, 1);
  98   sockdata->from_process = SIMIX_process_self();
  99   sockdata->to_process = active_socket_data->from_process;
 100
 101   res->peer_port = ((gras_trp_procdata_t)
 102                     gras_libdata_by_name_from_remote("gras_trp",
 103                                                      sockdata->to_process))->
 104     myport;
 105   sockdata->to_socket = active_socket;
 106   /*update the peer to_socket  variable */
 107   active_socket_data->to_socket = res;
 108   sockdata->cond = SIMIX_cond_init();
 109   sockdata->mutex = SIMIX_mutex_init();
 110
 111   sockdata->to_host =
 112     SIMIX_process_get_host(active_socket_data->from_process);
 113
 114   res->data = sockdata;
 115   res->peer_name = strdup(SIMIX_host_get_name(sockdata->to_host));
 116
 117   gras_trp_buf_init_sock(res);
 118
 119   DEBUG4("Create socket to process:%s(Port %d) from process: %s(Port %d)",
 120          SIMIX_process_get_name(sockdata->from_process),
 121          res->peer_port,
 122          SIMIX_process_get_name(sockdata->to_process), res->port);
 123
 124   return res;
 125 }
 126
 127
 128 /* dummy implementations of the functions used in RL mode */
 129
 130 void gras_trp_tcp_setup(gras_trp_plugin_t plug)
 131 {
 132   THROW0(mismatch_error, 0, NULL);
 133 }
 134
 135 void gras_trp_file_setup(gras_trp_plugin_t plug)
 136 {
 137   THROW0(mismatch_error, 0, NULL);
 138 }
 139
 140 void gras_trp_iov_setup(gras_trp_plugin_t plug)
 141 {
 142   THROW0(mismatch_error, 0, NULL);
 143 }
 144
 145 gras_socket_t gras_trp_buf_init_sock(gras_socket_t sock)
 146 {
 147   return sock;
 148 }