src/gras/Transport/sg_transport.c

   1 /* $Id$ */
   2
   3 /* sg_transport - SG 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 "gras/Transport/transport_private.h"
  12 #include "msg/msg.h"
  13 #include "gras/Virtu/virtu_sg.h"
  14
  15 XBT_LOG_EXTERNAL_DEFAULT_CATEGORY(gras_trp);
  16
  17 /* check transport_private.h for an explanation of this variable; this just need to be defined to NULL in SG */
  18 gras_socket_t _gras_lastly_selected_socket = NULL;
  19
  20 /**
  21  * gras_trp_select:
  22  *
  23  * Returns the next socket to service having a message awaiting.
  24  *
  25  * if timeout<0, we ought to implement the adaptative timeout (FIXME)
  26  *
  27  * if timeout=0, do not wait for new message, only handle the ones already there.
  28  *
  29  * if timeout>0 and no message there, wait at most that amount of time before giving up.
  30  */
  31 gras_socket_t gras_trp_select(double timeout) {
  32
  33   gras_socket_t res;
  34   gras_trp_procdata_t pd =
  35     (gras_trp_procdata_t) gras_libdata_by_id(gras_trp_libdata_id);
  36   gras_trp_sg_sock_data_t *sockdata;
  37   gras_trp_plugin_t trp;
  38
  39   gras_socket_t sock_iter; /* iterating over all sockets */
  40   int cursor,cpt;
  41
  42   gras_sg_portrec_t pr;     /* iterating to find the chanel of expeditor */
  43
  44   int r_pid;
  45   gras_hostdata_t *remote_hd;
  46
  47   DEBUG3("select on %s@%s with timeout=%f",
  48          MSG_process_get_name(MSG_process_self()),
  49          MSG_host_get_name(MSG_host_self()),
  50          timeout);
  51
  52   MSG_channel_select_from((m_channel_t) pd->chan, timeout, &r_pid);
  53
  54   if (r_pid < 0) {
  55     DEBUG0("TIMEOUT");
  56     THROW0(timeout_error,0,"Timeout");
  57   }
  58
  59   /* Ok, got something. Open a socket back to the expeditor */
  60
  61   /* Try to reuse an already openned socket to that expeditor */
  62   xbt_dynar_foreach(pd->sockets,cursor,sock_iter) {
  63     DEBUG1("Consider %p as outgoing socket to expeditor",sock_iter);
  64
  65     if (sock_iter->meas || !sock_iter->outgoing)
  66       continue;
  67
  68     sockdata = sock_iter->data;
  69     if (sockdata->to_PID == r_pid) {
  70       return sock_iter;
  71     }
  72   }
  73
  74   /* Socket to expeditor not created yet */
  75   DEBUG0("Create a socket to the expeditor");
  76
  77   trp = gras_trp_plugin_get_by_name("sg");
  78
  79   gras_trp_socket_new(1,&res);
  80   res->plugin   = trp;
  81
  82   res->incoming  = 1;
  83   res->outgoing  = 1;
  84   res->accepting = 0;
  85   res->sd        = -1;
  86
  87   res->port = -1;
  88
  89   sockdata = xbt_new(gras_trp_sg_sock_data_t,1);
  90   sockdata->from_PID = MSG_process_self_PID();
  91   sockdata->to_PID   = r_pid;
  92   sockdata->to_host  = MSG_process_get_host(MSG_process_from_PID(r_pid));
  93   res->data = sockdata;
  94   gras_trp_buf_init_sock(res);
  95
  96   res->peer_name = strdup(MSG_host_get_name(sockdata->to_host));
  97
  98   remote_hd=(gras_hostdata_t *)MSG_host_get_data(sockdata->to_host);
  99   xbt_assert0(remote_hd,"Run gras_process_init!!");
 100
 101   sockdata->to_chan = -1;
 102   res->peer_port = -10;
 103   for (cursor=0; cursor<XBT_MAX_CHANNEL; cursor++) {
 104     if (remote_hd->proc[cursor] == r_pid) {
 105       sockdata->to_chan = cursor;
 106       DEBUG2("Chan %d on %s is for my pal",
 107              cursor,res->peer_name);
 108
 109       xbt_dynar_foreach(remote_hd->ports, cpt, pr) {
 110         if (sockdata->to_chan == pr.tochan) {
 111           if (pr.meas) {
 112             DEBUG0("Damn, it's for measurement");
 113             continue;
 114           }
 115
 116           res->peer_port = pr.port;
 117           DEBUG1("Cool, it points to port %d", pr.port);
 118           break;
 119         } else {
 120           DEBUG2("Wrong port (tochan=%d, looking for %d)\n",
 121                  pr.tochan,sockdata->to_chan);
 122         }
 123       }
 124       if (res->peer_port == -10) {
 125         /* was for measurement */
 126         sockdata->to_chan = -1;
 127       } else {
 128         /* found it, don't let it override by meas */
 129         break;
 130       }
 131     }
 132   }
 133   res->peer_port = ((gras_trp_procdata_t) gras_libdata_by_name_from_remote("gras_trp",MSG_process_from_PID(r_pid)))->myport;
 134   xbt_assert0(sockdata->to_chan != -1,
 135               "Got a message from a process without channel");
 136
 137   return res;
 138 }
 139
 140
 141 /* dummy implementations of the functions used in RL mode */
 142
 143 void gras_trp_tcp_setup(gras_trp_plugin_t plug) {  THROW0(mismatch_error,0,NULL); }
 144 void gras_trp_file_setup(gras_trp_plugin_t plug){  THROW0(mismatch_error,0,NULL); }
 145 void gras_trp_iov_setup(gras_trp_plugin_t plug) {  THROW0(mismatch_error,0,NULL); }
 146
 147 gras_socket_t gras_trp_buf_init_sock(gras_socket_t sock) { return sock;}
 148