1/* SPDX-License-Identifier: GPL-2.0 */
2#ifndef _BCACHEFS_EC_H
3#define _BCACHEFS_EC_H
4
5#include "ec_types.h"
6#include "buckets_types.h"
7#include "extents_types.h"
8
9enum bkey_invalid_flags;
10
11int bch2_stripe_invalid(struct bch_fs *, struct bkey_s_c,
12			enum bkey_invalid_flags, struct printbuf *);
13void bch2_stripe_to_text(struct printbuf *, struct bch_fs *,
14			 struct bkey_s_c);
15int bch2_trigger_stripe(struct btree_trans *, enum btree_id, unsigned,
16			struct bkey_s_c, struct bkey_s, unsigned);
17
18#define bch2_bkey_ops_stripe ((struct bkey_ops) {	\
19	.key_invalid	= bch2_stripe_invalid,		\
20	.val_to_text	= bch2_stripe_to_text,		\
21	.swab		= bch2_ptr_swab,		\
22	.trigger	= bch2_trigger_stripe,		\
23	.min_val_size	= 8,				\
24})
25
26static inline unsigned stripe_csums_per_device(const struct bch_stripe *s)
27{
28	return DIV_ROUND_UP(le16_to_cpu(s->sectors),
29			    1 << s->csum_granularity_bits);
30}
31
32static inline unsigned stripe_csum_offset(const struct bch_stripe *s,
33					  unsigned dev, unsigned csum_idx)
34{
35	EBUG_ON(s->csum_type >= BCH_CSUM_NR);
36
37	unsigned csum_bytes = bch_crc_bytes[s->csum_type];
38
39	return sizeof(struct bch_stripe) +
40		sizeof(struct bch_extent_ptr) * s->nr_blocks +
41		(dev * stripe_csums_per_device(s) + csum_idx) * csum_bytes;
42}
43
44static inline unsigned stripe_blockcount_offset(const struct bch_stripe *s,
45						unsigned idx)
46{
47	return stripe_csum_offset(s, s->nr_blocks, 0) +
48		sizeof(u16) * idx;
49}
50
51static inline unsigned stripe_blockcount_get(const struct bch_stripe *s,
52					     unsigned idx)
53{
54	return le16_to_cpup((void *) s + stripe_blockcount_offset(s, idx));
55}
56
57static inline void stripe_blockcount_set(struct bch_stripe *s,
58					 unsigned idx, unsigned v)
59{
60	__le16 *p = (void *) s + stripe_blockcount_offset(s, idx);
61
62	*p = cpu_to_le16(v);
63}
64
65static inline unsigned stripe_val_u64s(const struct bch_stripe *s)
66{
67	return DIV_ROUND_UP(stripe_blockcount_offset(s, s->nr_blocks),
68			    sizeof(u64));
69}
70
71static inline void *stripe_csum(struct bch_stripe *s,
72				unsigned block, unsigned csum_idx)
73{
74	EBUG_ON(block >= s->nr_blocks);
75	EBUG_ON(csum_idx >= stripe_csums_per_device(s));
76
77	return (void *) s + stripe_csum_offset(s, block, csum_idx);
78}
79
80static inline struct bch_csum stripe_csum_get(struct bch_stripe *s,
81				   unsigned block, unsigned csum_idx)
82{
83	struct bch_csum csum = { 0 };
84
85	memcpy(&csum, stripe_csum(s, block, csum_idx), bch_crc_bytes[s->csum_type]);
86	return csum;
87}
88
89static inline void stripe_csum_set(struct bch_stripe *s,
90				   unsigned block, unsigned csum_idx,
91				   struct bch_csum csum)
92{
93	memcpy(stripe_csum(s, block, csum_idx), &csum, bch_crc_bytes[s->csum_type]);
94}
95
96static inline bool __bch2_ptr_matches_stripe(const struct bch_extent_ptr *stripe_ptr,
97					     const struct bch_extent_ptr *data_ptr,
98					     unsigned sectors)
99{
100	return  data_ptr->dev    == stripe_ptr->dev &&
101		data_ptr->gen    == stripe_ptr->gen &&
102		data_ptr->offset >= stripe_ptr->offset &&
103		data_ptr->offset  < stripe_ptr->offset + sectors;
104}
105
106static inline bool bch2_ptr_matches_stripe(const struct bch_stripe *s,
107					   struct extent_ptr_decoded p)
108{
109	unsigned nr_data = s->nr_blocks - s->nr_redundant;
110
111	BUG_ON(!p.has_ec);
112
113	if (p.ec.block >= nr_data)
114		return false;
115
116	return __bch2_ptr_matches_stripe(&s->ptrs[p.ec.block], &p.ptr,
117					 le16_to_cpu(s->sectors));
118}
119
120static inline bool bch2_ptr_matches_stripe_m(const struct gc_stripe *m,
121					     struct extent_ptr_decoded p)
122{
123	unsigned nr_data = m->nr_blocks - m->nr_redundant;
124
125	BUG_ON(!p.has_ec);
126
127	if (p.ec.block >= nr_data)
128		return false;
129
130	return __bch2_ptr_matches_stripe(&m->ptrs[p.ec.block], &p.ptr,
131					 m->sectors);
132}
133
134struct bch_read_bio;
135
136struct ec_stripe_buf {
137	/* might not be buffering the entire stripe: */
138	unsigned		offset;
139	unsigned		size;
140	unsigned long		valid[BITS_TO_LONGS(BCH_BKEY_PTRS_MAX)];
141
142	void			*data[BCH_BKEY_PTRS_MAX];
143
144	__BKEY_PADDED(key, 255);
145};
146
147struct ec_stripe_head;
148
149enum ec_stripe_ref {
150	STRIPE_REF_io,
151	STRIPE_REF_stripe,
152	STRIPE_REF_NR
153};
154
155struct ec_stripe_new {
156	struct bch_fs		*c;
157	struct ec_stripe_head	*h;
158	struct mutex		lock;
159	struct list_head	list;
160
161	struct hlist_node	hash;
162	u64			idx;
163
164	struct closure		iodone;
165
166	atomic_t		ref[STRIPE_REF_NR];
167
168	int			err;
169
170	u8			nr_data;
171	u8			nr_parity;
172	bool			allocated;
173	bool			pending;
174	bool			have_existing_stripe;
175
176	unsigned long		blocks_gotten[BITS_TO_LONGS(BCH_BKEY_PTRS_MAX)];
177	unsigned long		blocks_allocated[BITS_TO_LONGS(BCH_BKEY_PTRS_MAX)];
178	open_bucket_idx_t	blocks[BCH_BKEY_PTRS_MAX];
179	struct disk_reservation	res;
180
181	struct ec_stripe_buf	new_stripe;
182	struct ec_stripe_buf	existing_stripe;
183};
184
185struct ec_stripe_head {
186	struct list_head	list;
187	struct mutex		lock;
188
189	unsigned		target;
190	unsigned		algo;
191	unsigned		redundancy;
192	enum bch_watermark	watermark;
193
194	struct bch_devs_mask	devs;
195	unsigned		nr_active_devs;
196
197	unsigned		blocksize;
198
199	struct dev_stripe_state	block_stripe;
200	struct dev_stripe_state	parity_stripe;
201
202	struct ec_stripe_new	*s;
203};
204
205int bch2_ec_read_extent(struct btree_trans *, struct bch_read_bio *);
206
207void *bch2_writepoint_ec_buf(struct bch_fs *, struct write_point *);
208
209void bch2_ec_bucket_cancel(struct bch_fs *, struct open_bucket *);
210
211int bch2_ec_stripe_new_alloc(struct bch_fs *, struct ec_stripe_head *);
212
213void bch2_ec_stripe_head_put(struct bch_fs *, struct ec_stripe_head *);
214struct ec_stripe_head *bch2_ec_stripe_head_get(struct btree_trans *,
215			unsigned, unsigned, unsigned,
216			enum bch_watermark, struct closure *);
217
218void bch2_stripes_heap_update(struct bch_fs *, struct stripe *, size_t);
219void bch2_stripes_heap_del(struct bch_fs *, struct stripe *, size_t);
220void bch2_stripes_heap_insert(struct bch_fs *, struct stripe *, size_t);
221
222void bch2_do_stripe_deletes(struct bch_fs *);
223void bch2_ec_do_stripe_creates(struct bch_fs *);
224void bch2_ec_stripe_new_free(struct bch_fs *, struct ec_stripe_new *);
225
226static inline void ec_stripe_new_get(struct ec_stripe_new *s,
227				     enum ec_stripe_ref ref)
228{
229	atomic_inc(&s->ref[ref]);
230}
231
232static inline void ec_stripe_new_put(struct bch_fs *c, struct ec_stripe_new *s,
233				     enum ec_stripe_ref ref)
234{
235	BUG_ON(atomic_read(&s->ref[ref]) <= 0);
236
237	if (atomic_dec_and_test(&s->ref[ref]))
238		switch (ref) {
239		case STRIPE_REF_stripe:
240			bch2_ec_stripe_new_free(c, s);
241			break;
242		case STRIPE_REF_io:
243			bch2_ec_do_stripe_creates(c);
244			break;
245		default:
246			BUG();
247		}
248}
249
250void bch2_ec_stop_dev(struct bch_fs *, struct bch_dev *);
251void bch2_fs_ec_stop(struct bch_fs *);
252void bch2_fs_ec_flush(struct bch_fs *);
253
254int bch2_stripes_read(struct bch_fs *);
255
256void bch2_stripes_heap_to_text(struct printbuf *, struct bch_fs *);
257void bch2_new_stripes_to_text(struct printbuf *, struct bch_fs *);
258
259void bch2_fs_ec_exit(struct bch_fs *);
260void bch2_fs_ec_init_early(struct bch_fs *);
261int bch2_fs_ec_init(struct bch_fs *);
262
263#endif /* _BCACHEFS_EC_H */
264