Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
Adapt to upcomming changes in MSG even before Arnaud's done with implementing
[simgrid.git] / src / gras / Transport / sg_transport.c
index b17f5fc..c261a05 100644 (file)
@@ -2,18 +2,17 @@
 
 /* sg_transport - SG specific functions for transport                       */
 
-/* Authors: Martin Quinson                                                  */
-/* Copyright (C) 2004 Martin Quinson.                                       */
+/* Copyright (c) 2004 Martin Quinson. All rights reserved.                  */
 
 /* This program is free software; you can redistribute it and/or modify it
  under the terms of the license (GNU LGPL) which comes with this package. */
* under the terms of the license (GNU LGPL) which comes with this package. */
 
-#include "Transport/transport_private.h"
-#include <msg.h>
-#include "Virtu/virtu_sg.h"
+#include "gras/Transport/transport_private.h"
+#include "msg/msg.h"
+#include "gras/Virtu/virtu_sg.h"
 
-GRAS_LOG_EXTERNAL_CATEGORY(transport);
-GRAS_LOG_DEFAULT_CATEGORY(transport);
+XBT_LOG_EXTERNAL_CATEGORY(transport);
+XBT_LOG_DEFAULT_CATEGORY(transport);
 
 /**
  * gras_trp_select:
@@ -26,17 +25,17 @@ GRAS_LOG_DEFAULT_CATEGORY(transport);
  *
  * if timeout>0 and no message there, wait at most that amount of time before giving up.
  */
-gras_error_t 
+xbt_error_t 
 gras_trp_select(double timeout, 
-               gras_socket_t **dst) {
+               gras_socket_t *dst) {
 
-  gras_error_t errcode;
-  double startTime=gras_time();
-  gras_procdata_t *pd=gras_procdata_get();
+  xbt_error_t errcode;
+  double startTime=gras_os_time();
+  gras_trp_procdata_t pd=(gras_trp_procdata_t)gras_libdata_get("gras_trp");
   gras_trp_sg_sock_data_t *sockdata;
-  gras_trp_plugin_t *trp;
+  gras_trp_plugin_t trp;
 
-  gras_socket_t *sock_iter; /* iterating over all sockets */
+  gras_socket_t sock_iter; /* iterating over all sockets */
   int cursor,cpt;
 
   gras_sg_portrec_t pr;     /* iterating to find the chanel of expeditor */
@@ -44,113 +43,108 @@ gras_trp_select(double timeout,
   int r_pid;
   gras_hostdata_t *remote_hd;
 
-  DEBUG3("select on %s@%s with timeout=%d",
+  DEBUG3("select on %s@%s with timeout=%f",
         MSG_process_get_name(MSG_process_self()),
         MSG_host_get_name(MSG_host_self()),
         timeout);
-  do {
-    r_pid = MSG_task_probe_from((m_channel_t) pd->chan);
-    if (r_pid >= 0) {
-      /* Try to reuse an already openned socket to that expeditor */
-      gras_dynar_foreach(pd->sockets,cursor,sock_iter) {
-       DEBUG1("Consider %p as outgoing socket to expeditor",sock_iter);
-       sockdata = sock_iter->data;
-
-       if (sock_iter->raw || !sock_iter->outgoing)
-         continue;
-
-       if (sockdata->from_PID == r_pid) {
-         *dst=sock_iter;
-         return no_error;
-       }
-      }
-
-      /* Socket to expeditor not created yet */
-      DEBUG0("Create a socket to the expeditor");
-
-      TRY(gras_trp_plugin_get_by_name("sg",&trp));
-
-      TRY(gras_trp_socket_new(1,dst));
-      (*dst)->plugin   = trp;
-
-      (*dst)->incoming  = 1;
-      (*dst)->outgoing  = 1;
-      (*dst)->accepting = 0;
-      (*dst)->sd        = -1;
-
-      (*dst)->port      = -1;
-
-      if (!(sockdata = malloc(sizeof(gras_trp_sg_sock_data_t))))
-       RAISE_MALLOC;
-
-      sockdata->from_PID = MSG_process_self_PID();
-      sockdata->to_PID   = r_pid;
-      sockdata->to_host  = MSG_process_get_host(MSG_process_from_PID(r_pid));
-      (*dst)->data = sockdata;
 
-      (*dst)->peer_name = strdup(MSG_host_get_name(sockdata->to_host));
-
-      remote_hd=(gras_hostdata_t *)MSG_host_get_data(sockdata->to_host);
-      gras_assert0(remote_hd,"Run gras_process_init!!");
-
-      sockdata->to_chan = -1;
-      (*dst)->peer_port = -10;
-      for (cursor=0; cursor<GRAS_MAX_CHANNEL; cursor++) {
-       if (remote_hd->proc[cursor] == r_pid) {
-         sockdata->to_chan = cursor;
-         DEBUG2("Chan %d on %s is for my pal",
-                cursor,(*dst)->peer_name);
-
-         gras_dynar_foreach(remote_hd->ports, cpt, pr) {
-           if (sockdata->to_chan == pr.tochan) {
-             if (pr.raw) {
-               DEBUG0("Damn, it's raw");
-               continue;
-             }
-
-             (*dst)->peer_port = pr.port;
-             DEBUG1("Cool, it points to port %d", pr.port);
-             break;
-           } else {
-             DEBUG2("Wrong port (tochan=%d, looking for %d)\n",
-                    pr.tochan,sockdata->to_chan);
-           }
-         }
-         if ((*dst)->peer_port == -10) {
-           /* was raw */
-           sockdata->to_chan = -1;
-         } else {
-           /* found it, don't let it override by raw */
-           break;
+  r_pid = MSG_task_select_from((m_channel_t) pd->chan, timeout);
+
+  if (r_pid < 0) {
+    DEBUG0("TIMEOUT");
+    return timeout_error;
+  }
+
+  /* Ok, got something. Open a socket back to the expeditor */
+
+  /* Try to reuse an already openned socket to that expeditor */
+  xbt_dynar_foreach(pd->sockets,cursor,sock_iter) {
+    DEBUG1("Consider %p as outgoing socket to expeditor",sock_iter);
+    sockdata = sock_iter->data;
+    
+    if (sock_iter->meas || !sock_iter->outgoing)
+      continue;
+    
+    if (sockdata->to_PID == r_pid) {
+      *dst=sock_iter;
+      return no_error;
+    }
+  }
+  
+  /* Socket to expeditor not created yet */
+  DEBUG0("Create a socket to the expeditor");
+  
+  TRY(gras_trp_plugin_get_by_name("buf",&trp));
+  
+  gras_trp_socket_new(1,dst);
+  (*dst)->plugin   = trp;
+  
+  (*dst)->incoming  = 1;
+  (*dst)->outgoing  = 1;
+  (*dst)->accepting = 0;
+  (*dst)->sd        = -1;
+  
+  (*dst)->port      = -1;
+  
+  sockdata = xbt_new(gras_trp_sg_sock_data_t,1);
+  sockdata->from_PID = MSG_process_self_PID();
+  sockdata->to_PID   = r_pid;
+  sockdata->to_host  = MSG_process_get_host(MSG_process_from_PID(r_pid));
+  (*dst)->data = sockdata;
+  gras_trp_buf_init_sock(*dst);
+  
+  (*dst)->peer_name = strdup(MSG_host_get_name(sockdata->to_host));
+  
+  remote_hd=(gras_hostdata_t *)MSG_host_get_data(sockdata->to_host);
+  xbt_assert0(remote_hd,"Run gras_process_init!!");
+  
+  sockdata->to_chan = -1;
+  (*dst)->peer_port = -10;
+  for (cursor=0; cursor<XBT_MAX_CHANNEL; cursor++) {
+    if (remote_hd->proc[cursor] == r_pid) {
+      sockdata->to_chan = cursor;
+      DEBUG2("Chan %d on %s is for my pal",
+            cursor,(*dst)->peer_name);
+      
+      xbt_dynar_foreach(remote_hd->ports, cpt, pr) {
+       if (sockdata->to_chan == pr.tochan) {
+         if (pr.meas) {
+           DEBUG0("Damn, it's for measurement");
+           continue;
          }
+         
+         (*dst)->peer_port = pr.port;
+         DEBUG1("Cool, it points to port %d", pr.port);
+         break;
+       } else {
+         DEBUG2("Wrong port (tochan=%d, looking for %d)\n",
+                pr.tochan,sockdata->to_chan);
        }
       }
-      gras_assert0(sockdata->to_chan != -1,
-                  "Got a message from a process without channel");
-
-      return no_error;
-    } else {
-      /*
-      DEBUG2("Select on %s@%s did not find anything yet",
-            MSG_process_get_name(MSG_process_self()),
-            MSG_host_get_name(MSG_host_self()));
-      */
-      // MSG_process_sleep(1);
-      MSG_process_sleep(0.01);
+      if ((*dst)->peer_port == -10) {
+       /* was for measurement */
+       sockdata->to_chan = -1;
+      } else {
+       /* found it, don't let it override by meas */
+       break;
+      }
     }
-  } while (gras_time()-startTime < timeout
-          || MSG_task_Iprobe((m_channel_t) pd->chan));
-
-  return timeout_error;
-
+  }
+  xbt_assert0(sockdata->to_chan != -1,
+             "Got a message from a process without channel");
+  
+  return no_error;
 }
 
   
 /* dummy implementations of the functions used in RL mode */
 
-gras_error_t gras_trp_tcp_setup(gras_trp_plugin_t *plug) {
+xbt_error_t gras_trp_tcp_setup(gras_trp_plugin_t plug) {
   return mismatch_error;
 }
-gras_error_t gras_trp_file_setup(gras_trp_plugin_t *plug) {
+xbt_error_t gras_trp_file_setup(gras_trp_plugin_t plug) {
   return mismatch_error;
 }
+
+
+