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

   1 /* SPDX-License-Identifier: GPL-2.0 */
   2 /*
   3  * Code for manipulating bucket marks for garbage collection.
   4  *
   5  * Copyright 2014 Datera, Inc.
   6  */
   7
   8 #ifndef _BUCKETS_H
   9 #define _BUCKETS_H
  10
  11 #include "buckets_types.h"
  12 #include "extents.h"
  13 #include "super.h"
  14
  15 #define for_each_bucket(_b, _buckets)                           \
  16         for (_b = (_buckets)->b + (_buckets)->first_bucket;     \
  17              _b < (_buckets)->b + (_buckets)->nbuckets; _b++)
  18
  19 static inline void bucket_unlock(struct bucket *b)
  20 {
  21         smp_store_release(&b->lock, 0);
  22 }
  23
  24 static inline void bucket_lock(struct bucket *b)
  25 {
  26         while (xchg(&b->lock, 1))
  27                 cpu_relax();
  28 }
  29
  30 static inline struct bucket_array *gc_bucket_array(struct bch_dev *ca)
  31 {
  32         return rcu_dereference_check(ca->buckets_gc,
  33                                      !ca->fs ||
  34                                      percpu_rwsem_is_held(&ca->fs->mark_lock) ||
  35                                      lockdep_is_held(&ca->fs->gc_lock) ||
  36                                      lockdep_is_held(&ca->bucket_lock));
  37 }
  38
  39 static inline struct bucket *gc_bucket(struct bch_dev *ca, size_t b)
  40 {
  41         struct bucket_array *buckets = gc_bucket_array(ca);
  42
  43         BUG_ON(b < buckets->first_bucket || b >= buckets->nbuckets);
  44         return buckets->b + b;
  45 }
  46
  47 static inline struct bucket_gens *bucket_gens(struct bch_dev *ca)
  48 {
  49         return rcu_dereference_check(ca->bucket_gens,
  50                                      !ca->fs ||
  51                                      percpu_rwsem_is_held(&ca->fs->mark_lock) ||
  52                                      lockdep_is_held(&ca->fs->gc_lock) ||
  53                                      lockdep_is_held(&ca->bucket_lock));
  54 }
  55
  56 static inline u8 *bucket_gen(struct bch_dev *ca, size_t b)
  57 {
  58         struct bucket_gens *gens = bucket_gens(ca);
  59
  60         BUG_ON(b < gens->first_bucket || b >= gens->nbuckets);
  61         return gens->b + b;
  62 }
  63
  64 static inline size_t PTR_BUCKET_NR(const struct bch_dev *ca,
  65                                    const struct bch_extent_ptr *ptr)
  66 {
  67         return sector_to_bucket(ca, ptr->offset);
  68 }
  69
  70 static inline struct bpos PTR_BUCKET_POS(const struct bch_fs *c,
  71                                    const struct bch_extent_ptr *ptr)
  72 {
  73         struct bch_dev *ca = bch_dev_bkey_exists(c, ptr->dev);
  74
  75         return POS(ptr->dev, PTR_BUCKET_NR(ca, ptr));
  76 }
  77
  78 static inline struct bpos PTR_BUCKET_POS_OFFSET(const struct bch_fs *c,
  79                                                 const struct bch_extent_ptr *ptr,
  80                                                 u32 *bucket_offset)
  81 {
  82         struct bch_dev *ca = bch_dev_bkey_exists(c, ptr->dev);
  83
  84         return POS(ptr->dev, sector_to_bucket_and_offset(ca, ptr->offset, bucket_offset));
  85 }
  86
  87 static inline struct bucket *PTR_GC_BUCKET(struct bch_dev *ca,
  88                                            const struct bch_extent_ptr *ptr)
  89 {
  90         return gc_bucket(ca, PTR_BUCKET_NR(ca, ptr));
  91 }
  92
  93 static inline enum bch_data_type ptr_data_type(const struct bkey *k,
  94                                                const struct bch_extent_ptr *ptr)
  95 {
  96         if (bkey_is_btree_ptr(k))
  97                 return BCH_DATA_btree;
  98
  99         return ptr->cached ? BCH_DATA_cached : BCH_DATA_user;
 100 }
 101
 102 static inline s64 ptr_disk_sectors(s64 sectors, struct extent_ptr_decoded p)
 103 {
 104         EBUG_ON(sectors < 0);
 105
 106         return crc_is_compressed(p.crc)
 107                 ? DIV_ROUND_UP_ULL(sectors * p.crc.compressed_size,
 108                                    p.crc.uncompressed_size)
 109                 : sectors;
 110 }
 111
 112 static inline int gen_cmp(u8 a, u8 b)
 113 {
 114         return (s8) (a - b);
 115 }
 116
 117 static inline int gen_after(u8 a, u8 b)
 118 {
 119         int r = gen_cmp(a, b);
 120
 121         return r > 0 ? r : 0;
 122 }
 123
 124 /**
 125  * ptr_stale() - check if a pointer points into a bucket that has been
 126  * invalidated.
 127  */
 128 static inline u8 ptr_stale(struct bch_dev *ca,
 129                            const struct bch_extent_ptr *ptr)
 130 {
 131         u8 ret;
 132
 133         rcu_read_lock();
 134         ret = gen_after(*bucket_gen(ca, PTR_BUCKET_NR(ca, ptr)), ptr->gen);
 135         rcu_read_unlock();
 136
 137         return ret;
 138 }
 139
 140 /* Device usage: */
 141
 142 void bch2_dev_usage_read_fast(struct bch_dev *, struct bch_dev_usage *);
 143 static inline struct bch_dev_usage bch2_dev_usage_read(struct bch_dev *ca)
 144 {
 145         struct bch_dev_usage ret;
 146
 147         bch2_dev_usage_read_fast(ca, &ret);
 148         return ret;
 149 }
 150
 151 void bch2_dev_usage_init(struct bch_dev *);
 152
 153 static inline u64 bch2_dev_buckets_reserved(struct bch_dev *ca, enum bch_watermark watermark)
 154 {
 155         s64 reserved = 0;
 156
 157         switch (watermark) {
 158         case BCH_WATERMARK_NR:
 159                 unreachable();
 160         case BCH_WATERMARK_stripe:
 161                 reserved += ca->mi.nbuckets >> 6;
 162                 fallthrough;
 163         case BCH_WATERMARK_normal:
 164                 reserved += ca->mi.nbuckets >> 6;
 165                 fallthrough;
 166         case BCH_WATERMARK_copygc:
 167                 reserved += ca->nr_btree_reserve;
 168                 fallthrough;
 169         case BCH_WATERMARK_btree:
 170                 reserved += ca->nr_btree_reserve;
 171                 fallthrough;
 172         case BCH_WATERMARK_btree_copygc:
 173         case BCH_WATERMARK_reclaim:
 174                 break;
 175         }
 176
 177         return reserved;
 178 }
 179
 180 static inline u64 dev_buckets_free(struct bch_dev *ca,
 181                                    struct bch_dev_usage usage,
 182                                    enum bch_watermark watermark)
 183 {
 184         return max_t(s64, 0,
 185                      usage.d[BCH_DATA_free].buckets -
 186                      ca->nr_open_buckets -
 187                      bch2_dev_buckets_reserved(ca, watermark));
 188 }
 189
 190 static inline u64 __dev_buckets_available(struct bch_dev *ca,
 191                                           struct bch_dev_usage usage,
 192                                           enum bch_watermark watermark)
 193 {
 194         return max_t(s64, 0,
 195                        usage.d[BCH_DATA_free].buckets
 196                      + usage.d[BCH_DATA_cached].buckets
 197                      + usage.d[BCH_DATA_need_gc_gens].buckets
 198                      + usage.d[BCH_DATA_need_discard].buckets
 199                      - ca->nr_open_buckets
 200                      - bch2_dev_buckets_reserved(ca, watermark));
 201 }
 202
 203 static inline u64 dev_buckets_available(struct bch_dev *ca,
 204                                         enum bch_watermark watermark)
 205 {
 206         return __dev_buckets_available(ca, bch2_dev_usage_read(ca), watermark);
 207 }
 208
 209 /* Filesystem usage: */
 210
 211 static inline unsigned __fs_usage_u64s(unsigned nr_replicas)
 212 {
 213         return sizeof(struct bch_fs_usage) / sizeof(u64) + nr_replicas;
 214 }
 215
 216 static inline unsigned fs_usage_u64s(struct bch_fs *c)
 217 {
 218         return __fs_usage_u64s(READ_ONCE(c->replicas.nr));
 219 }
 220
 221 static inline unsigned __fs_usage_online_u64s(unsigned nr_replicas)
 222 {
 223         return sizeof(struct bch_fs_usage_online) / sizeof(u64) + nr_replicas;
 224 }
 225
 226 static inline unsigned fs_usage_online_u64s(struct bch_fs *c)
 227 {
 228         return __fs_usage_online_u64s(READ_ONCE(c->replicas.nr));
 229 }
 230
 231 static inline unsigned dev_usage_u64s(void)
 232 {
 233         return sizeof(struct bch_dev_usage) / sizeof(u64);
 234 }
 235
 236 u64 bch2_fs_usage_read_one(struct bch_fs *, u64 *);
 237
 238 struct bch_fs_usage_online *bch2_fs_usage_read(struct bch_fs *);
 239
 240 void bch2_fs_usage_acc_to_base(struct bch_fs *, unsigned);
 241
 242 void bch2_fs_usage_to_text(struct printbuf *,
 243                            struct bch_fs *, struct bch_fs_usage_online *);
 244
 245 u64 bch2_fs_sectors_used(struct bch_fs *, struct bch_fs_usage_online *);
 246
 247 struct bch_fs_usage_short
 248 bch2_fs_usage_read_short(struct bch_fs *);
 249
 250 /* key/bucket marking: */
 251
 252 static inline struct bch_fs_usage *fs_usage_ptr(struct bch_fs *c,
 253                                                 unsigned journal_seq,
 254                                                 bool gc)
 255 {
 256         percpu_rwsem_assert_held(&c->mark_lock);
 257         BUG_ON(!gc && !journal_seq);
 258
 259         return this_cpu_ptr(gc
 260                             ? c->usage_gc
 261                             : c->usage[journal_seq & JOURNAL_BUF_MASK]);
 262 }
 263
 264 int bch2_replicas_deltas_realloc(struct btree_trans *, unsigned);
 265
 266 void bch2_fs_usage_initialize(struct bch_fs *);
 267
 268 int bch2_mark_metadata_bucket(struct bch_fs *, struct bch_dev *,
 269                               size_t, enum bch_data_type, unsigned,
 270                               struct gc_pos, unsigned);
 271
 272 int bch2_mark_alloc(struct btree_trans *, enum btree_id, unsigned,
 273                     struct bkey_s_c, struct bkey_s_c, unsigned);
 274 int bch2_mark_extent(struct btree_trans *, enum btree_id, unsigned,
 275                      struct bkey_s_c, struct bkey_s_c, unsigned);
 276 int bch2_mark_stripe(struct btree_trans *, enum btree_id, unsigned,
 277                      struct bkey_s_c, struct bkey_s_c, unsigned);
 278 int bch2_mark_reservation(struct btree_trans *, enum btree_id, unsigned,
 279                           struct bkey_s_c, struct bkey_s_c, unsigned);
 280 int bch2_mark_reflink_p(struct btree_trans *, enum btree_id, unsigned,
 281                         struct bkey_s_c, struct bkey_s_c, unsigned);
 282
 283 int bch2_trans_mark_extent(struct btree_trans *, enum btree_id, unsigned, struct bkey_s_c, struct bkey_i *, unsigned);
 284 int bch2_trans_mark_stripe(struct btree_trans *, enum btree_id, unsigned, struct bkey_s_c, struct bkey_i *, unsigned);
 285 int bch2_trans_mark_reservation(struct btree_trans *, enum btree_id, unsigned, struct bkey_s_c, struct bkey_i *, unsigned);
 286 int bch2_trans_mark_reflink_p(struct btree_trans *, enum btree_id, unsigned, struct bkey_s_c, struct bkey_i *, unsigned);
 287
 288 void bch2_trans_fs_usage_revert(struct btree_trans *, struct replicas_delta_list *);
 289 int bch2_trans_fs_usage_apply(struct btree_trans *, struct replicas_delta_list *);
 290
 291 int bch2_trans_mark_metadata_bucket(struct btree_trans *, struct bch_dev *,
 292                                     size_t, enum bch_data_type, unsigned);
 293 int bch2_trans_mark_dev_sb(struct bch_fs *, struct bch_dev *);
 294
 295 /* disk reservations: */
 296
 297 static inline void bch2_disk_reservation_put(struct bch_fs *c,
 298                                              struct disk_reservation *res)
 299 {
 300         if (res->sectors) {
 301                 this_cpu_sub(*c->online_reserved, res->sectors);
 302                 res->sectors = 0;
 303         }
 304 }
 305
 306 #define BCH_DISK_RESERVATION_NOFAIL             (1 << 0)
 307
 308 int __bch2_disk_reservation_add(struct bch_fs *,
 309                                 struct disk_reservation *,
 310                                 u64, int);
 311
 312 static inline int bch2_disk_reservation_add(struct bch_fs *c, struct disk_reservation *res,
 313                                             u64 sectors, int flags)
 314 {
 315 #ifdef __KERNEL__
 316         u64 old, new;
 317
 318         do {
 319                 old = this_cpu_read(c->pcpu->sectors_available);
 320                 if (sectors > old)
 321                         return __bch2_disk_reservation_add(c, res, sectors, flags);
 322
 323                 new = old - sectors;
 324         } while (this_cpu_cmpxchg(c->pcpu->sectors_available, old, new) != old);
 325
 326         this_cpu_add(*c->online_reserved, sectors);
 327         res->sectors                    += sectors;
 328         return 0;
 329 #else
 330         return __bch2_disk_reservation_add(c, res, sectors, flags);
 331 #endif
 332 }
 333
 334 static inline struct disk_reservation
 335 bch2_disk_reservation_init(struct bch_fs *c, unsigned nr_replicas)
 336 {
 337         return (struct disk_reservation) {
 338                 .sectors        = 0,
 339 #if 0
 340                 /* not used yet: */
 341                 .gen            = c->capacity_gen,
 342 #endif
 343                 .nr_replicas    = nr_replicas,
 344         };
 345 }
 346
 347 static inline int bch2_disk_reservation_get(struct bch_fs *c,
 348                                             struct disk_reservation *res,
 349                                             u64 sectors, unsigned nr_replicas,
 350                                             int flags)
 351 {
 352         *res = bch2_disk_reservation_init(c, nr_replicas);
 353
 354         return bch2_disk_reservation_add(c, res, sectors * nr_replicas, flags);
 355 }
 356
 357 #define RESERVE_FACTOR  6
 358
 359 static inline u64 avail_factor(u64 r)
 360 {
 361         return div_u64(r << RESERVE_FACTOR, (1 << RESERVE_FACTOR) + 1);
 362 }
 363
 364 int bch2_dev_buckets_resize(struct bch_fs *, struct bch_dev *, u64);
 365 void bch2_dev_buckets_free(struct bch_dev *);
 366 int bch2_dev_buckets_alloc(struct bch_fs *, struct bch_dev *);
 367
 368 #endif /* _BUCKETS_H */