UBI: Fastmap: Ensure that all fastmap work is done upon WL shutdown

[firefly-linux-kernel-4.4.55.git] / drivers / mtd / ubi / wl.c

diff --git a/drivers/mtd/ubi/wl.c b/drivers/mtd/ubi/wl.c
index ae174f4ed674b20b2b90a3adeb9487f891b50c75..bf66890fefad430163e20a09cb0859dcc846352b 100644 (file)
--- a/drivers/mtd/ubi/wl.c
+++ b/drivers/mtd/ubi/wl.c
@@ -1856,6 +1856,9 @@ int ubi_thread(void *u)
  */
 static void shutdown_work(struct ubi_device *ubi)
 {
+#ifdef CONFIG_MTD_UBI_FASTMAP
+       flush_work(&ubi->fm_work);
+#endif
        while (!list_empty(&ubi->works)) {
                struct ubi_work *wrk;