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

   1 /* SPDX-License-Identifier: GPL-2.0 */
   2 #ifndef _BCACHEFS_SUPER_H
   3 #define _BCACHEFS_SUPER_H
   4
   5 #include "extents.h"
   6
   7 #include "bcachefs_ioctl.h"
   8
   9 #include <linux/math64.h>
  10
  11 static inline size_t sector_to_bucket(const struct bch_dev *ca, sector_t s)
  12 {
  13         return div_u64(s, ca->mi.bucket_size);
  14 }
  15
  16 static inline sector_t bucket_to_sector(const struct bch_dev *ca, size_t b)
  17 {
  18         return ((sector_t) b) * ca->mi.bucket_size;
  19 }
  20
  21 static inline sector_t bucket_remainder(const struct bch_dev *ca, sector_t s)
  22 {
  23         u32 remainder;
  24
  25         div_u64_rem(s, ca->mi.bucket_size, &remainder);
  26         return remainder;
  27 }
  28
  29 static inline bool bch2_dev_is_online(struct bch_dev *ca)
  30 {
  31         return !percpu_ref_is_zero(&ca->io_ref);
  32 }
  33
  34 static inline bool bch2_dev_is_readable(struct bch_dev *ca)
  35 {
  36         return bch2_dev_is_online(ca) &&
  37                 ca->mi.state != BCH_MEMBER_STATE_failed;
  38 }
  39
  40 static inline bool bch2_dev_get_ioref(struct bch_dev *ca, int rw)
  41 {
  42         if (!percpu_ref_tryget(&ca->io_ref))
  43                 return false;
  44
  45         if (ca->mi.state == BCH_MEMBER_STATE_rw ||
  46             (ca->mi.state == BCH_MEMBER_STATE_ro && rw == READ))
  47                 return true;
  48
  49         percpu_ref_put(&ca->io_ref);
  50         return false;
  51 }
  52
  53 static inline unsigned dev_mask_nr(const struct bch_devs_mask *devs)
  54 {
  55         return bitmap_weight(devs->d, BCH_SB_MEMBERS_MAX);
  56 }
  57
  58 static inline bool bch2_dev_list_has_dev(struct bch_devs_list devs,
  59                                          unsigned dev)
  60 {
  61         unsigned i;
  62
  63         for (i = 0; i < devs.nr; i++)
  64                 if (devs.devs[i] == dev)
  65                         return true;
  66
  67         return false;
  68 }
  69
  70 static inline void bch2_dev_list_drop_dev(struct bch_devs_list *devs,
  71                                           unsigned dev)
  72 {
  73         unsigned i;
  74
  75         for (i = 0; i < devs->nr; i++)
  76                 if (devs->devs[i] == dev) {
  77                         array_remove_item(devs->devs, devs->nr, i);
  78                         return;
  79                 }
  80 }
  81
  82 static inline void bch2_dev_list_add_dev(struct bch_devs_list *devs,
  83                                          unsigned dev)
  84 {
  85         BUG_ON(bch2_dev_list_has_dev(*devs, dev));
  86         BUG_ON(devs->nr >= BCH_REPLICAS_MAX);
  87         devs->devs[devs->nr++] = dev;
  88 }
  89
  90 static inline struct bch_devs_list bch2_dev_list_single(unsigned dev)
  91 {
  92         return (struct bch_devs_list) { .nr = 1, .devs[0] = dev };
  93 }
  94
  95 static inline struct bch_dev *__bch2_next_dev(struct bch_fs *c, unsigned *iter,
  96                                               const struct bch_devs_mask *mask)
  97 {
  98         struct bch_dev *ca = NULL;
  99
 100         while ((*iter = mask
 101                 ? find_next_bit(mask->d, c->sb.nr_devices, *iter)
 102                 : *iter) < c->sb.nr_devices &&
 103                !(ca = rcu_dereference_check(c->devs[*iter],
 104                                             lockdep_is_held(&c->state_lock))))
 105                 (*iter)++;
 106
 107         return ca;
 108 }
 109
 110 #define for_each_member_device_rcu(ca, c, iter, mask)                   \
 111         for ((iter) = 0; ((ca) = __bch2_next_dev((c), &(iter), mask)); (iter)++)
 112
 113 static inline struct bch_dev *bch2_get_next_dev(struct bch_fs *c, unsigned *iter)
 114 {
 115         struct bch_dev *ca;
 116
 117         rcu_read_lock();
 118         if ((ca = __bch2_next_dev(c, iter, NULL)))
 119                 percpu_ref_get(&ca->ref);
 120         rcu_read_unlock();
 121
 122         return ca;
 123 }
 124
 125 /*
 126  * If you break early, you must drop your ref on the current device
 127  */
 128 #define for_each_member_device(ca, c, iter)                             \
 129         for ((iter) = 0;                                                \
 130              (ca = bch2_get_next_dev(c, &(iter)));                      \
 131              percpu_ref_put(&ca->ref), (iter)++)
 132
 133 static inline struct bch_dev *bch2_get_next_online_dev(struct bch_fs *c,
 134                                                       unsigned *iter,
 135                                                       int state_mask)
 136 {
 137         struct bch_dev *ca;
 138
 139         rcu_read_lock();
 140         while ((ca = __bch2_next_dev(c, iter, NULL)) &&
 141                (!((1 << ca->mi.state) & state_mask) ||
 142                 !percpu_ref_tryget(&ca->io_ref)))
 143                 (*iter)++;
 144         rcu_read_unlock();
 145
 146         return ca;
 147 }
 148
 149 #define __for_each_online_member(ca, c, iter, state_mask)               \
 150         for ((iter) = 0;                                                \
 151              (ca = bch2_get_next_online_dev(c, &(iter), state_mask));   \
 152              percpu_ref_put(&ca->io_ref), (iter)++)
 153
 154 #define for_each_online_member(ca, c, iter)                             \
 155         __for_each_online_member(ca, c, iter, ~0)
 156
 157 #define for_each_rw_member(ca, c, iter)                                 \
 158         __for_each_online_member(ca, c, iter, 1 << BCH_MEMBER_STATE_rw)
 159
 160 #define for_each_readable_member(ca, c, iter)                           \
 161         __for_each_online_member(ca, c, iter,                           \
 162                 (1 << BCH_MEMBER_STATE_rw)|(1 << BCH_MEMBER_STATE_ro))
 163
 164 /*
 165  * If a key exists that references a device, the device won't be going away and
 166  * we can omit rcu_read_lock():
 167  */
 168 static inline struct bch_dev *bch_dev_bkey_exists(const struct bch_fs *c, unsigned idx)
 169 {
 170         EBUG_ON(idx >= c->sb.nr_devices || !c->devs[idx]);
 171
 172         return rcu_dereference_check(c->devs[idx], 1);
 173 }
 174
 175 static inline struct bch_dev *bch_dev_locked(struct bch_fs *c, unsigned idx)
 176 {
 177         EBUG_ON(idx >= c->sb.nr_devices || !c->devs[idx]);
 178
 179         return rcu_dereference_protected(c->devs[idx],
 180                                          lockdep_is_held(&c->sb_lock) ||
 181                                          lockdep_is_held(&c->state_lock));
 182 }
 183
 184 /* XXX kill, move to struct bch_fs */
 185 static inline struct bch_devs_mask bch2_online_devs(struct bch_fs *c)
 186 {
 187         struct bch_devs_mask devs;
 188         struct bch_dev *ca;
 189         unsigned i;
 190
 191         memset(&devs, 0, sizeof(devs));
 192         for_each_online_member(ca, c, i)
 193                 __set_bit(ca->dev_idx, devs.d);
 194         return devs;
 195 }
 196
 197 static inline bool is_superblock_bucket(struct bch_dev *ca, u64 b)
 198 {
 199         struct bch_sb_layout *layout = &ca->disk_sb.sb->layout;
 200         u64 b_offset    = bucket_to_sector(ca, b);
 201         u64 b_end       = bucket_to_sector(ca, b + 1);
 202         unsigned i;
 203
 204         if (!b)
 205                 return true;
 206
 207         for (i = 0; i < layout->nr_superblocks; i++) {
 208                 u64 offset = le64_to_cpu(layout->sb_offset[i]);
 209                 u64 end = offset + (1 << layout->sb_max_size_bits);
 210
 211                 if (!(offset >= b_end || end <= b_offset))
 212                         return true;
 213         }
 214
 215         return false;
 216 }
 217
 218 struct bch_fs *bch2_dev_to_fs(dev_t);
 219 struct bch_fs *bch2_uuid_to_fs(uuid_le);
 220
 221 bool bch2_dev_state_allowed(struct bch_fs *, struct bch_dev *,
 222                            enum bch_member_state, int);
 223 int __bch2_dev_set_state(struct bch_fs *, struct bch_dev *,
 224                         enum bch_member_state, int);
 225 int bch2_dev_set_state(struct bch_fs *, struct bch_dev *,
 226                       enum bch_member_state, int);
 227
 228 int bch2_dev_fail(struct bch_dev *, int);
 229 int bch2_dev_remove(struct bch_fs *, struct bch_dev *, int);
 230 int bch2_dev_add(struct bch_fs *, const char *);
 231 int bch2_dev_online(struct bch_fs *, const char *);
 232 int bch2_dev_offline(struct bch_fs *, struct bch_dev *, int);
 233 int bch2_dev_resize(struct bch_fs *, struct bch_dev *, u64);
 234 struct bch_dev *bch2_dev_lookup(struct bch_fs *, const char *);
 235
 236 bool bch2_fs_emergency_read_only(struct bch_fs *);
 237 void bch2_fs_read_only(struct bch_fs *);
 238
 239 int bch2_fs_read_write(struct bch_fs *);
 240 int bch2_fs_read_write_early(struct bch_fs *);
 241
 242 /*
 243  * Only for use in the recovery/fsck path:
 244  */
 245 static inline void bch2_fs_lazy_rw(struct bch_fs *c)
 246 {
 247         if (percpu_ref_is_zero(&c->writes))
 248                 bch2_fs_read_write_early(c);
 249 }
 250
 251 void __bch2_fs_stop(struct bch_fs *);
 252 void bch2_fs_free(struct bch_fs *);
 253 void bch2_fs_stop(struct bch_fs *);
 254
 255 int bch2_fs_start(struct bch_fs *);
 256 struct bch_fs *bch2_fs_open(char * const *, unsigned, struct bch_opts);
 257
 258 #endif /* _BCACHEFS_SUPER_H */