]> git.sesse.net Git - bcachefs-tools-debian/blobdiff - libbcachefs/ec.c
Update bcachefs sources to 33a60d9b05 bcachefs: Assorted fixes for clang
[bcachefs-tools-debian] / libbcachefs / ec.c
index 7d43fd4a6bb79a2727ba35c4883ba4b278f5a0cd..f58e84a2bf88eb09772d5da414ebdb29906f93b7 100644 (file)
@@ -105,7 +105,8 @@ struct ec_bio {
 /* Stripes btree keys: */
 
 int bch2_stripe_invalid(const struct bch_fs *c, struct bkey_s_c k,
-                       unsigned flags, struct printbuf *err)
+                       enum bkey_invalid_flags flags,
+                       struct printbuf *err)
 {
        const struct bch_stripe *s = bkey_s_c_to_stripe(k).v;
 
@@ -119,12 +120,6 @@ int bch2_stripe_invalid(const struct bch_fs *c, struct bkey_s_c k,
                return -BCH_ERR_invalid_bkey;
        }
 
-       if (bkey_val_bytes(k.k) < sizeof(*s)) {
-               prt_printf(err, "incorrect value size (%zu < %zu)",
-                      bkey_val_bytes(k.k), sizeof(*s));
-               return -BCH_ERR_invalid_bkey;
-       }
-
        if (bkey_val_u64s(k.k) < stripe_val_u64s(s)) {
                prt_printf(err, "incorrect value size (%zu < %u)",
                       bkey_val_u64s(k.k), stripe_val_u64s(s));
@@ -205,11 +200,14 @@ static bool extent_has_stripe_ptr(struct bkey_s_c k, u64 idx)
 
 static void ec_stripe_buf_exit(struct ec_stripe_buf *buf)
 {
-       unsigned i;
+       if (buf->key.k.type == KEY_TYPE_stripe) {
+               struct bkey_i_stripe *s = bkey_i_to_stripe(&buf->key);
+               unsigned i;
 
-       for (i = 0; i < buf->key.v.nr_blocks; i++) {
-               kvpfree(buf->data[i], buf->size << 9);
-               buf->data[i] = NULL;
+               for (i = 0; i < s->v.nr_blocks; i++) {
+                       kvpfree(buf->data[i], buf->size << 9);
+                       buf->data[i] = NULL;
+               }
        }
 }
 
@@ -217,7 +215,7 @@ static void ec_stripe_buf_exit(struct ec_stripe_buf *buf)
 static int ec_stripe_buf_init(struct ec_stripe_buf *buf,
                              unsigned offset, unsigned size)
 {
-       struct bch_stripe *v = &buf->key.v;
+       struct bch_stripe *v = &bkey_i_to_stripe(&buf->key)->v;
        unsigned csum_granularity = 1U << v->csum_granularity_bits;
        unsigned end = offset + size;
        unsigned i;
@@ -233,7 +231,7 @@ static int ec_stripe_buf_init(struct ec_stripe_buf *buf,
 
        memset(buf->valid, 0xFF, sizeof(buf->valid));
 
-       for (i = 0; i < buf->key.v.nr_blocks; i++) {
+       for (i = 0; i < v->nr_blocks; i++) {
                buf->data[i] = kvpmalloc(buf->size << 9, GFP_KERNEL);
                if (!buf->data[i])
                        goto err;
@@ -250,7 +248,7 @@ err:
 static struct bch_csum ec_block_checksum(struct ec_stripe_buf *buf,
                                         unsigned block, unsigned offset)
 {
-       struct bch_stripe *v = &buf->key.v;
+       struct bch_stripe *v = &bkey_i_to_stripe(&buf->key)->v;
        unsigned csum_granularity = 1 << v->csum_granularity_bits;
        unsigned end = buf->offset + buf->size;
        unsigned len = min(csum_granularity, end - offset);
@@ -269,7 +267,7 @@ static struct bch_csum ec_block_checksum(struct ec_stripe_buf *buf,
 
 static void ec_generate_checksums(struct ec_stripe_buf *buf)
 {
-       struct bch_stripe *v = &buf->key.v;
+       struct bch_stripe *v = &bkey_i_to_stripe(&buf->key)->v;
        unsigned i, j, csums_per_device = stripe_csums_per_device(v);
 
        if (!v->csum_type)
@@ -286,7 +284,7 @@ static void ec_generate_checksums(struct ec_stripe_buf *buf)
 
 static void ec_validate_checksums(struct bch_fs *c, struct ec_stripe_buf *buf)
 {
-       struct bch_stripe *v = &buf->key.v;
+       struct bch_stripe *v = &bkey_i_to_stripe(&buf->key)->v;
        unsigned csum_granularity = 1 << v->csum_granularity_bits;
        unsigned i;
 
@@ -309,7 +307,7 @@ static void ec_validate_checksums(struct bch_fs *c, struct ec_stripe_buf *buf)
                        if (bch2_crc_cmp(want, got)) {
                                struct printbuf buf2 = PRINTBUF;
 
-                               bch2_bkey_val_to_text(&buf2, c, bkey_i_to_s_c(&buf->key.k_i));
+                               bch2_bkey_val_to_text(&buf2, c, bkey_i_to_s_c(&buf->key));
 
                                bch_err_ratelimited(c,
                                        "stripe checksum error for %ps at %u:%u: csum type %u, expected %llx got %llx\n%s",
@@ -329,7 +327,7 @@ static void ec_validate_checksums(struct bch_fs *c, struct ec_stripe_buf *buf)
 
 static void ec_generate_ec(struct ec_stripe_buf *buf)
 {
-       struct bch_stripe *v = &buf->key.v;
+       struct bch_stripe *v = &bkey_i_to_stripe(&buf->key)->v;
        unsigned nr_data = v->nr_blocks - v->nr_redundant;
        unsigned bytes = le16_to_cpu(v->sectors) << 9;
 
@@ -338,13 +336,14 @@ static void ec_generate_ec(struct ec_stripe_buf *buf)
 
 static unsigned ec_nr_failed(struct ec_stripe_buf *buf)
 {
-       return buf->key.v.nr_blocks -
-               bitmap_weight(buf->valid, buf->key.v.nr_blocks);
+       struct bch_stripe *v = &bkey_i_to_stripe(&buf->key)->v;
+
+       return v->nr_blocks - bitmap_weight(buf->valid, v->nr_blocks);
 }
 
 static int ec_do_recov(struct bch_fs *c, struct ec_stripe_buf *buf)
 {
-       struct bch_stripe *v = &buf->key.v;
+       struct bch_stripe *v = &bkey_i_to_stripe(&buf->key)->v;
        unsigned i, failed[BCH_BKEY_PTRS_MAX], nr_failed = 0;
        unsigned nr_data = v->nr_blocks - v->nr_redundant;
        unsigned bytes = buf->size << 9;
@@ -368,7 +367,7 @@ static int ec_do_recov(struct bch_fs *c, struct ec_stripe_buf *buf)
 static void ec_block_endio(struct bio *bio)
 {
        struct ec_bio *ec_bio = container_of(bio, struct ec_bio, bio);
-       struct bch_stripe *v = &ec_bio->buf->key.v;
+       struct bch_stripe *v = &bkey_i_to_stripe(&ec_bio->buf->key)->v;
        struct bch_extent_ptr *ptr = &v->ptrs[ec_bio->idx];
        struct bch_dev *ca = ec_bio->ca;
        struct closure *cl = bio->bi_private;
@@ -391,15 +390,16 @@ static void ec_block_endio(struct bio *bio)
 }
 
 static void ec_block_io(struct bch_fs *c, struct ec_stripe_buf *buf,
-                       unsigned rw, unsigned idx, struct closure *cl)
+                       blk_opf_t opf, unsigned idx, struct closure *cl)
 {
-       struct bch_stripe *v = &buf->key.v;
+       struct bch_stripe *v = &bkey_i_to_stripe(&buf->key)->v;
        unsigned offset = 0, bytes = buf->size << 9;
        struct bch_extent_ptr *ptr = &v->ptrs[idx];
        struct bch_dev *ca = bch_dev_bkey_exists(c, ptr->dev);
-       enum bch_data_type data_type = idx < buf->key.v.nr_blocks - buf->key.v.nr_redundant
+       enum bch_data_type data_type = idx < v->nr_blocks - v->nr_redundant
                ? BCH_DATA_user
                : BCH_DATA_parity;
+       int rw = op_is_write(opf);
 
        if (ptr_stale(ca, ptr)) {
                bch_err_ratelimited(c,
@@ -425,7 +425,7 @@ static void ec_block_io(struct bch_fs *c, struct ec_stripe_buf *buf,
 
                ec_bio = container_of(bio_alloc_bioset(ca->disk_sb.bdev,
                                                       nr_iovecs,
-                                                      rw,
+                                                      opf,
                                                       GFP_KERNEL,
                                                       &c->ec_bioset),
                                      struct ec_bio, bio);
@@ -458,9 +458,8 @@ static int get_stripe_key_trans(struct btree_trans *trans, u64 idx,
        struct bkey_s_c k;
        int ret;
 
-       bch2_trans_iter_init(trans, &iter, BTREE_ID_stripes,
-                            POS(0, idx), BTREE_ITER_SLOTS);
-       k = bch2_btree_iter_peek_slot(&iter);
+       k = bch2_bkey_get_iter(trans, &iter, BTREE_ID_stripes,
+                              POS(0, idx), BTREE_ITER_SLOTS);
        ret = bkey_err(k);
        if (ret)
                goto err;
@@ -468,7 +467,7 @@ static int get_stripe_key_trans(struct btree_trans *trans, u64 idx,
                ret = -ENOENT;
                goto err;
        }
-       bkey_reassemble(&stripe->key.k_i, k);
+       bkey_reassemble(&stripe->key, k);
 err:
        bch2_trans_iter_exit(trans, &iter);
        return ret;
@@ -492,9 +491,9 @@ int bch2_ec_read_extent(struct bch_fs *c, struct bch_read_bio *rbio)
 
        BUG_ON(!rbio->pick.has_ec);
 
-       buf = kzalloc(sizeof(*buf), GFP_NOIO);
+       buf = kzalloc(sizeof(*buf), GFP_NOFS);
        if (!buf)
-               return -ENOMEM;
+               return -BCH_ERR_ENOMEM_ec_read_extent;
 
        ret = get_stripe_key(c, rbio->pick.ec.idx, buf);
        if (ret) {
@@ -504,7 +503,7 @@ int bch2_ec_read_extent(struct bch_fs *c, struct bch_read_bio *rbio)
                return -EIO;
        }
 
-       v = &buf->key.v;
+       v = &bkey_i_to_stripe(&buf->key)->v;
 
        if (!bch2_ptr_matches_stripe(v, rbio->pick)) {
                bch_err_ratelimited(c,
@@ -559,7 +558,7 @@ static int __ec_stripe_mem_alloc(struct bch_fs *c, size_t idx, gfp_t gfp)
 
        if (idx >= h->size) {
                if (!init_heap(&n, max(1024UL, roundup_pow_of_two(idx + 1)), gfp))
-                       return -ENOMEM;
+                       return -BCH_ERR_ENOMEM_ec_stripe_mem_alloc;
 
                mutex_lock(&c->ec_stripes_heap_lock);
                if (n.size > h->size) {
@@ -573,11 +572,11 @@ static int __ec_stripe_mem_alloc(struct bch_fs *c, size_t idx, gfp_t gfp)
        }
 
        if (!genradix_ptr_alloc(&c->stripes, idx, gfp))
-               return -ENOMEM;
+               return -BCH_ERR_ENOMEM_ec_stripe_mem_alloc;
 
        if (c->gc_pos.phase != GC_PHASE_NOT_RUNNING &&
            !genradix_ptr_alloc(&c->gc_stripes, idx, gfp))
-               return -ENOMEM;
+               return -BCH_ERR_ENOMEM_ec_stripe_mem_alloc;
 
        return 0;
 }
@@ -585,15 +584,8 @@ static int __ec_stripe_mem_alloc(struct bch_fs *c, size_t idx, gfp_t gfp)
 static int ec_stripe_mem_alloc(struct btree_trans *trans,
                               struct btree_iter *iter)
 {
-       size_t idx = iter->pos.offset;
-
-       if (!__ec_stripe_mem_alloc(trans->c, idx, GFP_NOWAIT|__GFP_NOWARN))
-               return 0;
-
-       bch2_trans_unlock(trans);
-
-       return   __ec_stripe_mem_alloc(trans->c, idx, GFP_KERNEL) ?:
-               bch2_trans_relock(trans);
+       return allocate_dropping_locks_errcode(trans,
+                       __ec_stripe_mem_alloc(trans->c, iter->pos.offset, _gfp));
 }
 
 /*
@@ -659,14 +651,13 @@ static void bch2_stripe_close(struct bch_fs *c, struct ec_stripe_new *s)
 static u64 stripe_idx_to_delete(struct bch_fs *c)
 {
        ec_stripes_heap *h = &c->ec_stripes_heap;
-       size_t heap_idx;
 
        lockdep_assert_held(&c->ec_stripes_heap_lock);
 
-       for (heap_idx = 0; heap_idx < h->used; heap_idx++)
-               if (h->data[heap_idx].blocks_nonempty == 0 &&
-                   !bch2_stripe_is_open(c, h->data[heap_idx].idx))
-                       return h->data[heap_idx].idx;
+       if (h->used &&
+           h->data[0].blocks_nonempty == 0 &&
+           !bch2_stripe_is_open(c, h->data[0].idx))
+               return h->data[0].idx;
 
        return 0;
 }
@@ -762,9 +753,8 @@ static int ec_stripe_delete(struct btree_trans *trans, u64 idx)
        struct bkey_s_c_stripe s;
        int ret;
 
-       bch2_trans_iter_init(trans, &iter, BTREE_ID_stripes, POS(0, idx),
-                            BTREE_ITER_INTENT);
-       k = bch2_btree_iter_peek_slot(&iter);
+       k = bch2_bkey_get_iter(trans, &iter, BTREE_ID_stripes, POS(0, idx),
+                              BTREE_ITER_INTENT);
        ret = bkey_err(k);
        if (ret)
                goto err;
@@ -814,7 +804,7 @@ static void ec_stripe_delete_work(struct work_struct *work)
                ret = commit_do(&trans, NULL, NULL, BTREE_INSERT_NOFAIL,
                                ec_stripe_delete(&trans, idx));
                if (ret) {
-                       bch_err(c, "%s: err %s", __func__, bch2_err_str(ret));
+                       bch_err_fn(c, ret);
                        break;
                }
        }
@@ -827,7 +817,7 @@ static void ec_stripe_delete_work(struct work_struct *work)
 void bch2_do_stripe_deletes(struct bch_fs *c)
 {
        if (bch2_write_ref_tryget(c, BCH_WRITE_REF_stripe_delete) &&
-           !schedule_work(&c->ec_stripe_delete_work))
+           !queue_work(c->write_ref_wq, &c->ec_stripe_delete_work))
                bch2_write_ref_put(c, BCH_WRITE_REF_stripe_delete);
 }
 
@@ -842,9 +832,8 @@ static int ec_stripe_key_update(struct btree_trans *trans,
        struct bkey_s_c k;
        int ret;
 
-       bch2_trans_iter_init(trans, &iter, BTREE_ID_stripes,
-                            new->k.p, BTREE_ITER_INTENT);
-       k = bch2_btree_iter_peek_slot(&iter);
+       k = bch2_bkey_get_iter(trans, &iter, BTREE_ID_stripes,
+                              new->k.p, BTREE_ITER_INTENT);
        ret = bkey_err(k);
        if (ret)
                goto err;
@@ -888,8 +877,9 @@ err:
 static int ec_stripe_update_extent(struct btree_trans *trans,
                                   struct bpos bucket, u8 gen,
                                   struct ec_stripe_buf *s,
-                                  u64 *bp_offset)
+                                  struct bpos *bp_pos)
 {
+       struct bch_stripe *v = &bkey_i_to_stripe(&s->key)->v;
        struct bch_fs *c = trans->c;
        struct bch_backpointer bp;
        struct btree_iter iter;
@@ -901,10 +891,10 @@ static int ec_stripe_update_extent(struct btree_trans *trans,
        int ret, dev, block;
 
        ret = bch2_get_next_backpointer(trans, bucket, gen,
-                               bp_offset, &bp, BTREE_ITER_CACHED);
+                               bp_pos, &bp, BTREE_ITER_CACHED);
        if (ret)
                return ret;
-       if (*bp_offset == U64_MAX)
+       if (bpos_eq(*bp_pos, SPOS_MAX))
                return 0;
 
        if (bp.level) {
@@ -912,7 +902,7 @@ static int ec_stripe_update_extent(struct btree_trans *trans,
                struct btree_iter node_iter;
                struct btree *b;
 
-               b = bch2_backpointer_get_node(trans, &node_iter, bucket, *bp_offset, bp);
+               b = bch2_backpointer_get_node(trans, &node_iter, *bp_pos, bp);
                bch2_trans_iter_exit(trans, &node_iter);
 
                if (!b)
@@ -926,7 +916,7 @@ static int ec_stripe_update_extent(struct btree_trans *trans,
                return -EIO;
        }
 
-       k = bch2_backpointer_get_key(trans, &iter, bucket, *bp_offset, bp);
+       k = bch2_backpointer_get_key(trans, &iter, *bp_pos, bp, BTREE_ITER_INTENT);
        ret = bkey_err(k);
        if (ret)
                return ret;
@@ -941,7 +931,7 @@ static int ec_stripe_update_extent(struct btree_trans *trans,
        if (extent_has_stripe_ptr(k, s->key.k.p.offset))
                goto out;
 
-       ptr_c = bkey_matches_stripe(&s->key.v, k, &block);
+       ptr_c = bkey_matches_stripe(v, k, &block);
        /*
         * It doesn't generally make sense to erasure code cached ptrs:
         * XXX: should we be incrementing a counter?
@@ -949,7 +939,7 @@ static int ec_stripe_update_extent(struct btree_trans *trans,
        if (!ptr_c || ptr_c->cached)
                goto out;
 
-       dev = s->key.v.ptrs[block].dev;
+       dev = v->ptrs[block].dev;
 
        n = bch2_trans_kmalloc(trans, bkey_bytes(k.k) + sizeof(stripe_ptr));
        ret = PTR_ERR_OR_ZERO(n);
@@ -959,13 +949,13 @@ static int ec_stripe_update_extent(struct btree_trans *trans,
        bkey_reassemble(n, k);
 
        bch2_bkey_drop_ptrs(bkey_i_to_s(n), ptr, ptr->dev != dev);
-       ec_ptr = (void *) bch2_bkey_has_device(bkey_i_to_s_c(n), dev);
+       ec_ptr = bch2_bkey_has_device(bkey_i_to_s(n), dev);
        BUG_ON(!ec_ptr);
 
        stripe_ptr = (struct bch_extent_stripe_ptr) {
                .type = 1 << BCH_EXTENT_ENTRY_stripe_ptr,
                .block          = block,
-               .redundancy     = s->key.v.nr_redundant,
+               .redundancy     = v->nr_redundant,
                .idx            = s->key.k.p.offset,
        };
 
@@ -983,22 +973,24 @@ static int ec_stripe_update_bucket(struct btree_trans *trans, struct ec_stripe_b
                                   unsigned block)
 {
        struct bch_fs *c = trans->c;
-       struct bch_extent_ptr bucket = s->key.v.ptrs[block];
+       struct bch_stripe *v = &bkey_i_to_stripe(&s->key)->v;
+       struct bch_extent_ptr bucket = v->ptrs[block];
        struct bpos bucket_pos = PTR_BUCKET_POS(c, &bucket);
-       u64 bp_offset = 0;
+       struct bpos bp_pos = POS_MIN;
        int ret = 0;
 
        while (1) {
                ret = commit_do(trans, NULL, NULL,
+                               BTREE_INSERT_NOCHECK_RW|
                                BTREE_INSERT_NOFAIL,
                        ec_stripe_update_extent(trans, bucket_pos, bucket.gen,
-                                               s, &bp_offset));
+                                               s, &bp_pos));
                if (ret)
                        break;
-               if (bp_offset == U64_MAX)
+               if (bkey_eq(bp_pos, POS_MAX))
                        break;
 
-               bp_offset++;
+               bp_pos = bpos_nosnap_successor(bp_pos);
        }
 
        return ret;
@@ -1007,7 +999,7 @@ static int ec_stripe_update_bucket(struct btree_trans *trans, struct ec_stripe_b
 static int ec_stripe_update_extents(struct bch_fs *c, struct ec_stripe_buf *s)
 {
        struct btree_trans trans;
-       struct bch_stripe *v = &s->key.v;
+       struct bch_stripe *v = &bkey_i_to_stripe(&s->key)->v;
        unsigned i, nr_data = v->nr_blocks - v->nr_redundant;
        int ret = 0;
 
@@ -1038,7 +1030,7 @@ static void zero_out_rest_of_ec_bucket(struct bch_fs *c,
        int ret;
 
        if (!bch2_dev_get_ioref(ca, WRITE)) {
-               s->err = -EROFS;
+               s->err = -BCH_ERR_erofs_no_writes;
                return;
        }
 
@@ -1057,6 +1049,13 @@ static void zero_out_rest_of_ec_bucket(struct bch_fs *c,
                s->err = ret;
 }
 
+void bch2_ec_stripe_new_free(struct bch_fs *c, struct ec_stripe_new *s)
+{
+       if (s->idx)
+               bch2_stripe_close(c, s);
+       kfree(s);
+}
+
 /*
  * data buckets of new stripe all written: create the stripe
  */
@@ -1064,7 +1063,7 @@ static void ec_stripe_create(struct ec_stripe_new *s)
 {
        struct bch_fs *c = s->c;
        struct open_bucket *ob;
-       struct bch_stripe *v = &s->new_stripe.key.v;
+       struct bch_stripe *v = &bkey_i_to_stripe(&s->new_stripe.key)->v;
        unsigned i, nr_data = v->nr_blocks - v->nr_redundant;
        int ret;
 
@@ -1072,13 +1071,15 @@ static void ec_stripe_create(struct ec_stripe_new *s)
 
        closure_sync(&s->iodone);
 
-       for (i = 0; i < nr_data; i++)
-               if (s->blocks[i]) {
-                       ob = c->open_buckets + s->blocks[i];
+       if (!s->err) {
+               for (i = 0; i < nr_data; i++)
+                       if (s->blocks[i]) {
+                               ob = c->open_buckets + s->blocks[i];
 
-                       if (ob->sectors_free)
-                               zero_out_rest_of_ec_bucket(c, s, i, ob);
-               }
+                               if (ob->sectors_free)
+                                       zero_out_rest_of_ec_bucket(c, s, i, ob);
+                       }
+       }
 
        if (s->err) {
                if (!bch2_err_matches(s->err, EROFS))
@@ -1095,7 +1096,7 @@ static void ec_stripe_create(struct ec_stripe_new *s)
                }
 
                for (i = 0; i < nr_data; i++)
-                       if (stripe_blockcount_get(&s->existing_stripe.key.v, i))
+                       if (stripe_blockcount_get(&bkey_i_to_stripe(&s->existing_stripe.key)->v, i))
                                swap(s->new_stripe.data[i],
                                     s->existing_stripe.data[i]);
 
@@ -1119,9 +1120,12 @@ static void ec_stripe_create(struct ec_stripe_new *s)
                goto err;
        }
 
-       ret = bch2_trans_do(c, &s->res, NULL, BTREE_INSERT_NOFAIL,
-                           ec_stripe_key_update(&trans, &s->new_stripe.key,
-                                                !s->have_existing_stripe));
+       ret = bch2_trans_do(c, &s->res, NULL,
+                           BTREE_INSERT_NOCHECK_RW|
+                           BTREE_INSERT_NOFAIL,
+                           ec_stripe_key_update(&trans,
+                                       bkey_i_to_stripe(&s->new_stripe.key),
+                                       !s->have_existing_stripe));
        if (ret) {
                bch_err(c, "error creating stripe: error creating stripe key");
                goto err;
@@ -1151,14 +1155,13 @@ err:
        mutex_lock(&c->ec_stripe_new_lock);
        list_del(&s->list);
        mutex_unlock(&c->ec_stripe_new_lock);
-
-       if (s->idx)
-               bch2_stripe_close(c, s);
+       wake_up(&c->ec_stripe_new_wait);
 
        ec_stripe_buf_exit(&s->existing_stripe);
        ec_stripe_buf_exit(&s->new_stripe);
        closure_debug_destroy(&s->iodone);
-       kfree(s);
+
+       ec_stripe_new_put(c, s, STRIPE_REF_stripe);
 }
 
 static struct ec_stripe_new *get_pending_stripe(struct bch_fs *c)
@@ -1167,7 +1170,7 @@ static struct ec_stripe_new *get_pending_stripe(struct bch_fs *c)
 
        mutex_lock(&c->ec_stripe_new_lock);
        list_for_each_entry(s, &c->ec_stripe_new_list, list)
-               if (!atomic_read(&s->pin))
+               if (!atomic_read(&s->ref[STRIPE_REF_io]))
                        goto out;
        s = NULL;
 out:
@@ -1209,7 +1212,7 @@ static void ec_stripe_set_pending(struct bch_fs *c, struct ec_stripe_head *h)
        list_add(&s->list, &c->ec_stripe_new_list);
        mutex_unlock(&c->ec_stripe_new_lock);
 
-       ec_stripe_new_put(c, s);
+       ec_stripe_new_put(c, s, STRIPE_REF_io);
 }
 
 void bch2_ec_bucket_cancel(struct bch_fs *c, struct open_bucket *ob)
@@ -1283,14 +1286,14 @@ static bool may_create_new_stripe(struct bch_fs *c)
 }
 
 static void ec_stripe_key_init(struct bch_fs *c,
-                              struct bkey_i_stripe *s,
+                              struct bkey_i *k,
                               unsigned nr_data,
                               unsigned nr_parity,
                               unsigned stripe_size)
 {
+       struct bkey_i_stripe *s = bkey_stripe_init(k);
        unsigned u64s;
 
-       bkey_stripe_init(&s->k_i);
        s->v.sectors                    = cpu_to_le16(stripe_size);
        s->v.algorithm                  = 0;
        s->v.nr_blocks                  = nr_data + nr_parity;
@@ -1317,19 +1320,20 @@ static int ec_new_stripe_alloc(struct bch_fs *c, struct ec_stripe_head *h)
 
        s = kzalloc(sizeof(*s), GFP_KERNEL);
        if (!s)
-               return -ENOMEM;
+               return -BCH_ERR_ENOMEM_ec_new_stripe_alloc;
 
        mutex_init(&s->lock);
        closure_init(&s->iodone, NULL);
-       atomic_set(&s->pin, 1);
+       atomic_set(&s->ref[STRIPE_REF_stripe], 1);
+       atomic_set(&s->ref[STRIPE_REF_io], 1);
        s->c            = c;
        s->h            = h;
        s->nr_data      = min_t(unsigned, h->nr_active_devs,
                                BCH_BKEY_PTRS_MAX) - h->redundancy;
        s->nr_parity    = h->redundancy;
 
-       ec_stripe_key_init(c, &s->new_stripe.key, s->nr_data,
-                          s->nr_parity, h->blocksize);
+       ec_stripe_key_init(c, &s->new_stripe.key,
+                          s->nr_data, s->nr_parity, h->blocksize);
 
        h->s = s;
        return 0;
@@ -1338,7 +1342,7 @@ static int ec_new_stripe_alloc(struct bch_fs *c, struct ec_stripe_head *h)
 static struct ec_stripe_head *
 ec_new_stripe_head_alloc(struct bch_fs *c, unsigned target,
                         unsigned algo, unsigned redundancy,
-                        enum alloc_reserve reserve)
+                        enum bch_watermark watermark)
 {
        struct ec_stripe_head *h;
        struct bch_dev *ca;
@@ -1354,7 +1358,7 @@ ec_new_stripe_head_alloc(struct bch_fs *c, unsigned target,
        h->target       = target;
        h->algo         = algo;
        h->redundancy   = redundancy;
-       h->reserve      = reserve;
+       h->watermark    = watermark;
 
        rcu_read_lock();
        h->devs = target_rw_devs(c, BCH_DATA_user, target);
@@ -1385,11 +1389,12 @@ void bch2_ec_stripe_head_put(struct bch_fs *c, struct ec_stripe_head *h)
        mutex_unlock(&h->lock);
 }
 
-struct ec_stripe_head *__bch2_ec_stripe_head_get(struct btree_trans *trans,
-                                                unsigned target,
-                                                unsigned algo,
-                                                unsigned redundancy,
-                                                enum alloc_reserve reserve)
+static struct ec_stripe_head *
+__bch2_ec_stripe_head_get(struct btree_trans *trans,
+                         unsigned target,
+                         unsigned algo,
+                         unsigned redundancy,
+                         enum bch_watermark watermark)
 {
        struct bch_fs *c = trans->c;
        struct ec_stripe_head *h;
@@ -1402,39 +1407,45 @@ struct ec_stripe_head *__bch2_ec_stripe_head_get(struct btree_trans *trans,
        if (ret)
                return ERR_PTR(ret);
 
+       if (test_bit(BCH_FS_GOING_RO, &c->flags)) {
+               h = ERR_PTR(-BCH_ERR_erofs_no_writes);
+               goto found;
+       }
+
        list_for_each_entry(h, &c->ec_stripe_head_list, list)
                if (h->target           == target &&
                    h->algo             == algo &&
                    h->redundancy       == redundancy &&
-                   h->reserve          == reserve) {
+                   h->watermark        == watermark) {
                        ret = bch2_trans_mutex_lock(trans, &h->lock);
                        if (ret)
                                h = ERR_PTR(ret);
                        goto found;
                }
 
-       h = ec_new_stripe_head_alloc(c, target, algo, redundancy, reserve);
+       h = ec_new_stripe_head_alloc(c, target, algo, redundancy, watermark);
 found:
        mutex_unlock(&c->ec_stripe_head_lock);
        return h;
 }
 
 static int new_stripe_alloc_buckets(struct btree_trans *trans, struct ec_stripe_head *h,
-                                   enum alloc_reserve reserve, struct closure *cl)
+                                   enum bch_watermark watermark, struct closure *cl)
 {
        struct bch_fs *c = trans->c;
        struct bch_devs_mask devs = h->devs;
        struct open_bucket *ob;
        struct open_buckets buckets;
+       struct bch_stripe *v = &bkey_i_to_stripe(&h->s->new_stripe.key)->v;
        unsigned i, j, nr_have_parity = 0, nr_have_data = 0;
        bool have_cache = true;
        int ret = 0;
 
-       BUG_ON(h->s->new_stripe.key.v.nr_blocks         != h->s->nr_data + h->s->nr_parity);
-       BUG_ON(h->s->new_stripe.key.v.nr_redundant      != h->s->nr_parity);
+       BUG_ON(v->nr_blocks     != h->s->nr_data + h->s->nr_parity);
+       BUG_ON(v->nr_redundant  != h->s->nr_parity);
 
-       for_each_set_bit(i, h->s->blocks_gotten, h->s->new_stripe.key.v.nr_blocks) {
-               __clear_bit(h->s->new_stripe.key.v.ptrs[i].dev, devs.d);
+       for_each_set_bit(i, h->s->blocks_gotten, v->nr_blocks) {
+               __clear_bit(v->ptrs[i].dev, devs.d);
                if (i < h->s->nr_data)
                        nr_have_data++;
                else
@@ -1451,9 +1462,9 @@ static int new_stripe_alloc_buckets(struct btree_trans *trans, struct ec_stripe_
                                            &devs,
                                            h->s->nr_parity,
                                            &nr_have_parity,
-                                           &have_cache,
+                                           &have_cache, 0,
                                            BCH_DATA_parity,
-                                           reserve,
+                                           watermark,
                                            cl);
 
                open_bucket_for_each(c, &buckets, ob, i) {
@@ -1463,7 +1474,7 @@ static int new_stripe_alloc_buckets(struct btree_trans *trans, struct ec_stripe_
                        BUG_ON(j >= h->s->nr_data + h->s->nr_parity);
 
                        h->s->blocks[j] = buckets.v[i];
-                       h->s->new_stripe.key.v.ptrs[j] = bch2_ob_ptr(c, ob);
+                       v->ptrs[j] = bch2_ob_ptr(c, ob);
                        __set_bit(j, h->s->blocks_gotten);
                }
 
@@ -1478,9 +1489,9 @@ static int new_stripe_alloc_buckets(struct btree_trans *trans, struct ec_stripe_
                                            &devs,
                                            h->s->nr_data,
                                            &nr_have_data,
-                                           &have_cache,
+                                           &have_cache, 0,
                                            BCH_DATA_user,
-                                           reserve,
+                                           watermark,
                                            cl);
 
                open_bucket_for_each(c, &buckets, ob, i) {
@@ -1489,7 +1500,7 @@ static int new_stripe_alloc_buckets(struct btree_trans *trans, struct ec_stripe_
                        BUG_ON(j >= h->s->nr_data);
 
                        h->s->blocks[j] = buckets.v[i];
-                       h->s->new_stripe.key.v.ptrs[j] = bch2_ob_ptr(c, ob);
+                       v->ptrs[j] = bch2_ob_ptr(c, ob);
                        __set_bit(j, h->s->blocks_gotten);
                }
 
@@ -1539,6 +1550,8 @@ static s64 get_existing_stripe(struct bch_fs *c,
 static int __bch2_ec_stripe_head_reuse(struct btree_trans *trans, struct ec_stripe_head *h)
 {
        struct bch_fs *c = trans->c;
+       struct bch_stripe *new_v = &bkey_i_to_stripe(&h->s->new_stripe.key)->v;
+       struct bch_stripe *existing_v;
        unsigned i;
        s64 idx;
        int ret;
@@ -1559,9 +1572,11 @@ static int __bch2_ec_stripe_head_reuse(struct btree_trans *trans, struct ec_stri
                return ret;
        }
 
-       BUG_ON(h->s->existing_stripe.key.v.nr_redundant != h->s->nr_parity);
-       h->s->nr_data = h->s->existing_stripe.key.v.nr_blocks -
-               h->s->existing_stripe.key.v.nr_redundant;
+       existing_v = &bkey_i_to_stripe(&h->s->existing_stripe.key)->v;
+
+       BUG_ON(existing_v->nr_redundant != h->s->nr_parity);
+       h->s->nr_data = existing_v->nr_blocks -
+               existing_v->nr_redundant;
 
        ret = ec_stripe_buf_init(&h->s->existing_stripe, 0, h->blocksize);
        if (ret) {
@@ -1570,21 +1585,21 @@ static int __bch2_ec_stripe_head_reuse(struct btree_trans *trans, struct ec_stri
        }
 
        BUG_ON(h->s->existing_stripe.size != h->blocksize);
-       BUG_ON(h->s->existing_stripe.size != h->s->existing_stripe.key.v.sectors);
+       BUG_ON(h->s->existing_stripe.size != le16_to_cpu(existing_v->sectors));
 
        /*
         * Free buckets we initially allocated - they might conflict with
         * blocks from the stripe we're reusing:
         */
-       for_each_set_bit(i, h->s->blocks_gotten, h->s->new_stripe.key.v.nr_blocks) {
+       for_each_set_bit(i, h->s->blocks_gotten, new_v->nr_blocks) {
                bch2_open_bucket_put(c, c->open_buckets + h->s->blocks[i]);
                h->s->blocks[i] = 0;
        }
        memset(h->s->blocks_gotten, 0, sizeof(h->s->blocks_gotten));
        memset(h->s->blocks_allocated, 0, sizeof(h->s->blocks_allocated));
 
-       for (i = 0; i < h->s->existing_stripe.key.v.nr_blocks; i++) {
-               if (stripe_blockcount_get(&h->s->existing_stripe.key.v, i)) {
+       for (i = 0; i < existing_v->nr_blocks; i++) {
+               if (stripe_blockcount_get(existing_v, i)) {
                        __set_bit(i, h->s->blocks_gotten);
                        __set_bit(i, h->s->blocks_allocated);
                }
@@ -1592,7 +1607,7 @@ static int __bch2_ec_stripe_head_reuse(struct btree_trans *trans, struct ec_stri
                ec_block_io(c, &h->s->existing_stripe, READ, i, &h->s->iodone);
        }
 
-       bkey_copy(&h->s->new_stripe.key.k_i, &h->s->existing_stripe.key.k_i);
+       bkey_copy(&h->s->new_stripe.key, &h->s->existing_stripe.key);
        h->s->have_existing_stripe = true;
 
        return 0;
@@ -1658,7 +1673,7 @@ struct ec_stripe_head *bch2_ec_stripe_head_get(struct btree_trans *trans,
                                               unsigned target,
                                               unsigned algo,
                                               unsigned redundancy,
-                                              enum alloc_reserve reserve,
+                                              enum bch_watermark watermark,
                                               struct closure *cl)
 {
        struct bch_fs *c = trans->c;
@@ -1666,15 +1681,15 @@ struct ec_stripe_head *bch2_ec_stripe_head_get(struct btree_trans *trans,
        bool waiting = false;
        int ret;
 
-       h = __bch2_ec_stripe_head_get(trans, target, algo, redundancy, reserve);
+       h = __bch2_ec_stripe_head_get(trans, target, algo, redundancy, watermark);
        if (!h)
                bch_err(c, "no stripe head");
        if (IS_ERR_OR_NULL(h))
                return h;
 
        if (!h->s) {
-               if (ec_new_stripe_alloc(c, h)) {
-                       ret = -ENOMEM;
+               ret = ec_new_stripe_alloc(c, h);
+               if (ret) {
                        bch_err(c, "failed to allocate new stripe");
                        goto err;
                }
@@ -1687,7 +1702,7 @@ struct ec_stripe_head *bch2_ec_stripe_head_get(struct btree_trans *trans,
                goto alloc_existing;
 
        /* First, try to allocate a full stripe: */
-       ret =   new_stripe_alloc_buckets(trans, h, RESERVE_stripe, NULL) ?:
+       ret =   new_stripe_alloc_buckets(trans, h, BCH_WATERMARK_stripe, NULL) ?:
                __bch2_ec_stripe_head_reserve(trans, h);
        if (!ret)
                goto allocate_buf;
@@ -1706,6 +1721,14 @@ struct ec_stripe_head *bch2_ec_stripe_head_get(struct btree_trans *trans,
                if (waiting || !cl || ret != -BCH_ERR_stripe_alloc_blocked)
                        goto err;
 
+               if (watermark == BCH_WATERMARK_copygc) {
+                       ret =   new_stripe_alloc_buckets(trans, h, watermark, NULL) ?:
+                               __bch2_ec_stripe_head_reserve(trans, h);
+                       if (ret)
+                               goto err;
+                       goto allocate_buf;
+               }
+
                /* XXX freelist_wait? */
                closure_wait(&c->freelist_wait, cl);
                waiting = true;
@@ -1715,10 +1738,10 @@ struct ec_stripe_head *bch2_ec_stripe_head_get(struct btree_trans *trans,
                closure_wake_up(&c->freelist_wait);
 alloc_existing:
        /*
-        * Retry allocating buckets, with the reserve watermark for this
+        * Retry allocating buckets, with the watermark for this
         * particular write:
         */
-       ret = new_stripe_alloc_buckets(trans, h, reserve, cl);
+       ret = new_stripe_alloc_buckets(trans, h, watermark, cl);
        if (ret)
                goto err;
 
@@ -1738,7 +1761,7 @@ err:
        return ERR_PTR(ret);
 }
 
-void bch2_ec_stop_dev(struct bch_fs *c, struct bch_dev *ca)
+static void __bch2_ec_stop(struct bch_fs *c, struct bch_dev *ca)
 {
        struct ec_stripe_head *h;
        struct open_bucket *ob;
@@ -1746,12 +1769,14 @@ void bch2_ec_stop_dev(struct bch_fs *c, struct bch_dev *ca)
 
        mutex_lock(&c->ec_stripe_head_lock);
        list_for_each_entry(h, &c->ec_stripe_head_list, list) {
-
                mutex_lock(&h->lock);
                if (!h->s)
                        goto unlock;
 
-               for (i = 0; i < h->s->new_stripe.key.v.nr_blocks; i++) {
+               if (!ca)
+                       goto found;
+
+               for (i = 0; i < bkey_i_to_stripe(&h->s->new_stripe.key)->v.nr_blocks; i++) {
                        if (!h->s->blocks[i])
                                continue;
 
@@ -1761,7 +1786,7 @@ void bch2_ec_stop_dev(struct bch_fs *c, struct bch_dev *ca)
                }
                goto unlock;
 found:
-               h->s->err = -EROFS;
+               h->s->err = -BCH_ERR_erofs_no_writes;
                ec_stripe_set_pending(c, h);
 unlock:
                mutex_unlock(&h->lock);
@@ -1769,6 +1794,32 @@ unlock:
        mutex_unlock(&c->ec_stripe_head_lock);
 }
 
+void bch2_ec_stop_dev(struct bch_fs *c, struct bch_dev *ca)
+{
+       __bch2_ec_stop(c, ca);
+}
+
+void bch2_fs_ec_stop(struct bch_fs *c)
+{
+       __bch2_ec_stop(c, NULL);
+}
+
+static bool bch2_fs_ec_flush_done(struct bch_fs *c)
+{
+       bool ret;
+
+       mutex_lock(&c->ec_stripe_new_lock);
+       ret = list_empty(&c->ec_stripe_new_list);
+       mutex_unlock(&c->ec_stripe_new_lock);
+
+       return ret;
+}
+
+void bch2_fs_ec_flush(struct bch_fs *c)
+{
+       wait_event(c->ec_stripe_new_wait, bch2_fs_ec_flush_done(c));
+}
+
 int bch2_stripes_read(struct bch_fs *c)
 {
        struct btree_trans trans;
@@ -1809,7 +1860,7 @@ int bch2_stripes_read(struct bch_fs *c)
        bch2_trans_exit(&trans);
 
        if (ret)
-               bch_err(c, "error reading stripes: %i", ret);
+               bch_err_fn(c, ret);
 
        return ret;
 }
@@ -1821,13 +1872,16 @@ void bch2_stripes_heap_to_text(struct printbuf *out, struct bch_fs *c)
        size_t i;
 
        mutex_lock(&c->ec_stripes_heap_lock);
-       for (i = 0; i < min_t(size_t, h->used, 20); i++) {
+       for (i = 0; i < min_t(size_t, h->used, 50); i++) {
                m = genradix_ptr(&c->stripes, h->data[i].idx);
 
-               prt_printf(out, "%zu %u/%u+%u\n", h->data[i].idx,
+               prt_printf(out, "%zu %u/%u+%u", h->data[i].idx,
                       h->data[i].blocks_nonempty,
                       m->nr_blocks - m->nr_redundant,
                       m->nr_redundant);
+               if (bch2_stripe_is_open(c, h->data[i].idx))
+                       prt_str(out, " open");
+               prt_newline(out);
        }
        mutex_unlock(&c->ec_stripes_heap_lock);
 }
@@ -1839,22 +1893,27 @@ void bch2_new_stripes_to_text(struct printbuf *out, struct bch_fs *c)
 
        mutex_lock(&c->ec_stripe_head_lock);
        list_for_each_entry(h, &c->ec_stripe_head_list, list) {
-               prt_printf(out, "target %u algo %u redundancy %u:\n",
-                      h->target, h->algo, h->redundancy);
+               prt_printf(out, "target %u algo %u redundancy %u %s:\n",
+                      h->target, h->algo, h->redundancy,
+                      bch2_watermarks[h->watermark]);
 
                if (h->s)
-                       prt_printf(out, "\tpending: idx %llu blocks %u+%u allocated %u\n",
+                       prt_printf(out, "\tidx %llu blocks %u+%u allocated %u\n",
                               h->s->idx, h->s->nr_data, h->s->nr_parity,
                               bitmap_weight(h->s->blocks_allocated,
                                             h->s->nr_data));
        }
        mutex_unlock(&c->ec_stripe_head_lock);
 
+       prt_printf(out, "in flight:\n");
+
        mutex_lock(&c->ec_stripe_new_lock);
        list_for_each_entry(s, &c->ec_stripe_new_list, list) {
-               prt_printf(out, "\tin flight: idx %llu blocks %u+%u pin %u\n",
+               prt_printf(out, "\tidx %llu blocks %u+%u ref %u %u %s\n",
                           s->idx, s->nr_data, s->nr_parity,
-                          atomic_read(&s->pin));
+                          atomic_read(&s->ref[STRIPE_REF_io]),
+                          atomic_read(&s->ref[STRIPE_REF_stripe]),
+                          bch2_watermarks[s->h->watermark]);
        }
        mutex_unlock(&c->ec_stripe_new_lock);
 }
@@ -1875,7 +1934,7 @@ void bch2_fs_ec_exit(struct bch_fs *c)
                        break;
 
                if (h->s) {
-                       for (i = 0; i < h->s->new_stripe.key.v.nr_blocks; i++)
+                       for (i = 0; i < bkey_i_to_stripe(&h->s->new_stripe.key)->v.nr_blocks; i++)
                                BUG_ON(h->s->blocks[i]);
 
                        kfree(h->s);
@@ -1892,14 +1951,22 @@ void bch2_fs_ec_exit(struct bch_fs *c)
 
 void bch2_fs_ec_init_early(struct bch_fs *c)
 {
+       spin_lock_init(&c->ec_stripes_new_lock);
+       mutex_init(&c->ec_stripes_heap_lock);
+
+       INIT_LIST_HEAD(&c->ec_stripe_head_list);
+       mutex_init(&c->ec_stripe_head_lock);
+
+       INIT_LIST_HEAD(&c->ec_stripe_new_list);
+       mutex_init(&c->ec_stripe_new_lock);
+       init_waitqueue_head(&c->ec_stripe_new_wait);
+
        INIT_WORK(&c->ec_stripe_create_work, ec_stripe_create_work);
        INIT_WORK(&c->ec_stripe_delete_work, ec_stripe_delete_work);
 }
 
 int bch2_fs_ec_init(struct bch_fs *c)
 {
-       spin_lock_init(&c->ec_stripes_new_lock);
-
        return bioset_init(&c->ec_bioset, 1, offsetof(struct ec_bio, bio),
                           BIOSET_NEED_BVECS);
 }