KVM: MMU: cleanup for __mmu_unsync_walk()

Decrease sp->unsync_children after clear unsync_child_bitmap bit

Signed-off-by: Xiao Guangrong <xiaoguangrong@cn.fujitsu.com>
Signed-off-by: Marcelo Tosatti <mtosatti@redhat.com>

diff --git a/arch/x86/kvm/mmu.c b/arch/x86/kvm/mmu.c
index d1e09f3c5614e057bafb7e0451c4f331b2b143b4..41e801b530647079249b2c7ba670c5e17b0a3aa4 100644 (file)
--- a/arch/x86/kvm/mmu.c
+++ b/arch/x86/kvm/mmu.c
@@ -1160,9 +1160,11 @@ static int __mmu_unsync_walk(struct kvm_mmu_page *sp,
                                        return -ENOSPC;
 
                                ret = __mmu_unsync_walk(child, pvec);
-                               if (!ret)
+                               if (!ret) {
                                        __clear_bit(i, sp->unsync_child_bitmap);
-                               else if (ret > 0)
+                                       sp->unsync_children--;
+                                       WARN_ON((int)sp->unsync_children < 0);
+                               } else if (ret > 0)
                                        nr_unsync_leaf += ret;
                                else
                                        return ret;
@@ -1176,8 +1178,6 @@ static int __mmu_unsync_walk(struct kvm_mmu_page *sp,
                }
        }
 
-       if (find_first_bit(sp->unsync_child_bitmap, 512) == 512)
-               sp->unsync_children = 0;
 
        return nr_unsync_leaf;
 }