Dirty hacks again: shot more bullets, it may be a zombie

[simgrid.git] / examples / msg / actions / actions.c

diff --git a/examples/msg/actions/actions.c b/examples/msg/actions/actions.c
index 091f7ea..52c89a4 100644 (file)
--- a/examples/msg/actions/actions.c
+++ b/examples/msg/actions/actions.c
@@ -12,6 +12,9 @@
 #include "xbt.h"                /* calloc, printf */
 #include "instr/instr_private.h"
 
+#include "msg/private.h" /* You don't want to know why, trust us */
+#include "simix/private.h"
+
 void SIMIX_ctx_raw_factory_init(smx_context_factory_t *factory);
 
 XBT_LOG_NEW_DEFAULT_CATEGORY(actions,
@@ -100,6 +103,11 @@ static void action_send(const char *const *action)
   asynchronous_cleanup();
 }
 
+static int task_recv_matching(void*ignored,void*recv_task) {
+  XBT_DEBUG("Trying to recv_match with %p", recv_task);
+  return recv_task != NULL;
+}
+
 static void action_Isend(const char *const *action)
 {
   char to[250];
@@ -110,16 +118,36 @@ static void action_Isend(const char *const *action)
 
   sprintf(to, "%s_%s", MSG_process_get_name(MSG_process_self()),action[2]);
   m_task_t task = MSG_task_create(to,0,parse_double(size),NULL);
+
+  smx_rdv_t rdv = MSG_mailbox_get_by_alias(to);
+
+  if(SIMIX_comm_has_recv_match(rdv, task_recv_matching, NULL)) {
+    XBT_DEBUG("Switching back to MSG_task_send: %s", to);
+    MSG_task_send(task, to);
+    return;
+  }
+
   msg_comm_t comm = MSG_task_isend_with_matching(task, to, /*matching madness*/NULL,task);
   xbt_dynar_push(globals->isends,&comm);
 
+  if (task->simdata->message_size < 65536) {
+    /* Close your eyes, it burns ! */
+    comm->s_comm->comm.dst_proc = SIMIX_process_get_by_name(action[2]);
+    comm->s_comm->comm.dst_buff = NULL;
+    comm->s_comm->comm.dst_buff_size = NULL;
+    comm->s_comm->comm.dst_data = NULL;
+    comm->s_comm->state = SIMIX_READY;
+    comm->s_comm->comm.refcount++;
+    SIMIX_comm_start(comm->s_comm);
+  }
+
   XBT_DEBUG("Isend on %s", MSG_process_get_name(MSG_process_self()));
   XBT_VERB("%s %f", xbt_str_join_array(action, " "), MSG_get_clock() - clock);
 
   asynchronous_cleanup();
 }
 
-static int task_matching(void*sent_task,void*ignored) {
+static int task_sent_matching(void*ignored,void*sent_task) {
   m_task_t t = (m_task_t)sent_task;
   if (t!=NULL && MSG_task_get_data_size(t)<65536)
     return 1; /* that's supposed to be already arrived */
@@ -148,11 +176,19 @@ static void action_recv(const char *const *action)
 
   /* make sure the rdv is created on need by asking to MSG instead of simix directly */
   smx_rdv_t rdv = MSG_mailbox_get_by_alias(mailbox_name);
-  smx_action_t act = SIMIX_comm_get_send_match(rdv,task_matching,NULL);
+  smx_action_t act = SIMIX_comm_get_send_match(rdv, task_sent_matching, NULL);
   if (act!=NULL){
     /* FIXME account for the memcopy time if needed */
-    SIMIX_comm_finish(act);
-    return;
+    task = act->comm.src_data;
+
+    if (task->simdata->message_size < 65536) {
+      act->comm.refcount--; /* See action_send for more pain */
+      if(act->state == SIMIX_DONE)
+        SIMIX_comm_finish(act);
+      else
+        SIMIX_req_comm_wait(act, -1.0);
+      return;
+    }
   }
 
 #ifdef HAVE_TRACING
@@ -162,7 +198,9 @@ static void action_recv(const char *const *action)
 #endif
 
   XBT_DEBUG("Receiving: %s", name);
-  MSG_task_receive(&task, mailbox_name);
+  /* Mimic a call to MSG_task_receive(&task, mailbox_name); */
+  SIMIX_req_comm_recv(rdv, &task, NULL, NULL, &task, -1.0);
+
   //  MSG_task_receive(&task, MSG_process_get_name(MSG_process_self()));
   XBT_VERB("%s %f", name, MSG_get_clock() - clock);
   MSG_task_destroy(task);