From 145e627fb31fa35087140eb8d7388af6e889823b Mon Sep 17 00:00:00 2001
From: mquinson <mquinson@48e7efb5-ca39-0410-a469-dd3cf9ba447f>
Date: Thu, 20 Jul 2006 11:48:18 +0000
Subject: [PATCH] some more paje state poping

git-svn-id: svn+ssh://scm.gforge.inria.fr/svn/simgrid/simgrid/trunk@2646 48e7efb5-ca39-0410-a469-dd3cf9ba447f
---
 src/msg/gos.c | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/src/msg/gos.c b/src/msg/gos.c
index d255d1d5ff..5951487fb5 100644
--- a/src/msg/gos.c
+++ b/src/msg/gos.c
@@ -82,6 +82,8 @@ static MSG_error_t __MSG_task_get_with_time_out_from_host(m_task_t * task,
     first_time = 0;
     if(surf_workstation_resource->extension_public->get_state(h_simdata->host) 
        == SURF_CPU_OFF)
+      PAJE_PROCESS_POP_STATE(process);
+      PAJE_COMM_STOP(process,t,channel);
       MSG_RETURN(MSG_HOST_FAILURE);
     /* OK, we should both be ready now. Are you there ? */
   }
@@ -458,6 +460,8 @@ MSG_error_t MSG_task_put_with_timeout(m_task_t task, m_host_t dest,
   while(!(task_simdata->comm)) {
     if(max_duration>0) {
       if(!first_time) {
+	PAJE_PROCESS_POP_STATE(process);
+	PAJE_COMM_STOP(process,task,channel);
 	MSG_RETURN(MSG_TRANSFER_FAILURE);
       }
     }
@@ -473,6 +477,8 @@ MSG_error_t MSG_task_put_with_timeout(m_task_t task, m_host_t dest,
        get_state(local_host->simdata->host) == SURF_CPU_OFF) {
       xbt_fifo_remove(((simdata_host_t) remote_host->simdata)->mbox[channel],
 		      task);
+      PAJE_PROCESS_POP_STATE(process);
+      PAJE_COMM_STOP(process,task,channel);
       MSG_task_destroy(task);
       MSG_RETURN(MSG_HOST_FAILURE);
     }
-- 
2.20.1