]> git.sesse.net Git - bcachefs-tools-debian/blobdiff - libbcachefs/ec.c
Update bcachefs sources to bed61fae3b bcachefs: Delete a faulty assertion
[bcachefs-tools-debian] / libbcachefs / ec.c
index 7bd68880995f28bebb9ac3efc1391b87920ae53f..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,18 +200,22 @@ 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;
+               }
        }
 }
 
+/* XXX: this is a non-mempoolified memory allocation: */
 static int ec_stripe_buf_init(struct ec_stripe_buf *buf,
-                              unsigned offset, unsigned size)
+                             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;
@@ -232,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;
@@ -241,7 +240,7 @@ static int ec_stripe_buf_init(struct ec_stripe_buf *buf,
        return 0;
 err:
        ec_stripe_buf_exit(buf);
-       return -ENOMEM;
+       return -BCH_ERR_ENOMEM_stripe_buf;
 }
 
 /* Checksumming: */
@@ -249,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);
@@ -268,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)
@@ -285,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;
 
@@ -308,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",
@@ -328,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;
 
@@ -337,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;
@@ -367,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;
@@ -390,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,
@@ -424,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);
@@ -457,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;
@@ -467,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;
@@ -491,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) {
@@ -503,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,
@@ -558,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) {
@@ -572,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;
 }
@@ -584,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));
 }
 
 /*
@@ -647,7 +640,7 @@ static void bch2_stripe_close(struct bch_fs *c, struct ec_stripe_new *s)
        BUG_ON(!s->idx);
 
        spin_lock(&c->ec_stripes_new_lock);
-       hlist_del(&s->hash);
+       hlist_del_init(&s->hash);
        spin_unlock(&c->ec_stripes_new_lock);
 
        s->idx = 0;
@@ -658,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;
 }
@@ -761,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;
@@ -813,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;
                }
        }
@@ -826,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);
 }
 
@@ -841,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;
@@ -866,8 +856,16 @@ static int ec_stripe_key_update(struct btree_trans *trans,
                        goto err;
                }
 
-               for (i = 0; i < new->v.nr_blocks; i++)
-                       stripe_blockcount_set(&new->v, i, stripe_blockcount_get(old, i));
+               for (i = 0; i < new->v.nr_blocks; i++) {
+                       unsigned v = stripe_blockcount_get(old, i);
+
+                       BUG_ON(v &&
+                              (old->ptrs[i].dev != new->v.ptrs[i].dev ||
+                               old->ptrs[i].gen != new->v.ptrs[i].gen ||
+                               old->ptrs[i].offset != new->v.ptrs[i].offset));
+
+                       stripe_blockcount_set(&new->v, i, v);
+               }
        }
 
        ret = bch2_trans_update(trans, &iter, &new->k_i, 0);
@@ -879,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;
@@ -892,16 +891,32 @@ 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 (bch2_fs_inconsistent_on(bp.level, c, "found btree node in erasure coded bucket!?"))
+       if (bp.level) {
+               struct printbuf buf = PRINTBUF;
+               struct btree_iter node_iter;
+               struct btree *b;
+
+               b = bch2_backpointer_get_node(trans, &node_iter, *bp_pos, bp);
+               bch2_trans_iter_exit(trans, &node_iter);
+
+               if (!b)
+                       return 0;
+
+               prt_printf(&buf, "found btree node in erasure coded bucket: b=%px\n", b);
+               bch2_backpointer_to_text(&buf, &bp);
+
+               bch2_fs_inconsistent(c, "%s", buf.buf);
+               printbuf_exit(&buf);
                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;
@@ -916,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?
@@ -924,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);
@@ -934,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,
        };
 
@@ -958,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;
@@ -982,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;
 
@@ -1013,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;
        }
 
@@ -1032,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
  */
@@ -1039,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;
 
@@ -1047,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))
@@ -1070,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]);
 
@@ -1078,6 +1104,7 @@ static void ec_stripe_create(struct ec_stripe_new *s)
        }
 
        BUG_ON(!s->allocated);
+       BUG_ON(!s->idx);
 
        ec_generate_ec(&s->new_stripe);
 
@@ -1093,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;
@@ -1107,8 +1137,6 @@ static void ec_stripe_create(struct ec_stripe_new *s)
                        bch2_err_str(ret));
                goto err;
        }
-
-       bch2_stripe_close(c, s);
 err:
        bch2_disk_reservation_put(c, &s->res);
 
@@ -1124,10 +1152,16 @@ err:
                        }
                }
 
+       mutex_lock(&c->ec_stripe_new_lock);
+       list_del(&s->list);
+       mutex_unlock(&c->ec_stripe_new_lock);
+       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)
@@ -1136,10 +1170,8 @@ 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)) {
-                       list_del(&s->list);
+               if (!atomic_read(&s->ref[STRIPE_REF_io]))
                        goto out;
-               }
        s = NULL;
 out:
        mutex_unlock(&c->ec_stripe_new_lock);
@@ -1167,14 +1199,6 @@ void bch2_ec_do_stripe_creates(struct bch_fs *c)
                bch2_write_ref_put(c, BCH_WRITE_REF_stripe_create);
 }
 
-static void ec_stripe_new_put(struct bch_fs *c, struct ec_stripe_new *s)
-{
-       BUG_ON(atomic_read(&s->pin) <= 0);
-
-       if (atomic_dec_and_test(&s->pin))
-               bch2_ec_do_stripe_creates(c);
-}
-
 static void ec_stripe_set_pending(struct bch_fs *c, struct ec_stripe_head *h)
 {
        struct ec_stripe_new *s = h->s;
@@ -1188,15 +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);
-}
-
-/* have a full bucket - hand it off to be erasure coded: */
-void bch2_ec_bucket_written(struct bch_fs *c, struct open_bucket *ob)
-{
-       struct ec_stripe_new *s = ob->ec;
-
-       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)
@@ -1215,6 +1231,8 @@ void *bch2_writepoint_ec_buf(struct bch_fs *c, struct write_point *wp)
        if (!ob)
                return NULL;
 
+       BUG_ON(!ob->ec->new_stripe.data[ob->ec_idx]);
+
        ca      = bch_dev_bkey_exists(c, ob->dev);
        offset  = ca->mi.bucket_size - ob->sectors_free;
 
@@ -1268,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;
@@ -1302,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;
@@ -1323,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,
-                        bool copygc)
+                        enum bch_watermark watermark)
 {
        struct ec_stripe_head *h;
        struct bch_dev *ca;
@@ -1339,7 +1358,7 @@ ec_new_stripe_head_alloc(struct bch_fs *c, unsigned target,
        h->target       = target;
        h->algo         = algo;
        h->redundancy   = redundancy;
-       h->copygc       = copygc;
+       h->watermark    = watermark;
 
        rcu_read_lock();
        h->devs = target_rw_devs(c, BCH_DATA_user, target);
@@ -1370,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,
-                                                bool copygc)
+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;
@@ -1387,42 +1407,49 @@ 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->copygc           == copygc) {
+                   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, copygc);
+       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,
-                                   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;
 
-       for (i = 0; i < h->s->new_stripe.key.v.nr_blocks; i++) {
-               if (test_bit(i, h->s->blocks_gotten)) {
-                       __clear_bit(h->s->new_stripe.key.v.ptrs[i].dev, devs.d);
-                       if (i < h->s->nr_data)
-                               nr_have_data++;
-                       else
-                               nr_have_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, v->nr_blocks) {
+               __clear_bit(v->ptrs[i].dev, devs.d);
+               if (i < h->s->nr_data)
+                       nr_have_data++;
+               else
+                       nr_have_parity++;
        }
 
        BUG_ON(nr_have_data     > h->s->nr_data);
@@ -1435,11 +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,
-                                           h->copygc
-                                           ? RESERVE_movinggc
-                                           : RESERVE_none,
+                                           watermark,
                                            cl);
 
                open_bucket_for_each(c, &buckets, ob, i) {
@@ -1449,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);
                }
 
@@ -1464,11 +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,
-                                           h->copygc
-                                           ? RESERVE_movinggc
-                                           : RESERVE_none,
+                                           watermark,
                                            cl);
 
                open_bucket_for_each(c, &buckets, ob, i) {
@@ -1477,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);
                }
 
@@ -1527,34 +1550,56 @@ 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;
 
+       /*
+        * If we can't allocate a new stripe, and there's no stripes with empty
+        * blocks for us to reuse, that means we have to wait on copygc:
+        */
        idx = get_existing_stripe(c, h);
        if (idx < 0)
-               return -BCH_ERR_ENOSPC_stripe_reuse;
+               return -BCH_ERR_stripe_alloc_blocked;
 
-       h->s->have_existing_stripe = true;
        ret = get_stripe_key_trans(trans, idx, &h->s->existing_stripe);
        if (ret) {
-               bch2_fs_fatal_error(c, "error reading stripe key: %i", ret);
+               bch2_stripe_close(c, h->s);
+               if (!bch2_err_matches(ret, BCH_ERR_transaction_restart))
+                       bch2_fs_fatal_error(c, "error reading stripe key: %s", bch2_err_str(ret));
                return ret;
        }
 
-       if (ec_stripe_buf_init(&h->s->existing_stripe, 0, h->blocksize)) {
-               /*
-                * this is a problem: we have deleted from the
-                * stripes heap already
-                */
-               BUG();
+       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) {
+               bch2_stripe_close(c, h->s);
+               return ret;
        }
 
        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));
 
-       for (i = 0; i < h->s->existing_stripe.key.v.nr_blocks; i++) {
-               if (stripe_blockcount_get(&h->s->existing_stripe.key.v, i)) {
+       /*
+        * 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, 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 < 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);
                }
@@ -1562,8 +1607,8 @@ 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;
 }
@@ -1577,13 +1622,14 @@ static int __bch2_ec_stripe_head_reserve(struct btree_trans *trans, struct ec_st
        struct bpos start_pos = bpos_max(min_pos, POS(0, c->ec_stripe_hint));
        int ret;
 
-       BUG_ON(h->s->res.sectors);
-
-       ret = bch2_disk_reservation_get(c, &h->s->res,
+       if (!h->s->res.sectors) {
+               ret = bch2_disk_reservation_get(c, &h->s->res,
                                        h->blocksize,
-                                       h->s->nr_parity, 0);
-       if (ret)
-               return ret;
+                                       h->s->nr_parity,
+                                       BCH_DISK_RESERVATION_NOFAIL);
+               if (ret)
+                       return ret;
+       }
 
        for_each_btree_key_norestart(trans, iter, BTREE_ID_stripes, start_pos,
                           BTREE_ITER_SLOTS|BTREE_ITER_INTENT, k, ret) {
@@ -1627,58 +1673,87 @@ struct ec_stripe_head *bch2_ec_stripe_head_get(struct btree_trans *trans,
                                               unsigned target,
                                               unsigned algo,
                                               unsigned redundancy,
-                                              bool copygc,
+                                              enum bch_watermark watermark,
                                               struct closure *cl)
 {
        struct bch_fs *c = trans->c;
        struct ec_stripe_head *h;
+       bool waiting = false;
        int ret;
-       bool needs_stripe_new;
 
-       h = __bch2_ec_stripe_head_get(trans, target, algo, redundancy, copygc);
+       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;
 
-       needs_stripe_new = !h->s;
-       if (needs_stripe_new) {
-               if (ec_new_stripe_alloc(c, h)) {
-                       ret = -ENOMEM;
+       if (!h->s) {
+               ret = ec_new_stripe_alloc(c, h);
+               if (ret) {
                        bch_err(c, "failed to allocate new stripe");
                        goto err;
                }
-
-               if (ec_stripe_buf_init(&h->s->new_stripe, 0, h->blocksize))
-                       BUG();
        }
 
-       /*
-        * Try reserve a new stripe before reusing an
-        * existing stripe. This will prevent unnecessary
-        * read amplification during write oriented workloads.
-        */
-       ret = 0;
-       if (!h->s->allocated && !h->s->res.sectors && !h->s->have_existing_stripe)
-               ret = __bch2_ec_stripe_head_reserve(trans, h);
-       if (bch2_err_matches(ret, BCH_ERR_transaction_restart))
-               goto err;
+       if (h->s->allocated)
+               goto allocated;
 
-       if (ret && needs_stripe_new)
-               ret = __bch2_ec_stripe_head_reuse(trans, h);
-       if (ret) {
-               bch_err_ratelimited(c, "failed to get stripe: %s", bch2_err_str(ret));
+       if (h->s->have_existing_stripe)
+               goto alloc_existing;
+
+       /* First, try to allocate a full stripe: */
+       ret =   new_stripe_alloc_buckets(trans, h, BCH_WATERMARK_stripe, NULL) ?:
+               __bch2_ec_stripe_head_reserve(trans, h);
+       if (!ret)
+               goto allocate_buf;
+       if (bch2_err_matches(ret, BCH_ERR_transaction_restart) ||
+           bch2_err_matches(ret, ENOMEM))
                goto err;
-       }
 
-       if (!h->s->allocated) {
-               ret = new_stripe_alloc_buckets(trans, h, cl);
-               if (ret)
+       /*
+        * Not enough buckets available for a full stripe: we must reuse an
+        * existing stripe:
+        */
+       while (1) {
+               ret = __bch2_ec_stripe_head_reuse(trans, h);
+               if (!ret)
+                       break;
+               if (waiting || !cl || ret != -BCH_ERR_stripe_alloc_blocked)
                        goto err;
 
-               h->s->allocated = true;
+               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;
        }
 
+       if (waiting)
+               closure_wake_up(&c->freelist_wait);
+alloc_existing:
+       /*
+        * Retry allocating buckets, with the watermark for this
+        * particular write:
+        */
+       ret = new_stripe_alloc_buckets(trans, h, watermark, cl);
+       if (ret)
+               goto err;
+
+allocate_buf:
+       ret = ec_stripe_buf_init(&h->s->new_stripe, 0, h->blocksize);
+       if (ret)
+               goto err;
+
+       h->s->allocated = true;
+allocated:
+       BUG_ON(!h->s->idx);
+       BUG_ON(!h->s->new_stripe.data[0]);
        BUG_ON(trans->restarted);
        return h;
 err:
@@ -1686,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;
@@ -1694,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;
 
@@ -1709,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);
@@ -1717,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;
@@ -1757,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;
 }
@@ -1769,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);
 }
@@ -1787,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: blocks %u+%u allocated %u\n",
-                              h->s->nr_data, h->s->nr_parity,
+                       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: blocks %u+%u pin %u\n",
-                      s->nr_data, s->nr_parity,
-                      atomic_read(&s->pin));
+               prt_printf(out, "\tidx %llu blocks %u+%u ref %u %u %s\n",
+                          s->idx, s->nr_data, s->nr_parity,
+                          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);
 }
@@ -1823,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);
@@ -1840,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);
 }