1 files changed, 23 insertions, 0 deletions

diff --git a/block/badblocks.c b/block/badblocks.c
index 6610e282a03e..6ebcef282314 100644
--- a/block/badblocks.c
+++ b/block/badblocks.c
@@ -133,6 +133,26 @@ retry:
 }
 EXPORT_SYMBOL_GPL(badblocks_check);
 
+static void badblocks_update_acked(struct badblocks *bb)
+{
+        u64 *p = bb->page;
+        int i;
+        bool unacked = false;
+
+        if (!bb->unacked_exist)
+                return;
+
+        for (i = 0; i < bb->count ; i++) {
+                if (!BB_ACK(p[i])) {
+                        unacked = true;
+                        break;
+                }
+        }
+
+        if (!unacked)
+                bb->unacked_exist = 0;
+}
+
 /**
  * badblocks_set() - Add a range of bad blocks to the table.
  * @bb:         the badblocks structure that holds all badblock information
@@ -294,6 +314,8 @@ int badblocks_set(struct badblocks *bb, sector_t s, int sectors,
         bb->changed = 1;
         if (!acknowledged)
                 bb->unacked_exist = 1;
+        else
+                badblocks_update_acked(bb);
         write_sequnlock_irqrestore(&bb->lock, flags);
 
         return rv;
@@ -401,6 +423,7 @@ int badblocks_clear(struct badblocks *bb, sector_t s, int sectors)
                 }
         }
 
+        badblocks_update_acked(bb);
         bb->changed = 1;
 out:
         write_sequnlock_irq(&bb->lock);