X-Git-Url: http://info.iut-bm.univ-fcomte.fr/pub/gitweb/simgrid.git/blobdiff_plain/5864e92472443bbdccd7f8d9caac1059529fa50c..8c8d46824d6c4860d1123a9feca25c08a2ce9511:/src/gras/Msg/msg.c

diff --git a/src/gras/Msg/msg.c b/src/gras/Msg/msg.c
index 6a4c32956a..65ffedd66f 100644
--- a/src/gras/Msg/msg.c
+++ b/src/gras/Msg/msg.c
@@ -8,6 +8,7 @@
  * under the terms of the license (GNU LGPL) which comes with this package. */
 
 #include "xbt/ex.h"
+#include "xbt/ex_interface.h"
 #include "gras/Msg/msg_private.h"
 #include "gras/Virtu/virtu_interface.h"
 #include "gras/DataDesc/datadesc_interface.h"
@@ -213,12 +214,12 @@ gras_msgtype_declare_v(const char           *name,
 			    e_gras_msg_kind_oneway, payload, NULL);
 }
 
-/** @brief retrive an existing message type from its name. */
+/** @brief retrieve an existing message type from its name. */
 gras_msgtype_t gras_msgtype_by_name (const char *name) {
   return gras_msgtype_by_namev(name,0);
 }
 
-/** @brief retrive an existing message type from its name and version. */
+/** @brief retrieve an existing message type from its name and version. */
 gras_msgtype_t gras_msgtype_by_namev(const char      *name,
 				     short int        version) {
   gras_msgtype_t res = NULL;
@@ -236,7 +237,7 @@ gras_msgtype_t gras_msgtype_by_namev(const char      *name,
   
   return res;
 }
-/** @brief retrive an existing message type from its name and version. */
+/** @brief retrieve an existing message type from its name and version. */
 gras_msgtype_t gras_msgtype_by_id(int id) {
   return (gras_msgtype_t)xbt_set_get_by_id(_gras_msgtype_set, id);
 }
@@ -269,12 +270,10 @@ gras_msg_wait_ext(double           timeout,
   gras_msg_procdata_t pd=(gras_msg_procdata_t)gras_libdata_by_id(gras_msg_libdata_id);
   int cpt;
 
-  xbt_assert0(msgt_want,"Cannot wait for the NULL message");
   xbt_assert0(msg_got,"msg_got is an output parameter");
 
-  VERB1("Waiting for message '%s'",msgt_want->name);
-
-  start = now = gras_os_time();
+  start = gras_os_time();
+  VERB1("Waiting for message '%s'",msgt_want?msgt_want->name:"(any)");
 
   xbt_dynar_foreach(pd->msg_queue,cpt,msg){
     if ( (   !msgt_want || (msg.type->code == msgt_want->code)) 
@@ -290,10 +289,30 @@ gras_msg_wait_ext(double           timeout,
   }
 
   while (1) {
+    int need_restart;
+    xbt_ex_t e;
+
+  restart_receive: /* Goto here when the receive of a message failed */
+    need_restart=0;
+    now=gras_os_time();
     memset(&msg,sizeof(msg),0);
 
-    msg.expe = gras_trp_select(timeout ? timeout - now + start : 0);
-    gras_msg_recv(msg.expe, &msg);
+    TRY {
+      msg.expe = gras_trp_select(timeout ? timeout - now + start : 0);
+      gras_msg_recv(msg.expe, &msg);
+    } CATCH(e) {
+      if (e.category == system_error &&
+	  !strncmp("Socket closed by remote side",e.msg,
+		  strlen("Socket closed by remote side"))) {
+	xbt_ex_free(e);
+	need_restart=1;
+      }	else {
+	RETHROW;
+      }
+    }
+    if (need_restart)
+      goto restart_receive;
+
     DEBUG0("Got a message from the socket");
 
     if ( (   !msgt_want || (msg.type->code == msgt_want->code)) 
@@ -311,9 +330,10 @@ gras_msg_wait_ext(double           timeout,
     xbt_dynar_push(pd->msg_queue,&msg);
     
     now=gras_os_time();
-    if (now - start + 0.001 < timeout) {
+    if (now - start + 0.001 > timeout) {
       THROW1(timeout_error,  now-start+0.001-timeout,
-	     "Timeout while waiting for msg %s",msgt_want->name);
+	     "Timeout while waiting for msg '%s'",
+	     msgt_want?msgt_want->name:"(any)");
     }
   }
 
@@ -341,6 +361,16 @@ gras_msg_wait(double           timeout,
 		    msgt_want, NULL,      NULL, NULL,
 		    &msg);
 
+  if (msgt_want->ctn_type) {
+    xbt_assert1(payload,
+		"Message type '%s' convey a payload you must accept",
+		msgt_want->name);
+  } else {
+    xbt_assert1(!payload,
+		"No payload was declared for message type '%s'",
+		msgt_want->name);
+  }
+
   if (payload) {
     memcpy(payload,msg.payl,msg.payl_size);
     free(msg.payl);
@@ -350,6 +380,61 @@ gras_msg_wait(double           timeout,
     *expeditor = msg.expe;
 }
 
+static int gras_msg_wait_or_filter(gras_msg_t msg, void *ctx) {
+  xbt_dynar_t dyn=(xbt_dynar_t)ctx;
+  int res =  xbt_dynar_member(dyn,msg->type);
+  if (res)
+    VERB1("Got matching message (type=%s)",msg->type->name);
+  else
+    VERB0("Got message not matching our expectations");
+  return res;
+}
+/** \brief Waits for a message to come in over a given socket. 
+ *
+ * @param timeout: How long should we wait for this message.
+ * @param msgt_want: a dynar containing all accepted message type
+ * @param[out] ctx: the context of received message (in case it's a RPC call we want to answer to)
+ * @param[out] msgt_got: indice in the dynar of the type of the received message 
+ * @param[out] payload: where to write the payload of the incomming message
+ * @return the error code (or no_error).
+ *
+ * Every message of a type not in the accepted list received before the one
+ * waited will be queued and used by subsequent call to this function or
+ * gras_msg_handle().
+ *
+ * If you are interested in the context, pass the address of a s_gras_msg_cb_ctx_t variable.
+ */
+void gras_msg_wait_or(double         timeout,
+                      xbt_dynar_t    msgt_want,
+		      gras_msg_cb_ctx_t *ctx,
+                      int           *msgt_got,
+                      void          *payload) {
+  s_gras_msg_t msg;
+
+  VERB1("Wait %f seconds for several message types",timeout);
+  gras_msg_wait_ext(timeout,
+		    NULL, NULL,      
+		    &gras_msg_wait_or_filter, (void*)msgt_want,
+		    &msg);
+
+  if (msg.type->ctn_type) {
+    xbt_assert1(payload,
+		"Message type '%s' convey a payload you must accept",
+		msg.type->name);
+  } /* don't check the other side since some of the types may have a payload */
+
+  if (payload && msg.type->ctn_type) {
+    memcpy(payload,msg.payl,msg.payl_size);
+    free(msg.payl);
+  }
+
+  if (ctx) 
+    *ctx=gras_msg_cb_ctx_new(msg.expe,msg.type,msg.ID,60);
+
+  if (msgt_got)
+    *msgt_got = xbt_dynar_search(msgt_want,msg.type);
+}
+
 
 /** \brief Send the data pointed by \a payload as a message of type
  * \a msgtype to the peer \a sock */
@@ -358,7 +443,49 @@ gras_msg_send(gras_socket_t   sock,
 	      gras_msgtype_t  msgtype,
 	      void           *payload) {
 
+  if (msgtype->ctn_type) {
+    xbt_assert1(payload,
+		"Message type '%s' convey a payload you must provide",
+		msgtype->name);
+  } else {
+    xbt_assert1(!payload,
+		"No payload was declared for message type '%s'",
+		msgtype->name);
+  }
+
+  DEBUG2("Send a oneway message of type '%s'. Payload=%p",
+	 msgtype->name,payload);
   gras_msg_send_ext(sock, e_gras_msg_kind_oneway,0, msgtype, payload);
+  VERB2("Sent a oneway message of type '%s'. Payload=%p",
+	msgtype->name,payload);
+}
+
+/** @brief Handle all messages arriving within the given period
+ *
+ * @param period: How long to wait for incoming messages (in seconds)
+ *
+ * Messages are dealed with just like gras_msg_handle() would do. The
+ * difference is that gras_msg_handle() handles at most one message (or wait up
+ * to timeout second when no message arrives) while this function handles any
+ * amount of messages, and lasts the given period in any case.
+ */
+void 
+gras_msg_handleall(double period) {
+  xbt_ex_t e;
+  double begin=gras_os_time();
+  double now;
+
+  do {
+    now=gras_os_time();
+    TRY{
+      if (period - now + begin > 0)
+	gras_msg_handle(period - now + begin);
+    } CATCH(e) {
+      if (e.category != timeout_error) 
+	RETHROW0("Error while waiting for messages: %s");
+      xbt_ex_free(e);
+    }
+  } while (now - begin < period);
 }
 
 /** @brief Handle an incomming message or timer (or wait up to \a timeOut seconds)
@@ -366,14 +493,15 @@ gras_msg_send(gras_socket_t   sock,
  * @param timeOut: How long to wait for incoming messages (in seconds)
  * @return the error code (or no_error).
  *
- * Messages are passed to the callbacks.
+ * Messages are passed to the callbacks. See also gras_msg_handleall().
  */
 void
 gras_msg_handle(double timeOut) {
   
   double          untiltimer;
    
-  int             cpt, ran_ok;
+  int             cpt;
+  int volatile ran_ok;
 
   s_gras_msg_t    msg;
 
@@ -423,8 +551,10 @@ gras_msg_handle(double timeOut) {
 	       e_gras_msg_kind_names[msg.kind]);
     
       } CATCH(e) {
-	RETHROW1("Error caught while receiving a message on select()ed socket %p: %s",
-	  	 msg.expe);
+	RETHROW4("Error while receiving a message on select()ed socket %p to [%s]%s:%d: %s",
+		 msg.expe,
+		 gras_socket_peer_proc(msg.expe),gras_socket_peer_name(msg.expe),
+		 gras_socket_peer_port(msg.expe));
       }
     }
   }
@@ -448,7 +578,8 @@ gras_msg_handle(double timeOut) {
 	
      } else {
 	/* select timeouted, and no timer elapsed. Nothing to do */
-       THROW0(timeout_error, 0, "No new message or timer");
+       THROW1(timeout_error, 0, "No new message or timer (delay was %f)",
+	      timeOut);
      }
      
   }
@@ -462,8 +593,9 @@ gras_msg_handle(double timeOut) {
     }
   }
   if (!list) {
-    INFO1("No callback for the incomming '%s' message. Discarded.", 
-	  msg.type->name);
+    INFO3("No callback for the incomming '%s' message (from %s:%d). Discarded.", 
+	  msg.type->name,
+	  gras_socket_peer_name(msg.expe),gras_socket_peer_port(msg.expe));
     WARN0("FIXME: gras_datadesc_free not implemented => leaking the payload");
     return;
   }
@@ -478,13 +610,14 @@ gras_msg_handle(double timeOut) {
     ran_ok=0;
     TRY {
       xbt_dynar_foreach(list->cbs,cpt,cb) { 
-	VERB3("Use the callback #%d (@%p) for incomming msg %s",
-	      cpt+1,cb,msg.type->name);
-	if ((*cb)(&ctx,msg.payl)) {
-	  /* cb handled the message */
-	  free(msg.payl);
-	  ran_ok = 1;
-	  break;
+	if (!ran_ok) {
+	  DEBUG4("Use the callback #%d (@%p) for incomming msg %s (payload_size=%d)",
+		cpt+1,cb,msg.type->name,msg.payl_size);
+	  if ((*cb)(&ctx,msg.payl)) {
+	    /* cb handled the message */
+	    free(msg.payl);
+	    ran_ok = 1;
+	  }
 	}
       }
     } CATCH(e) {
@@ -493,21 +626,21 @@ gras_msg_handle(double timeOut) {
 	/* The callback raised an exception, propagate it on the network */
 	if (!e.remote) { /* the exception is born on this machine */
 	  e.host = (char*)gras_os_myname();
-#ifdef HAVE_EXECINFO_H
-	  e.bt_strings = backtrace_symbols (e.bt, e.used);
-#endif
+	  xbt_ex_setup_backtrace(&e);
 	} 
-	gras_msg_send_ext(msg.expe, e_gras_msg_kind_rpcerror,
-			  msg.ID, msg.type, &e);
-	INFO4("Propagated %s exception from '%s' RPC cb back to %s:%d",
+	VERB5("Propagate %s exception ('%s') from '%s' RPC cb back to %s:%d",
 	      (e.remote ? "remote" : "local"),
+	      e.msg,
 	      msg.type->name,
 	      gras_socket_peer_name(msg.expe),
 	      gras_socket_peer_port(msg.expe));
-	e.host = NULL;
+	gras_msg_send_ext(msg.expe, e_gras_msg_kind_rpcerror,
+			  msg.ID, msg.type, &e);
 	xbt_ex_free(e);
-      } else
-	RETHROW;
+	ran_ok=1;
+      } else {
+	RETHROW0("Callback raised an exception: %s");
+      }
     }
     if (!ran_ok)
       THROW1(mismatch_error,0,
@@ -615,3 +748,22 @@ gras_cb_unregister(gras_msgtype_t msgtype,
 gras_socket_t gras_msg_cb_ctx_from(gras_msg_cb_ctx_t ctx) {
   return ctx->expeditor;
 }
+/* \brief Creates a new message exchange context (user should never have to) */
+gras_msg_cb_ctx_t gras_msg_cb_ctx_new(gras_socket_t expe, 
+				      gras_msgtype_t msgtype,
+				      unsigned long int ID,
+				      double timeout) {
+  gras_msg_cb_ctx_t res=xbt_new(s_gras_msg_cb_ctx_t,1);
+  res->expeditor = expe;
+  res->msgtype = msgtype;
+  res->ID = ID;
+  res->timeout = timeout;
+  return res;
+}
+/* \brief Frees a message exchange context 
+ *
+ * This function is mainly useful with \ref gras_msg_wait_or, ie seldom.
+ */
+void gras_msg_cb_ctx_free(gras_msg_cb_ctx_t ctx) {
+  free(ctx);
+}