git.sesse.net Git - bcachefs-tools-debian/blob - libbcachefs/ec.h

   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 */