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