/* time.c */
 #define BILLION (1000 * 1000 * 1000)
 
-extern void switch_timers(int to_real);
+extern int switch_timers(int to_real);
 extern void idle_sleep(int secs);
 extern int set_interval(int is_virtual);
 extern void disable_timer(void);
 
        do {
                current->thread.saved_task = NULL;
 
-               /* XXX need to check runqueues[cpu].idle */
-               if (current->pid == 0)
-                       switch_timers(0);
-
                switch_threads(&from->thread.switch_buf,
                               &to->thread.switch_buf);
 
                arch_switch_to(current->thread.prev_sched, current);
 
-               if (current->pid == 0)
-                       switch_timers(1);
-
                if (current->thread.saved_task)
                        show_regs(&(current->thread.regs));
                next= current->thread.saved_task;
                if (need_resched())
                        schedule();
 
+               switch_timers(1);
                idle_sleep(10);
+               switch_timers(0);
        }
 }
 
 
 {
        struct uml_pt_regs regs;
 
-       if (sig == SIGALRM)
-               switch_timers(0);
-
        if (sc != NULL)
                copy_sc(®s, sc);
        regs.is_user = 0;
        unblock_signals();
        timer_handler(sig, ®s);
-
-       if (sig == SIGALRM)
-               switch_timers(1);
 }
 
 void alarm_handler(int sig, struct sigcontext *sc)
 void handle_signal(int sig, struct sigcontext *sc)
 {
        unsigned long pending = 1UL << sig;
+       int timer = switch_timers(0);
 
        do {
                int nested, bail;
                if (!nested)
                        pending = from_irq_stack(nested);
        } while (pending);
+
+       switch_timers(timer);
 }
 
 extern void hard_handler(int sig);
 
 #include "os.h"
 #include "user.h"
 
+static int is_real_timer = 0;
+
 int set_interval(int is_virtual)
 {
        int usec = 1000000/UM_HZ;
        signal(SIGVTALRM, SIG_IGN);
 }
 
-void switch_timers(int to_real)
+int switch_timers(int to_real)
 {
        struct itimerval disable = ((struct itimerval) { { 0, 0 }, { 0, 0 }});
-       struct itimerval enable = ((struct itimerval) { { 0, 1000000/UM_HZ },
-                                                       { 0, 1000000/UM_HZ }});
-       int old, new;
+       struct itimerval enable;
+       int old, new, old_type = is_real_timer;
+
+       if(to_real == old_type)
+               return to_real;
 
        if (to_real) {
                old = ITIMER_VIRTUAL;
                new = ITIMER_VIRTUAL;
        }
 
-       if ((setitimer(old, &disable, NULL) < 0) ||
-           (setitimer(new, &enable, NULL)))
-               printk(UM_KERN_ERR "switch_timers - setitimer failed, "
+       if (setitimer(old, &disable, &enable) < 0)
+               printk(UM_KERN_ERR "switch_timers - setitimer disable failed, "
+                      "errno = %d\n", errno);
+
+       if((enable.it_value.tv_sec == 0) && (enable.it_value.tv_usec == 0))
+               enable.it_value = enable.it_interval;
+
+       if (setitimer(new, &enable, NULL))
+               printk(UM_KERN_ERR "switch_timers - setitimer enable failed, "
                       "errno = %d\n", errno);
+
+       is_real_timer = to_real;
+       return old_type;
 }
 
 unsigned long long os_nsecs(void)