From: Yu Kuai <yuku...@huawei.com>

Include following APIs:
 - llbitmap_create
 - llbitmap_resize
 - llbitmap_load
 - llbitmap_destroy

Signed-off-by: Yu Kuai <yuku...@huawei.com>
---
 drivers/md/md-bitmap.h   |   1 +
 drivers/md/md-llbitmap.c | 306 +++++++++++++++++++++++++++++++++++++++
 2 files changed, 307 insertions(+)

diff --git a/drivers/md/md-bitmap.h b/drivers/md/md-bitmap.h
index 5d579f0b0c3a..9e8bc7895751 100644
--- a/drivers/md/md-bitmap.h
+++ b/drivers/md/md-bitmap.h
@@ -22,6 +22,7 @@ typedef __u16 bitmap_counter_t;
 enum bitmap_state {
        BITMAP_STALE       = 1,  /* the bitmap file is out of date or had -EIO 
*/
        BITMAP_WRITE_ERROR = 2, /* A write error has occurred */
+       BITMAP_FIRST_USE   = 3, /* llbtimap is just created */
        BITMAP_HOSTENDIAN  =15,
 };
 
diff --git a/drivers/md/md-llbitmap.c b/drivers/md/md-llbitmap.c
index 7d4a0e81f8e1..1452887ffc5d 100644
--- a/drivers/md/md-llbitmap.c
+++ b/drivers/md/md-llbitmap.c
@@ -80,9 +80,15 @@
  *  4) resume the page;
  */
 
+#define LLBITMAP_MAJOR_HI 6
+
 #define BITMAP_MAX_SECTOR (128 * 2)
 #define BITMAP_MAX_PAGES 32
 #define BITMAP_SB_SIZE 1024
+/* 64k is the max IO size of sync IO for raid1/raid10 */
+#define MIN_CHUNK_SIZE (64 * 2)
+
+#define DEFAULT_DAEMON_SLEEP 30
 
 #define BARRIER_IDLE 5
 
@@ -177,6 +183,7 @@ struct llbitmap_bio {
 };
 
 static struct workqueue_struct *md_llbitmap_io_wq;
+static struct workqueue_struct *md_llbitmap_unplug_wq;
 
 static char state_machine[nr_llbitmap_state][nr_llbitmap_action] = {
        [BitUnwritten] = {BitDirty, BitNone, BitNone, BitNone, BitNone, 
BitNone, BitNone, BitNone},
@@ -611,3 +618,302 @@ static int llbitmap_cache_pages(struct llbitmap *llbitmap)
        return 0;
 }
 
+static int llbitmap_check_support(struct mddev *mddev)
+{
+       if (test_bit(MD_HAS_JOURNAL, &mddev->flags)) {
+               pr_notice("md/llbitmap: %s: array with journal cannot have 
bitmap\n",
+                         mdname(mddev));
+               return -EBUSY;
+       }
+
+       if (mddev->bitmap_info.space == 0) {
+               if (mddev->bitmap_info.default_space == 0) {
+                       pr_notice("md/llbitmap: %s: no space for bitmap\n",
+                                 mdname(mddev));
+                       return -ENOSPC;
+               }
+       }
+
+       if (!mddev->persistent) {
+               pr_notice("md/llbitmap: %s: array must be persistent\n",
+                         mdname(mddev));
+               return -EOPNOTSUPP;
+       }
+
+       if (mddev->bitmap_info.file) {
+               pr_notice("md/llbitmap: %s: doesn't support bitmap file\n",
+                         mdname(mddev));
+               return -EOPNOTSUPP;
+       }
+
+       if (mddev->bitmap_info.external) {
+               pr_notice("md/llbitmap: %s: doesn't support external 
metadata\n",
+                         mdname(mddev));
+               return -EOPNOTSUPP;
+       }
+
+       if (mddev_is_dm(mddev)) {
+               pr_notice("md/llbitmap: %s: doesn't support dm-raid\n",
+                         mdname(mddev));
+               return -EOPNOTSUPP;
+       }
+
+       return 0;
+}
+
+static int llbitmap_init(struct llbitmap *llbitmap)
+{
+       struct mddev *mddev = llbitmap->mddev;
+       sector_t blocks = mddev->resync_max_sectors;
+       unsigned long chunksize = MIN_CHUNK_SIZE;
+       unsigned long chunks = DIV_ROUND_UP(blocks, chunksize);
+       unsigned long space = mddev->bitmap_info.space << SECTOR_SHIFT;
+       int ret;
+
+       while (chunks > space) {
+               chunksize = chunksize << 1;
+               chunks = DIV_ROUND_UP(blocks, chunksize);
+       }
+
+       llbitmap->chunkshift = ffz(~chunksize);
+       llbitmap->chunksize = chunksize;
+       llbitmap->chunks = chunks;
+       mddev->bitmap_info.daemon_sleep = DEFAULT_DAEMON_SLEEP;
+
+       ret = llbitmap_cache_pages(llbitmap);
+       if (ret)
+               return ret;
+
+       llbitmap_state_machine(llbitmap, 0, llbitmap->chunks - 1, 
BitmapActionInit);
+       return 0;
+}
+
+static int llbitmap_read_sb(struct llbitmap *llbitmap)
+{
+       struct mddev *mddev = llbitmap->mddev;
+       unsigned long daemon_sleep;
+       unsigned long chunksize;
+       unsigned long events;
+       struct page *sb_page;
+       bitmap_super_t *sb;
+       int ret = -EINVAL;
+
+       if (!mddev->bitmap_info.offset) {
+               pr_err("md/llbitmap: %s: no super block found", mdname(mddev));
+               return -EINVAL;
+       }
+
+       sb_page = read_mapping_page(llbitmap->bitmap_file->f_mapping, 0, NULL);
+       if (IS_ERR(sb_page)) {
+               pr_err("md/llbitmap: %s: read super block failed",
+                      mdname(mddev));
+               ret = -EIO;
+               goto out;
+       }
+
+       sb = kmap_local_page(sb_page);
+       if (sb->magic != cpu_to_le32(BITMAP_MAGIC)) {
+               pr_err("md/llbitmap: %s: invalid super block magic number",
+                      mdname(mddev));
+               goto out_put_page;
+       }
+
+       if (sb->version != cpu_to_le32(LLBITMAP_MAJOR_HI)) {
+               pr_err("md/llbitmap: %s: invalid super block version",
+                      mdname(mddev));
+               goto out_put_page;
+       }
+
+       if (memcmp(sb->uuid, mddev->uuid, 16)) {
+               pr_err("md/llbitmap: %s: bitmap superblock UUID mismatch\n",
+                      mdname(mddev));
+               goto out_put_page;
+       }
+
+       if (mddev->bitmap_info.space == 0) {
+               int room = le32_to_cpu(sb->sectors_reserved);
+
+               if (room)
+                       mddev->bitmap_info.space = room;
+               else
+                       mddev->bitmap_info.space = 
mddev->bitmap_info.default_space;
+       }
+       llbitmap->flags = le32_to_cpu(sb->state);
+       if (test_and_clear_bit(BITMAP_FIRST_USE, &llbitmap->flags)) {
+               ret = llbitmap_init(llbitmap);
+               goto out_put_page;
+       }
+
+       chunksize = le32_to_cpu(sb->chunksize);
+       if (!is_power_of_2(chunksize)) {
+               pr_err("md/llbitmap: %s: chunksize not a power of 2",
+                      mdname(mddev));
+               goto out_put_page;
+       }
+
+       if (chunksize < DIV_ROUND_UP(mddev->resync_max_sectors,
+                                    mddev->bitmap_info.space << SECTOR_SHIFT)) 
{
+               pr_err("md/llbitmap: %s: chunksize too small %lu < %llu / %lu",
+                      mdname(mddev), chunksize, mddev->resync_max_sectors,
+                      mddev->bitmap_info.space);
+               goto out_put_page;
+       }
+
+       daemon_sleep = le32_to_cpu(sb->daemon_sleep);
+       if (daemon_sleep < 1 || daemon_sleep > MAX_SCHEDULE_TIMEOUT / HZ) {
+               pr_err("md/llbitmap: %s: daemon sleep %lu period out of range",
+                      mdname(mddev), daemon_sleep);
+               goto out_put_page;
+       }
+
+       if (le32_to_cpu(sb->write_behind))
+               pr_warn("md/llbitmap: %s: slow disk is not supported",
+                       mdname(mddev));
+
+       events = le64_to_cpu(sb->events);
+       if (events < mddev->events) {
+               pr_warn("md/llbitmap :%s: bitmap file is out of date (%lu < 
%llu) -- forcing full recovery",
+                       mdname(mddev), events, mddev->events);
+               set_bit(BITMAP_STALE, &llbitmap->flags);
+       }
+
+       sb->sync_size = cpu_to_le64(mddev->resync_max_sectors);
+       mddev->bitmap_info.chunksize = chunksize;
+       mddev->bitmap_info.daemon_sleep = daemon_sleep;
+
+       llbitmap->chunksize = chunksize;
+       llbitmap->chunks = DIV_ROUND_UP(mddev->resync_max_sectors, chunksize);
+       llbitmap->chunkshift = ffz(~chunksize);
+       ret = llbitmap_cache_pages(llbitmap);
+
+out_put_page:
+       put_page(sb_page);
+out:
+       kunmap_local(sb);
+       return ret;
+}
+
+static int llbitmap_create(struct mddev *mddev)
+{
+       struct llbitmap *llbitmap;
+       int ret;
+
+       ret = llbitmap_check_support(mddev);
+       if (ret)
+               return ret;
+
+       llbitmap = kzalloc(sizeof(*llbitmap), GFP_KERNEL);
+       if (!llbitmap)
+               return -ENOMEM;
+
+       llbitmap->mddev = mddev;
+       bio_list_init(&llbitmap->retry_list);
+       spin_lock_init(&llbitmap->retry_lock);
+
+       timer_setup(&llbitmap->pending_timer, llbitmap_pending_timer_fn, 0);
+       INIT_WORK(&llbitmap->retry_work, md_llbitmap_retry_fn);
+       INIT_WORK(&llbitmap->daemon_work, md_llbitmap_daemon_fn);
+
+       ret = bioset_init(&llbitmap->bio_set, BIO_POOL_SIZE,
+                         offsetof(struct llbitmap_bio, bio), 0);
+       if (ret)
+               goto err_out;
+
+       ret = llbitmap_add_disk(llbitmap);
+       if (ret)
+               goto err_bio_set;
+
+       ret = llbitmap_open_disk(llbitmap);
+       if (ret)
+               goto err_del_disk;
+
+       mutex_lock(&mddev->bitmap_info.mutex);
+       mddev->bitmap = llbitmap;
+       ret = llbitmap_read_sb(llbitmap);
+       mutex_unlock(&mddev->bitmap_info.mutex);
+       if (ret)
+               goto err_close_disk;
+
+       return 0;
+
+err_close_disk:
+       mddev->bitmap = NULL;
+       llbitmap_close_disk(llbitmap);
+err_del_disk:
+       llbitmap_del_disk(llbitmap);
+err_bio_set:
+       bioset_exit(&llbitmap->bio_set);
+err_out:
+       kfree(llbitmap);
+       return ret;
+}
+
+static int llbitmap_resize(struct mddev *mddev, sector_t blocks, int chunksize)
+{
+       struct llbitmap *llbitmap = mddev->bitmap;
+       unsigned long chunks;
+
+       if (chunksize == 0)
+               chunksize = llbitmap->chunksize;
+
+       /* If there is enough space, leave the chunksize unchanged. */
+       chunks = DIV_ROUND_UP(blocks, chunksize);
+       while (chunks > mddev->bitmap_info.space << SECTOR_SHIFT) {
+               chunksize = chunksize << 1;
+               chunks = DIV_ROUND_UP(blocks, chunksize);
+       }
+
+       llbitmap->chunkshift = ffz(~chunksize);
+       llbitmap->chunksize = chunksize;
+       llbitmap->chunks = chunks;
+
+       return 0;
+}
+
+static int llbitmap_load(struct mddev *mddev)
+{
+       enum llbitmap_action action = BitmapActionReload;
+       struct llbitmap *llbitmap = mddev->bitmap;
+
+       if (test_and_clear_bit(BITMAP_STALE, &llbitmap->flags))
+               action = BitmapActionStale;
+
+       llbitmap_state_machine(llbitmap, 0, llbitmap->chunks - 1, action);
+       return 0;
+}
+
+static void llbitmap_destroy(struct mddev *mddev)
+{
+       struct llbitmap *llbitmap = mddev->bitmap;
+
+       if (!llbitmap)
+               return;
+
+       mutex_lock(&mddev->bitmap_info.mutex);
+
+       del_timer_sync(&llbitmap->pending_timer);
+       flush_workqueue(md_llbitmap_io_wq);
+       flush_workqueue(md_llbitmap_unplug_wq);
+
+       llbitmap_del_disk(llbitmap);
+       llbitmap_close_disk(llbitmap);
+       mddev->bitmap = NULL;
+
+       llbitmap_free_pages(llbitmap);
+       bioset_exit(&llbitmap->bio_set);
+       kfree(llbitmap);
+       mutex_unlock(&mddev->bitmap_info.mutex);
+}
+
+static struct bitmap_operations llbitmap_ops = {
+       .head = {
+               .type   = MD_BITMAP,
+               .id     = ID_LLBITMAP,
+               .name   = "llbitmap",
+       },
+
+       .create                 = llbitmap_create,
+       .resize                 = llbitmap_resize,
+       .load                   = llbitmap_load,
+       .destroy                = llbitmap_destroy,
+};
-- 
2.39.2


Reply via email to