2 files changed, 45 insertions, 0 deletions

diff --git a/drivers/md/bitmap.c b/drivers/md/bitmap.c
index ecc56765d949..0a44298fb353 100644
--- a/drivers/md/bitmap.c
+++ b/drivers/md/bitmap.c
@@ -613,6 +613,7 @@ static inline unsigned long file_page_offset(unsigned long chunk)
 static inline struct page *filemap_get_page(struct bitmap *bitmap,
                                         unsigned long chunk)
 {
+        if (file_page_index(chunk) >= bitmap->file_pages) return NULL;
         return bitmap->filemap[file_page_index(chunk) - file_page_index(0)];
 }
 
@@ -739,6 +740,7 @@ static void bitmap_file_set_bit(struct bitmap *bitmap, sector_t block)
         }
 
         page = filemap_get_page(bitmap, chunk);
+        if (!page) return;
         bit = file_page_offset(chunk);
 
         /* set the bit */
@@ -1322,6 +1324,18 @@ static void bitmap_set_memory_bits(struct bitmap *bitmap, sector_t offset, int n
 
 }
 
+/* dirty the memory and file bits for bitmap chunks "s" to "e" */
+void bitmap_dirty_bits(struct bitmap *bitmap, unsigned long s, unsigned long e)
+{
+        unsigned long chunk;
+
+        for (chunk = s; chunk <= e; chunk++) {
+                sector_t sec = chunk << CHUNK_BLOCK_SHIFT(bitmap);
+                bitmap_set_memory_bits(bitmap, sec, 1);
+                bitmap_file_set_bit(bitmap, sec);
+        }
+}
+
 /*
  * flush out any pending updates
  */
diff --git a/drivers/md/md.c b/drivers/md/md.c
index 8b08043f07ef..b95dd8a183ec 100644
--- a/drivers/md/md.c
+++ b/drivers/md/md.c
@@ -2524,6 +2524,36 @@ static struct md_sysfs_entry md_new_device =
 __ATTR(new_dev, S_IWUSR, null_show, new_dev_store);
 
 static ssize_t
+bitmap_store(mddev_t *mddev, const char *buf, size_t len)
+{
+        char *end;
+        unsigned long chunk, end_chunk;
+
+        if (!mddev->bitmap)
+                goto out;
+        /* buf should be <chunk> <chunk> ... or <chunk>-<chunk> ... (range) */
+        while (*buf) {
+                chunk = end_chunk = simple_strtoul(buf, &end, 0);
+                if (buf == end) break;
+                if (*end == '-') { /* range */
+                        buf = end + 1;
+                        end_chunk = simple_strtoul(buf, &end, 0);
+                        if (buf == end) break;
+                }
+                if (*end && !isspace(*end)) break;
+                bitmap_dirty_bits(mddev->bitmap, chunk, end_chunk);
+                buf = end;
+                while (isspace(*buf)) buf++;
+        }
+        bitmap_unplug(mddev->bitmap); /* flush the bits to disk */
+out:
+        return len;
+}
+
+static struct md_sysfs_entry md_bitmap =
+__ATTR(bitmap_set_bits, S_IWUSR, null_show, bitmap_store);
+
+static ssize_t
 size_show(mddev_t *mddev, char *page)
 {
         return sprintf(page, "%llu\n", (unsigned long long)mddev->size);
@@ -2843,6 +2873,7 @@ static struct attribute *md_redundancy_attrs[] = {
         &md_sync_completed.attr,
         &md_suspend_lo.attr,
         &md_suspend_hi.attr,
+        &md_bitmap.attr,
         NULL,
 };
 static struct attribute_group md_redundancy_group = {