1 files changed, 4 insertions, 1 deletions

diff --git a/litmus/sched_mc2.c b/litmus/sched_mc2.c
index f8ad5c615e5d..864e1b03d924 100644
--- a/litmus/sched_mc2.c
+++ b/litmus/sched_mc2.c
@@ -148,7 +148,7 @@ asmlinkage long sys_enact_mode(void)
                 }
                 mode_changed = false;
                 if (pending){ //MCR has entered
-                        raw_spin_lock_irqsave(&state->lock,flags);
+                        raw_spin_lock_irqsave(&state->lock, flags);
                         //TRACE_CUR("LOCK_MSG: GLOBAL LOCK\n");
                         raw_spin_lock(&global_lock);
                         raw_spin_lock(&mode_lock);
@@ -602,6 +602,7 @@ static void mc2_update_timer_and_unlock(struct mc2_cpu_state *state)
         //enum crit_level lv = get_task_crit_level(state->scheduled);
         struct next_timer_event *event, *next;
         int reschedule[NR_CPUS];
+        unsigned long flags;
 
         for (cpus = 0; cpus<NR_CPUS; cpus++)
                 reschedule[cpus] = 0;
@@ -616,6 +617,7 @@ static void mc2_update_timer_and_unlock(struct mc2_cpu_state *state)
         local = local_cpu_state() == state;
 
         //TRACE_CUR("LOCK_MSG: GLOBAL LOCK\n");
+        local_irq_save(flags);
         raw_spin_lock(&global_lock);
                 
         list_for_each_entry_safe(event, next, &_global_env->next_events, list) {
@@ -654,6 +656,7 @@ static void mc2_update_timer_and_unlock(struct mc2_cpu_state *state)
          * may raise a softirq, which in turn may wake ksoftirqd. */
         //TRACE_CUR("LOCK_MSG: GLOBAL UNLOCK\n");
         raw_spin_unlock(&global_lock);
+        local_irq_restore(flags);
         raw_spin_unlock(&state->lock);
 
         if ((update <= now) || reschedule[state->cpu]) {