1 // SPDX-License-Identifier: GPL-2.0
3 #include "disk_groups.h"
4 #include "sb-members.h"
7 #include <linux/sort.h>
9 static int group_cmp(const void *_l, const void *_r)
11 const struct bch_disk_group *l = _l;
12 const struct bch_disk_group *r = _r;
14 return ((BCH_GROUP_DELETED(l) > BCH_GROUP_DELETED(r)) -
15 (BCH_GROUP_DELETED(l) < BCH_GROUP_DELETED(r))) ?:
16 ((BCH_GROUP_PARENT(l) > BCH_GROUP_PARENT(r)) -
17 (BCH_GROUP_PARENT(l) < BCH_GROUP_PARENT(r))) ?:
18 strncmp(l->label, r->label, sizeof(l->label));
21 static int bch2_sb_disk_groups_validate(struct bch_sb *sb,
22 struct bch_sb_field *f,
25 struct bch_sb_field_disk_groups *groups =
26 field_to_type(f, disk_groups);
27 struct bch_disk_group *g, *sorted = NULL;
28 struct bch_sb_field_members *mi = bch2_sb_get_members(sb);
29 unsigned nr_groups = disk_groups_nr(groups);
33 for (i = 0; i < sb->nr_devices; i++) {
34 struct bch_member *m = mi->members + i;
37 if (!BCH_MEMBER_GROUP(m))
40 g = BCH_MEMBER_GROUP(m) - 1;
43 prt_printf(err, "disk %u has invalid label %u (have %u)",
45 return -BCH_ERR_invalid_sb_disk_groups;
48 if (BCH_GROUP_DELETED(&groups->entries[g])) {
49 prt_printf(err, "disk %u has deleted label %u", i, g);
50 return -BCH_ERR_invalid_sb_disk_groups;
57 for (i = 0; i < nr_groups; i++) {
58 g = groups->entries + i;
60 if (BCH_GROUP_DELETED(g))
63 len = strnlen(g->label, sizeof(g->label));
65 prt_printf(err, "label %u empty", i);
66 return -BCH_ERR_invalid_sb_disk_groups;
70 sorted = kmalloc_array(nr_groups, sizeof(*sorted), GFP_KERNEL);
72 return -BCH_ERR_ENOMEM_disk_groups_validate;
74 memcpy(sorted, groups->entries, nr_groups * sizeof(*sorted));
75 sort(sorted, nr_groups, sizeof(*sorted), group_cmp, NULL);
77 for (g = sorted; g + 1 < sorted + nr_groups; g++)
78 if (!BCH_GROUP_DELETED(g) &&
79 !group_cmp(&g[0], &g[1])) {
80 prt_printf(err, "duplicate label %llu.%.*s",
82 (int) sizeof(g->label), g->label);
83 ret = -BCH_ERR_invalid_sb_disk_groups;
91 void bch2_disk_groups_to_text(struct printbuf *out, struct bch_fs *c)
93 struct bch_disk_groups_cpu *g;
101 g = rcu_dereference(c->disk_groups);
105 for (i = 0; i < g->nr; i++) {
107 prt_printf(out, " ");
109 if (g->entries[i].deleted) {
110 prt_printf(out, "[deleted]");
114 prt_printf(out, "[parent %d devs", g->entries[i].parent);
115 for_each_member_device_rcu(ca, c, iter, &g->entries[i].devs)
116 prt_printf(out, " %s", ca->name);
117 prt_printf(out, "]");
125 static void bch2_sb_disk_groups_to_text(struct printbuf *out,
127 struct bch_sb_field *f)
129 struct bch_sb_field_disk_groups *groups =
130 field_to_type(f, disk_groups);
131 struct bch_disk_group *g;
132 unsigned nr_groups = disk_groups_nr(groups);
134 for (g = groups->entries;
135 g < groups->entries + nr_groups;
137 if (g != groups->entries)
138 prt_printf(out, " ");
140 if (BCH_GROUP_DELETED(g))
141 prt_printf(out, "[deleted]");
143 prt_printf(out, "[parent %llu name %s]",
144 BCH_GROUP_PARENT(g), g->label);
148 const struct bch_sb_field_ops bch_sb_field_ops_disk_groups = {
149 .validate = bch2_sb_disk_groups_validate,
150 .to_text = bch2_sb_disk_groups_to_text
153 int bch2_sb_disk_groups_to_cpu(struct bch_fs *c)
155 struct bch_sb_field_members *mi;
156 struct bch_sb_field_disk_groups *groups;
157 struct bch_disk_groups_cpu *cpu_g, *old_g;
158 unsigned i, g, nr_groups;
160 lockdep_assert_held(&c->sb_lock);
162 mi = bch2_sb_get_members(c->disk_sb.sb);
163 groups = bch2_sb_get_disk_groups(c->disk_sb.sb);
164 nr_groups = disk_groups_nr(groups);
169 cpu_g = kzalloc(sizeof(*cpu_g) +
170 sizeof(cpu_g->entries[0]) * nr_groups, GFP_KERNEL);
172 return -BCH_ERR_ENOMEM_disk_groups_to_cpu;
174 cpu_g->nr = nr_groups;
176 for (i = 0; i < nr_groups; i++) {
177 struct bch_disk_group *src = &groups->entries[i];
178 struct bch_disk_group_cpu *dst = &cpu_g->entries[i];
180 dst->deleted = BCH_GROUP_DELETED(src);
181 dst->parent = BCH_GROUP_PARENT(src);
184 for (i = 0; i < c->disk_sb.sb->nr_devices; i++) {
185 struct bch_member *m = mi->members + i;
186 struct bch_disk_group_cpu *dst =
187 &cpu_g->entries[BCH_MEMBER_GROUP(m)];
189 if (!bch2_member_exists(m))
192 g = BCH_MEMBER_GROUP(m);
194 dst = &cpu_g->entries[g - 1];
195 __set_bit(i, dst->devs.d);
200 old_g = rcu_dereference_protected(c->disk_groups,
201 lockdep_is_held(&c->sb_lock));
202 rcu_assign_pointer(c->disk_groups, cpu_g);
204 kfree_rcu(old_g, rcu);
209 const struct bch_devs_mask *bch2_target_to_mask(struct bch_fs *c, unsigned target)
211 struct target t = target_decode(target);
212 struct bch_devs_mask *devs;
221 struct bch_dev *ca = t.dev < c->sb.nr_devices
222 ? rcu_dereference(c->devs[t.dev])
224 devs = ca ? &ca->self : NULL;
228 struct bch_disk_groups_cpu *g = rcu_dereference(c->disk_groups);
230 devs = g && t.group < g->nr && !g->entries[t.group].deleted
231 ? &g->entries[t.group].devs
244 bool bch2_dev_in_target(struct bch_fs *c, unsigned dev, unsigned target)
246 struct target t = target_decode(target);
254 struct bch_disk_groups_cpu *g;
255 const struct bch_devs_mask *m;
259 g = rcu_dereference(c->disk_groups);
260 m = g && t.group < g->nr && !g->entries[t.group].deleted
261 ? &g->entries[t.group].devs
264 ret = m ? test_bit(dev, m->d) : false;
274 static int __bch2_disk_group_find(struct bch_sb_field_disk_groups *groups,
276 const char *name, unsigned namelen)
278 unsigned i, nr_groups = disk_groups_nr(groups);
280 if (!namelen || namelen > BCH_SB_LABEL_SIZE)
283 for (i = 0; i < nr_groups; i++) {
284 struct bch_disk_group *g = groups->entries + i;
286 if (BCH_GROUP_DELETED(g))
289 if (!BCH_GROUP_DELETED(g) &&
290 BCH_GROUP_PARENT(g) == parent &&
291 strnlen(g->label, sizeof(g->label)) == namelen &&
292 !memcmp(name, g->label, namelen))
299 static int __bch2_disk_group_add(struct bch_sb_handle *sb, unsigned parent,
300 const char *name, unsigned namelen)
302 struct bch_sb_field_disk_groups *groups =
303 bch2_sb_get_disk_groups(sb->sb);
304 unsigned i, nr_groups = disk_groups_nr(groups);
305 struct bch_disk_group *g;
307 if (!namelen || namelen > BCH_SB_LABEL_SIZE)
311 i < nr_groups && !BCH_GROUP_DELETED(&groups->entries[i]);
315 if (i == nr_groups) {
317 (sizeof(struct bch_sb_field_disk_groups) +
318 sizeof(struct bch_disk_group) * (nr_groups + 1)) /
321 groups = bch2_sb_resize_disk_groups(sb, u64s);
323 return -BCH_ERR_ENOSPC_disk_label_add;
325 nr_groups = disk_groups_nr(groups);
328 BUG_ON(i >= nr_groups);
330 g = &groups->entries[i];
332 memcpy(g->label, name, namelen);
333 if (namelen < sizeof(g->label))
334 g->label[namelen] = '\0';
335 SET_BCH_GROUP_DELETED(g, 0);
336 SET_BCH_GROUP_PARENT(g, parent);
337 SET_BCH_GROUP_DATA_ALLOWED(g, ~0);
342 int bch2_disk_path_find(struct bch_sb_handle *sb, const char *name)
344 struct bch_sb_field_disk_groups *groups =
345 bch2_sb_get_disk_groups(sb->sb);
349 const char *next = strchrnul(name, '.');
350 unsigned len = next - name;
355 v = __bch2_disk_group_find(groups, v + 1, name, len);
357 } while (*name && v >= 0);
362 int bch2_disk_path_find_or_create(struct bch_sb_handle *sb, const char *name)
364 struct bch_sb_field_disk_groups *groups;
369 const char *next = strchrnul(name, '.');
370 unsigned len = next - name;
375 groups = bch2_sb_get_disk_groups(sb->sb);
377 v = __bch2_disk_group_find(groups, parent, name, len);
379 v = __bch2_disk_group_add(sb, parent, name, len);
385 } while (*name && v >= 0);
390 void bch2_disk_path_to_text(struct printbuf *out, struct bch_sb *sb, unsigned v)
392 struct bch_sb_field_disk_groups *groups =
393 bch2_sb_get_disk_groups(sb);
394 struct bch_disk_group *g;
399 if (nr == ARRAY_SIZE(path))
402 if (v >= disk_groups_nr(groups))
405 g = groups->entries + v;
407 if (BCH_GROUP_DELETED(g))
412 if (!BCH_GROUP_PARENT(g))
415 v = BCH_GROUP_PARENT(g) - 1;
420 g = groups->entries + v;
422 prt_printf(out, "%.*s", (int) sizeof(g->label), g->label);
424 prt_printf(out, ".");
428 prt_printf(out, "invalid label %u", v);
431 int __bch2_dev_group_set(struct bch_fs *c, struct bch_dev *ca, const char *name)
433 struct bch_member *mi;
436 if (!strlen(name) || !strcmp(name, "none"))
439 v = bch2_disk_path_find_or_create(&c->disk_sb, name);
443 ret = bch2_sb_disk_groups_to_cpu(c);
447 mi = &bch2_sb_get_members(c->disk_sb.sb)->members[ca->dev_idx];
448 SET_BCH_MEMBER_GROUP(mi, v + 1);
452 int bch2_dev_group_set(struct bch_fs *c, struct bch_dev *ca, const char *name)
456 mutex_lock(&c->sb_lock);
457 ret = __bch2_dev_group_set(c, ca, name) ?:
459 mutex_unlock(&c->sb_lock);
464 int bch2_opt_target_parse(struct bch_fs *c, const char *val, u64 *res,
465 struct printbuf *err)
476 if (!strlen(val) || !strcmp(val, "none")) {
481 /* Is it a device? */
482 ca = bch2_dev_lookup(c, val);
484 *res = dev_to_target(ca->dev_idx);
485 percpu_ref_put(&ca->ref);
489 mutex_lock(&c->sb_lock);
490 g = bch2_disk_path_find(&c->disk_sb, val);
491 mutex_unlock(&c->sb_lock);
494 *res = group_to_target(g);
501 void bch2_opt_target_to_text(struct printbuf *out,
506 struct target t = target_decode(v);
510 prt_printf(out, "none");
517 ca = t.dev < c->sb.nr_devices
518 ? rcu_dereference(c->devs[t.dev])
521 if (ca && percpu_ref_tryget(&ca->io_ref)) {
522 prt_printf(out, "/dev/%pg", ca->disk_sb.bdev);
523 percpu_ref_put(&ca->io_ref);
525 prt_printf(out, "offline device %u", t.dev);
527 prt_printf(out, "invalid device %u", t.dev);
532 struct bch_sb_field_members *mi = bch2_sb_get_members(sb);
533 struct bch_member *m = mi->members + t.dev;
535 if (bch2_dev_exists(sb, mi, t.dev)) {
536 prt_printf(out, "Device ");
537 pr_uuid(out, m->uuid.b);
538 prt_printf(out, " (%u)", t.dev);
540 prt_printf(out, "Bad device %u", t.dev);
546 mutex_lock(&c->sb_lock);
547 bch2_disk_path_to_text(out, c->disk_sb.sb, t.group);
548 mutex_unlock(&c->sb_lock);
550 bch2_disk_path_to_text(out, sb, t.group);