* thread is removed from its waitqueue and made runnable when there's a timer
  * interrupt to handle.
  */
-void kvm_timer_schedule(struct kvm_vcpu *vcpu)
+static void kvm_timer_blocking(struct kvm_vcpu *vcpu)
 {
        struct arch_timer_cpu *timer = &vcpu->arch.timer_cpu;
        struct arch_timer_context *vtimer = vcpu_vtimer(vcpu);
        struct arch_timer_context *ptimer = vcpu_ptimer(vcpu);
 
-       vtimer_save_state(vcpu);
-
-       /*
-        * No need to schedule a background timer if any guest timer has
-        * already expired, because kvm_vcpu_block will return before putting
-        * the thread to sleep.
-        */
-       if (kvm_timer_should_fire(vtimer) || kvm_timer_should_fire(ptimer))
-               return;
-
        /*
         * If both timers are not capable of raising interrupts (disabled or
         * masked), then there's no more work for us to do.
                return;
 
        /*
-        * The guest timers have not yet expired, schedule a background timer.
+        * At least one guest time will expire. Schedule a background timer.
         * Set the earliest expiration time among the guest timers.
         */
        soft_timer_start(&timer->bg_timer, kvm_timer_earliest_exp(vcpu));
 }
 
+static void kvm_timer_unblocking(struct kvm_vcpu *vcpu)
+{
+       struct arch_timer_cpu *timer = &vcpu->arch.timer_cpu;
+
+       soft_timer_cancel(&timer->bg_timer);
+}
+
 static void vtimer_restore_state(struct kvm_vcpu *vcpu)
 {
        struct arch_timer_cpu *timer = &vcpu->arch.timer_cpu;
        local_irq_restore(flags);
 }
 
-void kvm_timer_unschedule(struct kvm_vcpu *vcpu)
-{
-       struct arch_timer_cpu *timer = &vcpu->arch.timer_cpu;
-
-       vtimer_restore_state(vcpu);
-
-       soft_timer_cancel(&timer->bg_timer);
-}
-
 static void set_cntvoff(u64 cntvoff)
 {
        u32 low = lower_32_bits(cntvoff);
        /* Set the background timer for the physical timer emulation. */
        phys_timer_emulate(vcpu);
 
+       kvm_timer_unblocking(vcpu);
+
        /* If the timer fired while we weren't running, inject it now */
        if (kvm_timer_should_fire(ptimer) != ptimer->irq.level)
                kvm_timer_update_irq(vcpu, !ptimer->irq.level, ptimer);
         */
        soft_timer_cancel(&timer->phys_timer);
 
+       if (swait_active(kvm_arch_vcpu_wq(vcpu)))
+               kvm_timer_blocking(vcpu);
+
        /*
         * The kernel may decide to run userspace after calling vcpu_put, so
         * we reset cntvoff to 0 to ensure a consistent read between user