1 files changed, 18 insertions, 4 deletions

diff --git a/litmus/litmus.c b/litmus/litmus.c
index 447f8e8af04d..fc63e6dee84a 100644
--- a/litmus/litmus.c
+++ b/litmus/litmus.c
@@ -386,7 +386,7 @@ out:
         return retval;
 }
 
-void litmus_exit_task(struct task_struct* tsk)
+void litmus_exit_task(struct task_struct* tsk, int free_mem)
 {
         if (is_realtime(tsk)) {
                 sched_trace_task_completion(tsk, 1);
@@ -394,8 +394,22 @@ void litmus_exit_task(struct task_struct* tsk)
                 litmus->task_exit(tsk);
 
                 BUG_ON(bheap_node_in_heap(tsk_rt(tsk)->heap_node));
-                bheap_node_free(tsk_rt(tsk)->heap_node);
-                release_heap_free(tsk_rt(tsk)->rel_heap);
+                if (free_mem) {
+                        bheap_node_free(tsk_rt(tsk)->heap_node);
+                        release_heap_free(tsk_rt(tsk)->rel_heap);
+                } else {
+                        /* XXX: in PreemptRT calling kfree here will deadlock
+                         * postponing the free when the rq->lock and the
+                         * pi_lock are released don't work (it breaks
+                         * other assumptions in the code). This is ugly,
+                         * but for now we leak memory. To avoid the leakage
+                         * the user should not reset the state of a litmus
+                         * task to a background task.
+                         */
+                        tsk_rt(tsk)->heap_node = NULL;
+                        hrtimer_cancel(&(tsk_rt(tsk)->rel_heap)->timer);
+                        tsk_rt(tsk)->rel_heap = NULL;
+                }
 
                 atomic_dec(&rt_task_count);
                 reinit_litmus_state(tsk, 1);
@@ -500,7 +514,7 @@ void exit_litmus(struct task_struct *dead_tsk)
 
         /* main cleanup only for RT tasks */
         if (is_realtime(dead_tsk))
-                litmus_exit_task(dead_tsk);
+                litmus_exit_task(dead_tsk, 1);
 }