1 files changed, 6 insertions, 0 deletions

diff --git a/drivers/mtd/ubi/io.c b/drivers/mtd/ubi/io.c
index ed0bcb35472f..5bbd1f094f4e 100644
--- a/drivers/mtd/ubi/io.c
+++ b/drivers/mtd/ubi/io.c
@@ -859,6 +859,9 @@ int ubi_io_write_ec_hdr(struct ubi_device *ubi, int pnum,
         if (err)
                 return err;
 
+        if (ubi_dbg_power_cut(ubi, POWER_CUT_EC_WRITE))
+                return -EROFS;
+
         err = ubi_io_write(ubi, ec_hdr, pnum, 0, ubi->ec_hdr_alsize);
         return err;
 }
@@ -1106,6 +1109,9 @@ int ubi_io_write_vid_hdr(struct ubi_device *ubi, int pnum,
         if (err)
                 return err;
 
+        if (ubi_dbg_power_cut(ubi, POWER_CUT_VID_WRITE))
+                return -EROFS;
+
         p = (char *)vid_hdr - ubi->vid_hdr_shift;
         err = ubi_io_write(ubi, p, pnum, ubi->vid_hdr_aloffset,
                            ubi->vid_hdr_alsize);