1 files changed, 18 insertions, 7 deletions

diff --git a/arch/x86/kvm/mmu.c b/arch/x86/kvm/mmu.c
index 6e2d2c8f230b..dce0df8150df 100644
--- a/arch/x86/kvm/mmu.c
+++ b/arch/x86/kvm/mmu.c
@@ -4421,13 +4421,12 @@ void kvm_mmu_invalidate_mmio_sptes(struct kvm *kvm)
         }
 }
 
-static int mmu_shrink(struct shrinker *shrink, struct shrink_control *sc)
+static unsigned long
+mmu_shrink_scan(struct shrinker *shrink, struct shrink_control *sc)
 {
         struct kvm *kvm;
         int nr_to_scan = sc->nr_to_scan;
-
-        if (nr_to_scan == 0)
-                goto out;
+        unsigned long freed = 0;
 
         raw_spin_lock(&kvm_lock);
 
@@ -4462,25 +4461,37 @@ static int mmu_shrink(struct shrinker *shrink, struct shrink_control *sc)
                         goto unlock;
                 }
 
-                prepare_zap_oldest_mmu_page(kvm, &invalid_list);
+                if (prepare_zap_oldest_mmu_page(kvm, &invalid_list))
+                        freed++;
                 kvm_mmu_commit_zap_page(kvm, &invalid_list);
 
 unlock:
                 spin_unlock(&kvm->mmu_lock);
                 srcu_read_unlock(&kvm->srcu, idx);
 
+                /*
+                 * unfair on small ones
+                 * per-vm shrinkers cry out
+                 * sadness comes quickly
+                 */
                 list_move_tail(&kvm->vm_list, &vm_list);
                 break;
         }
 
         raw_spin_unlock(&kvm_lock);
+        return freed;
 
-out:
+}
+
+static unsigned long
+mmu_shrink_count(struct shrinker *shrink, struct shrink_control *sc)
+{
         return percpu_counter_read_positive(&kvm_total_used_mmu_pages);
 }
 
 static struct shrinker mmu_shrinker = {
-        .shrink = mmu_shrink,
+        .count_objects = mmu_shrink_count,
+        .scan_objects = mmu_shrink_scan,
         .seeks = DEFAULT_SEEKS * 10,
 };