1 files changed, 5 insertions, 2 deletions

diff --git a/drivers/kvm/kvm_main.c b/drivers/kvm/kvm_main.c
index 3b046507ebc6..353e58527d15 100644
--- a/drivers/kvm/kvm_main.c
+++ b/drivers/kvm/kvm_main.c
@@ -1815,8 +1815,6 @@ static int complete_pio(struct kvm_vcpu *vcpu)
         io->count -= io->cur_count;
         io->cur_count = 0;
 
-        if (!io->count)
-                kvm_x86_ops->skip_emulated_instruction(vcpu);
         return 0;
 }
 
@@ -1876,6 +1874,8 @@ int kvm_emulate_pio (struct kvm_vcpu *vcpu, struct kvm_run *run, int in,
         memcpy(vcpu->pio_data, &vcpu->regs[VCPU_REGS_RAX], 4);
         kvm_x86_ops->decache_regs(vcpu);
 
+        kvm_x86_ops->skip_emulated_instruction(vcpu);
+
         pio_dev = vcpu_find_pio_dev(vcpu, port);
         if (pio_dev) {
                 kernel_pio(pio_dev, vcpu, vcpu->pio_data);
@@ -1938,6 +1938,9 @@ int kvm_emulate_pio_string(struct kvm_vcpu *vcpu, struct kvm_run *run, int in,
         vcpu->run->io.count = now;
         vcpu->pio.cur_count = now;
 
+        if (vcpu->pio.cur_count == vcpu->pio.count)
+                kvm_x86_ops->skip_emulated_instruction(vcpu);
+
         for (i = 0; i < nr_pages; ++i) {
                 mutex_lock(&vcpu->kvm->lock);
                 page = gva_to_page(vcpu, address + i * PAGE_SIZE);