Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
Corrections and improvements to the new network implementation for SIMIX
[simgrid.git] / src / simix / smx_network.c
index 5f40e44..4ba18f5 100644 (file)
@@ -24,7 +24,6 @@ smx_rdv_t SIMIX_rdv_create(char *name)
   rvp->name = xbt_strdup(name);
   rvp->read = SIMIX_mutex_init();
   rvp->write = SIMIX_mutex_init();
-  rvp->comm_mutex = SIMIX_mutex_init();
   rvp->comm_fifo = xbt_fifo_new();
 
   return rvp;
@@ -39,7 +38,6 @@ void SIMIX_rdv_destroy(smx_rdv_t rvp)
   xbt_free(rvp->name);
   SIMIX_mutex_destroy(rvp->read);
   SIMIX_mutex_destroy(rvp->write);
-  SIMIX_mutex_destroy(rvp->comm_mutex);
   xbt_fifo_free(rvp->comm_fifo);
   xbt_free(rvp);
 }
@@ -59,30 +57,31 @@ static inline void SIMIX_rdv_push(smx_rdv_t rvp, smx_comm_t comm)
  *  \brief Checks if there is a communication request queued in a rendez-vous matching our needs
  *  \param rvp The rendez-vous with the queue
  *  \param look_for_src boolean. True: we are receiver looking for sender; False: other way round
- *  \return The communication request if found, a newly created one otherwise.
+ *  \return The communication request if found, or a newly created one otherwise.
  */
 smx_comm_t SIMIX_rdv_get_request_or_create(smx_rdv_t rvp, int look_for_src) {
-  /* Get a communication request from the rendez-vous queue. If it is a receive
-     request then return it, otherwise put it again in the queue and return NULL
+  /* Get a communication request from the rendez-vous queue. If it is the kind
+     of request we are looking for then return it, otherwise put it again in the
+     queue.
    */
-  smx_comm_t comm_head = xbt_fifo_shift(rvp->comm_fifo);
+  smx_comm_t comm = xbt_fifo_shift(rvp->comm_fifo);
 
-  if(comm_head != NULL) {
-    if (( look_for_src && comm_head->src_host != NULL) ||
-        (!look_for_src && comm_head->dst_host != NULL)) {
+  if(comm != NULL) {
+    if (( look_for_src && comm->src_host != NULL) ||
+        (!look_for_src && comm->dst_host != NULL)) {
 
-      SIMIX_communication_use(comm_head);
-      return comm_head;
+      SIMIX_communication_use(comm);
+      return comm;
     }
   }
-  xbt_fifo_unshift(rvp->comm_fifo, comm_head);
+  xbt_fifo_unshift(rvp->comm_fifo, comm);
 
-  /* no relevant peer found. Create a new comm action and set it up */
-  comm_head = SIMIX_communication_new(rvp);
-  SIMIX_rdv_push(rvp, comm_head);
-  SIMIX_communication_use(comm_head);
+  /* no relevant request found. Create a new comm action and set it up */
+  comm = SIMIX_communication_new(rvp);
+  SIMIX_rdv_push(rvp, comm);
+  SIMIX_communication_use(comm);
 
-  return comm_head;
+  return comm;
 }
 
 /******************************************************************************/
@@ -100,7 +99,6 @@ smx_comm_t SIMIX_communication_new(smx_rdv_t rdv)
   /* alloc structures */
   smx_comm_t comm = xbt_new0(s_smx_comm_t, 1);
   comm->cond = SIMIX_cond_init();
-
   comm->rdv = rdv;
 
   return comm;
@@ -117,9 +115,6 @@ void SIMIX_communication_destroy(smx_comm_t comm)
     if(comm->act != NULL)
       SIMIX_action_destroy(comm->act);
 
-    if(comm->data != NULL)
-      xbt_free(comm->data);
-
     xbt_free(comm->cond);
     xbt_free(comm);
   }
@@ -140,57 +135,76 @@ static inline void SIMIX_communication_use(smx_comm_t comm)
 
 /**
  *  \brief Start the simulation of a communication request
- *  \param comm The communicatino request
+ *  \param comm The communication request
  */
 static inline void SIMIX_communication_start(smx_comm_t comm)
 {
   comm->act = SIMIX_action_communicate(comm->src_host, comm->dst_host, NULL, 
-      comm->data_size, comm->rate);
+      comm->task_size, comm->rate);
 
   SIMIX_register_action_to_condition(comm->act, comm->cond);
 }
 
-static inline void SIMIX_communication_wait_for_completion(smx_comm_t comm)
+/**
+ *  \brief Waits for communication completion and performs error checking
+ *  \param comm The communication
+ *  \param timeout The max amount of time to wait for the communication to finish
+ *
+ *  Throws:
+ *   - host_error if peer failed
+ *   - timeout_error if communication reached the timeout specified
+ *   - network_error if network failed or peer issued a timeout
+ */
+static inline void SIMIX_communication_wait_for_completion(smx_comm_t comm, double timeout)
 {
-  __SIMIX_cond_wait(comm->cond);
-  if (comm->act) {
-    SIMIX_unregister_action_to_condition(comm->act, comm->cond);
-    SIMIX_action_destroy(comm->act);
-    comm->act=NULL;
+  xbt_ex_t e;
+  
+  if(timeout > 0){
+    TRY{
+      SIMIX_cond_wait_timeout(comm->cond, NULL, timeout);
+    }
+    CATCH(e){
+      /* If it's a timeout then cancel the communication and signal the other peer */
+      if(e.category == timeout_error)
+        SIMIX_action_cancel(comm->act);
+        SIMIX_cond_signal(comm->cond);   
+      RETHROW;
+    }
+  }else{
+    SIMIX_cond_wait(comm->cond, NULL);
   }
+
+  SIMIX_unregister_action_to_condition(comm->act, comm->cond);
+  
   /* Check for errors */
   if (SIMIX_host_get_state(comm->dst_host) == 0){
     THROW1(host_error, 0, "Destination host %s failed", comm->dst_host->name);
   } else if (SIMIX_host_get_state(comm->src_host) == 0){
     THROW1(host_error, 0, "Source host %s failed", comm->src_host->name);
-  } else if (SIMIX_action_get_state(comm->act) == SURF_ACTION_FAILED) {
+  } else if (SIMIX_action_get_state(comm->act) == SURF_ACTION_FAILED){
     THROW0(network_error, 0, "Link failure");
   }
-
 }
+
 /******************************************************************************/
 /*                        Synchronous Communication                           */
 /******************************************************************************/
-/* Throws:
- *  - host_error if peer failed
- *  - network_error if network failed */
-
-void SIMIX_network_send(smx_rdv_t rdv, void *data, size_t size, double timeout, double rate)
+/*  Throws:
+ *   - host_error if peer failed
+ *   - timeout_error if communication reached the timeout specified
+ *   - network_error if network failed or peer issued a timeout
+ */
+void SIMIX_network_send(smx_rdv_t rdv, double task_size, double rate, double timeout, void *data, size_t data_size)
 {
-  /*double start_time = SIMIX_get_clock();*/
-  void *smx_net_data;
   smx_comm_t comm;
 
-  /* Copy the message to the network */
-  /*FIXME here the MC should allocate the space from the network storage area */
-  smx_net_data = xbt_malloc(size);
-  memcpy(smx_net_data, data, size);
-
-  comm = SIMIX_rdv_get_request_or_create(rdv,0);
+  /* Setup communication request */
+  comm = SIMIX_rdv_get_request_or_create(rdv, 0);
   comm->src_host = SIMIX_host_self();
-  comm->data = smx_net_data;
-  comm->data_size = size;
+  comm->task_size = task_size;
   comm->rate = rate;
+  comm->data = data;
+  comm->data_size = data_size;
 
   /* If the receiver is already there, start the communication */
   /* If it's not here already, it will start that communication itself later on */
@@ -200,20 +214,25 @@ void SIMIX_network_send(smx_rdv_t rdv, void *data, size_t size, double timeout,
   /* Wait for communication completion */
   /* FIXME: if the semantic is non blocking, it shouldn't wait on the condition here */
   /* FIXME: add timeout checking stuff */
-  SIMIX_communication_wait_for_completion(comm);
+  SIMIX_communication_wait_for_completion(comm, timeout);
 
   SIMIX_communication_destroy(comm);
 }
 
-void SIMIX_network_recv(smx_rdv_t rdv, void **data, size_t *size, double timeout)
+/*  Throws:
+ *   - host_error if peer failed
+ *   - timeout_error if communication reached the timeout specified
+ *   - network_error if network failed or peer issued a timeout
+ */
+void SIMIX_network_recv(smx_rdv_t rdv, double timeout, void **data, size_t *data_size)
 {
-  /*double start_time = SIMIX_get_clock();*/
   smx_comm_t comm;
-  smx_host_t my_host = SIMIX_host_self();
 
-  comm = SIMIX_rdv_get_request_or_create(rdv,1);
+  /* Setup communication request */
+  comm = SIMIX_rdv_get_request_or_create(rdv, 1);
   comm->dst_host = SIMIX_host_self();
   comm->dest_buff = data;
+  comm->dest_buff_size = data_size;
 
   /* If the sender is already there, start the communication */
   /* If it's not here already, it will start that communication itself later on */
@@ -223,11 +242,7 @@ void SIMIX_network_recv(smx_rdv_t rdv, void **data, size_t *size, double timeout
   /* Wait for communication completion */
   /* FIXME: if the semantic is non blocking, it shouldn't wait on the condition here */
   /* FIXME: add timeout checking stuff */
-  SIMIX_communication_wait_for_completion(comm);
-
-  /* We are OK, let's copy the message to receiver's buffer */
-  *size = *size < comm->data_size ? *size : comm->data_size;
-  memcpy(*data, comm->data, *size);
+  SIMIX_communication_wait_for_completion(comm, timeout);
 
   SIMIX_communication_destroy(comm);
 }
@@ -257,6 +272,10 @@ XBT_PUBLIC(int) SIMIX_network_test(smx_action_t comm)
 
 
 
+  /* FIXME: MOVE DATA COPY TO MAESTRO AT ACTION SIGNAL TIME 
+   We are OK, let's copy the message to receiver's buffer 
+  *size = *size < comm->data_size ? *size : comm->data_size;
+  memcpy(*data, comm->data, *size);*/