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