From: mingzhe <mingzhe.zou@xxxxxxxxxxxx>
When partial_stripes_expensive is false, resizing causes nr_stripes to change.
So, stripe_sectors_dirty and full_dirty_stripes memory must be reallocated.
If the device is smaller, only nr_stripes need to be modified.
Signed-off-by: mingzhe <mingzhe.zou@xxxxxxxxxxxx>
---
Changelog:
v4: Fix up overflow and null pointer
v3: Fix up errors.
v2: Fix up errors.
v1: Original verison.
---
drivers/md/bcache/bcache.h | 1 +
drivers/md/bcache/btree.c | 31 ++++++++++++++
drivers/md/bcache/btree.h | 2 +
drivers/md/bcache/super.c | 87 ++++++++++++++++++++++++++++++++++++++
drivers/md/bcache/sysfs.c | 14 ++++++
5 files changed, 135 insertions(+)
diff --git a/drivers/md/bcache/bcache.h b/drivers/md/bcache/bcache.h
index 5da991505b45..70e1f6ec12d5 100644
--- a/drivers/md/bcache/bcache.h
+++ b/drivers/md/bcache/bcache.h
@@ -1040,6 +1040,7 @@ void bcache_write_super(struct cache_set *c);
int bch_flash_dev_create(struct cache_set *c, uint64_t size);
+int bch_cached_dev_resize(struct cached_dev *dc, sector_t sectors);
int bch_cached_dev_attach(struct cached_dev *dc, struct cache_set *c,
uint8_t *set_uuid);
void bch_cached_dev_detach(struct cached_dev *dc);
diff --git a/drivers/md/bcache/btree.c b/drivers/md/bcache/btree.c
index 147c493a989a..07388e51ff9c 100644
--- a/drivers/md/bcache/btree.c
+++ b/drivers/md/bcache/btree.c
@@ -2467,6 +2467,37 @@ int bch_btree_insert(struct cache_set *c, struct keylist *keys,
return ret;
}
+int bch_btree_insert_invalidate(struct cache_set *c, unsigned int inode,
+ sector_t offset, sector_t length)
+{
+ int ret = 0;
+ sector_t num;
+ struct keylist insert_keys;
+
+ bch_keylist_init(&insert_keys);
+ while (!ret && length) {
+ num = min_t(sector_t, length, 1U << (KEY_SIZE_BITS - 1));
+
+ if ((ret = __bch_keylist_realloc(&insert_keys, 2))) {
+ pr_err("cannot allocate memory");
+ break;
+ }
+
+ offset += num;
+ length -= num;
+
+ bch_keylist_add(&insert_keys, &KEY(inode, offset, num));
+ if ((ret = bch_btree_insert(c, &insert_keys, NULL, NULL))) {
+ pr_err("invalidating %llu sectors from %llu error %d",
+ num, offset - num, ret);
+ break;
+ }
+ }
+ bch_keylist_free(&insert_keys);
+
+ return ret;
+}
+
void bch_btree_set_root(struct btree *b)
{
unsigned int i;
diff --git a/drivers/md/bcache/btree.h b/drivers/md/bcache/btree.h
index 1b5fdbc0d83e..28c8885ecea1 100644
--- a/drivers/md/bcache/btree.h
+++ b/drivers/md/bcache/btree.h
@@ -276,6 +276,8 @@ int bch_btree_insert_check_key(struct btree *b, struct btree_op *op,
struct bkey *check_key);
int bch_btree_insert(struct cache_set *c, struct keylist *keys,
atomic_t *journal_ref, struct bkey *replace_key);
+int bch_btree_insert_invalidate(struct cache_set *c, unsigned int inode,
+ sector_t offset, sector_t length);
int bch_gc_thread_start(struct cache_set *c);
void bch_initial_gc_finish(struct cache_set *c);
diff --git a/drivers/md/bcache/super.c b/drivers/md/bcache/super.c
index 125f607d58f0..f19496b4515e 100644
--- a/drivers/md/bcache/super.c
+++ b/drivers/md/bcache/super.c
@@ -1345,6 +1345,93 @@ int bch_cached_dev_attach(struct cached_dev *dc, struct cache_set *c,
return 0;
}
+int bch_cached_dev_resize(struct cached_dev *dc, sector_t sectors)
+{
+ struct cache_set *c = dc->disk.c;
+ uint64_t nr_stripes, n, i;
+ sector_t length, sectors_dirty;
+ atomic_t *stripe_sectors_dirty;
+ unsigned long *full_dirty_stripes;
+ const size_t max_stripes = min_t(size_t, INT_MAX,
+ SIZE_MAX / sizeof(atomic_t));
+
+ /* Block writeback thread and all requests */
+ down_write(&dc->writeback_lock);
+
+ if (!dc->partial_stripes_expensive)
+ goto set_capacity;
+
+ nr_stripes = DIV_ROUND_UP_ULL(sectors, dc->disk.stripe_size);
+ if (!nr_stripes || nr_stripes > max_stripes) {
+ pr_err("nr_stripes too large or invalid: %llu", nr_stripes);
+ up_write(&dc->writeback_lock);
+ return -ENOMEM;
+ }
+
+ if (nr_stripes > dc->disk.nr_stripes)
+ goto realloc;
+
+ for (i = nr_stripes; i < dc->disk.nr_stripes; i++) {
+ sectors_dirty = atomic_read(dc->disk.stripe_sectors_dirty + i);
+ atomic_long_sub(sectors_dirty, &dc->disk.dirty_sectors);
+ }
+ goto nr_stripes;
+
+realloc:
+ n = nr_stripes * sizeof(atomic_t);
+ stripe_sectors_dirty = kvzalloc(n, GFP_KERNEL);
+ if (!stripe_sectors_dirty) {
+ up_write(&dc->writeback_lock);
+ return -ENOMEM;
+ }
+
+ n = BITS_TO_LONGS(nr_stripes) * sizeof(unsigned long);
+ full_dirty_stripes = kvzalloc(n, GFP_KERNEL);
+ if (!full_dirty_stripes) {
+ kvfree(stripe_sectors_dirty);
+ up_write(&dc->writeback_lock);
+ return -ENOMEM;
+ }
+
+ for (i = 0; i < dc->disk.nr_stripes; i++) {
+ sectors_dirty = atomic_read(dc->disk.stripe_sectors_dirty + i);
+ atomic_set(stripe_sectors_dirty + i, sectors_dirty);
+ if (sectors_dirty == dc->disk.stripe_size)
+ set_bit(i, full_dirty_stripes);
+ }
+
+ kvfree(dc->disk.full_dirty_stripes);
+ kvfree(dc->disk.stripe_sectors_dirty);
+ dc->disk.stripe_sectors_dirty = stripe_sectors_dirty;
+ dc->disk.full_dirty_stripes = full_dirty_stripes;
+
+nr_stripes:
+ dc->disk.nr_stripes = nr_stripes;
+
+set_capacity:
+ length = get_capacity(dc->disk.disk);
+ set_capacity(dc->disk.disk, sectors);
+ revalidate_disk(dc->disk.disk);
+
+ if (!c || length <= sectors)
+ goto skip_invalidate;
+ length -= sectors;
+
+ /* invalidate dirty data not used */
+ pr_info("invalidating %llu sectors from %llu", length, sectors);
+ bch_btree_insert_invalidate(c, dc->disk.id, sectors, length);
+
+ /* recount dirty sectors */
+ if (!dc->partial_stripes_expensive) {
+ atomic_long_set(&dc->disk.dirty_sectors, 0);
+ bch_sectors_dirty_init(&dc->disk);
+ }
+
+skip_invalidate:
+ up_write(&dc->writeback_lock);
+ return 0;
+}
+
/* when dc->disk.kobj released */
void bch_cached_dev_release(struct kobject *kobj)
{
diff --git a/drivers/md/bcache/sysfs.c b/drivers/md/bcache/sysfs.c
index 8d1a86249f99..4f480c579b2a 100644
--- a/drivers/md/bcache/sysfs.c
+++ b/drivers/md/bcache/sysfs.c
@@ -199,6 +199,7 @@ SHOW(__bch_cached_dev)
sysfs_printf(data_csum, "%i", dc->disk.data_csum);
+ sysfs_hprint(size, get_capacity(dc->disk.disk) << 9);
var_printf(verify, "%i");
var_printf(bypass_torture_test, "%i");
var_printf(writeback_metadata, "%i");
@@ -312,6 +313,18 @@ STORE(__cached_dev)
#define d_strtoi_h(var) sysfs_hatoi(var, dc->var)
sysfs_strtoul(data_csum, dc->disk.data_csum);
+
+ if (attr == &sysfs_size) {
+ ssize_t ret;
+ sector_t v, sectors;
+
+ strtoi_h_or_return(buf, v);
+ sectors = clamp_t(sector_t, v >> 9, 0,
+ dc->bdev->bd_part->nr_sects - dc->sb.data_offset);
+ ret = bch_cached_dev_resize(dc, sectors);
+ return ret ? ret : size;
+ }
+
d_strtoul(verify);
sysfs_strtoul_bool(bypass_torture_test, dc->bypass_torture_test);
sysfs_strtoul_bool(writeback_metadata, dc->writeback_metadata);
@@ -558,6 +571,7 @@ static struct attribute *bch_cached_dev_attrs[] = {
&sysfs_running,
&sysfs_state,
&sysfs_label,
+ &sysfs_size,
#ifdef CONFIG_BCACHE_DEBUG
&sysfs_verify,
&sysfs_bypass_torture_test,
--
2.17.1
