1 files changed, 15 insertions, 1 deletions

diff --git a/drivers/lightnvm/pblk-recovery.c b/drivers/lightnvm/pblk-recovery.c
index 017874e03253..357e52980f2f 100644
--- a/drivers/lightnvm/pblk-recovery.c
+++ b/drivers/lightnvm/pblk-recovery.c
@@ -93,10 +93,24 @@ static int pblk_recov_l2p_from_emeta(struct pblk *pblk, struct pblk_line *line)
 static void pblk_update_line_wp(struct pblk *pblk, struct pblk_line *line,
                                 u64 written_secs)
 {
+        struct pblk_line_mgmt *l_mg = &pblk->l_mg;
         int i;
 
         for (i = 0; i < written_secs; i += pblk->min_write_pgs)
-                pblk_alloc_page(pblk, line, pblk->min_write_pgs);
+                __pblk_alloc_page(pblk, line, pblk->min_write_pgs);
+
+        spin_lock(&l_mg->free_lock);
+        if (written_secs > line->left_msecs) {
+                /*
+                 * We have all data sectors written
+                 * and some emeta sectors written too.
+                 */
+                line->left_msecs = 0;
+        } else {
+                /* We have only some data sectors written. */
+                line->left_msecs -= written_secs;
+        }
+        spin_unlock(&l_mg->free_lock);
 }
 
 static u64 pblk_sec_in_open_line(struct pblk *pblk, struct pblk_line *line)