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 unsigned nr_groups = disk_groups_nr(groups);
32 for (i = 0; i < sb->nr_devices; i++) {
33 struct bch_member m = bch2_sb_member_get(sb, i);
36 if (!BCH_MEMBER_GROUP(&m))
39 group_id = BCH_MEMBER_GROUP(&m) - 1;
41 if (group_id >= nr_groups) {
42 prt_printf(err, "disk %u has invalid label %u (have %u)",
43 i, group_id, nr_groups);
44 return -BCH_ERR_invalid_sb_disk_groups;
47 if (BCH_GROUP_DELETED(&groups->entries[group_id])) {
48 prt_printf(err, "disk %u has deleted label %u", i, group_id);
49 return -BCH_ERR_invalid_sb_disk_groups;
56 for (i = 0; i < nr_groups; i++) {
57 g = groups->entries + i;
59 if (BCH_GROUP_DELETED(g))
62 len = strnlen(g->label, sizeof(g->label));
64 prt_printf(err, "label %u empty", i);
65 return -BCH_ERR_invalid_sb_disk_groups;
69 sorted = kmalloc_array(nr_groups, sizeof(*sorted), GFP_KERNEL);
71 return -BCH_ERR_ENOMEM_disk_groups_validate;
73 memcpy(sorted, groups->entries, nr_groups * sizeof(*sorted));
74 sort(sorted, nr_groups, sizeof(*sorted), group_cmp, NULL);
76 for (g = sorted; g + 1 < sorted + nr_groups; g++)
77 if (!BCH_GROUP_DELETED(g) &&
78 !group_cmp(&g[0], &g[1])) {
79 prt_printf(err, "duplicate label %llu.%.*s",
81 (int) sizeof(g->label), g->label);
82 ret = -BCH_ERR_invalid_sb_disk_groups;
90 void bch2_disk_groups_to_text(struct printbuf *out, struct bch_fs *c)
95 struct bch_disk_groups_cpu *g = rcu_dereference(c->disk_groups);
99 for (unsigned i = 0; i < g->nr; i++) {
101 prt_printf(out, " ");
103 if (g->entries[i].deleted) {
104 prt_printf(out, "[deleted]");
108 prt_printf(out, "[parent %d devs", g->entries[i].parent);
109 for_each_member_device_rcu(c, ca, &g->entries[i].devs)
110 prt_printf(out, " %s", ca->name);
111 prt_printf(out, "]");
119 static void bch2_sb_disk_groups_to_text(struct printbuf *out,
121 struct bch_sb_field *f)
123 struct bch_sb_field_disk_groups *groups =
124 field_to_type(f, disk_groups);
125 struct bch_disk_group *g;
126 unsigned nr_groups = disk_groups_nr(groups);
128 for (g = groups->entries;
129 g < groups->entries + nr_groups;
131 if (g != groups->entries)
132 prt_printf(out, " ");
134 if (BCH_GROUP_DELETED(g))
135 prt_printf(out, "[deleted]");
137 prt_printf(out, "[parent %llu name %s]",
138 BCH_GROUP_PARENT(g), g->label);
142 const struct bch_sb_field_ops bch_sb_field_ops_disk_groups = {
143 .validate = bch2_sb_disk_groups_validate,
144 .to_text = bch2_sb_disk_groups_to_text
147 int bch2_sb_disk_groups_to_cpu(struct bch_fs *c)
149 struct bch_sb_field_disk_groups *groups;
150 struct bch_disk_groups_cpu *cpu_g, *old_g;
151 unsigned i, g, nr_groups;
153 lockdep_assert_held(&c->sb_lock);
155 groups = bch2_sb_field_get(c->disk_sb.sb, disk_groups);
156 nr_groups = disk_groups_nr(groups);
161 cpu_g = kzalloc(struct_size(cpu_g, entries, nr_groups), GFP_KERNEL);
163 return -BCH_ERR_ENOMEM_disk_groups_to_cpu;
165 cpu_g->nr = nr_groups;
167 for (i = 0; i < nr_groups; i++) {
168 struct bch_disk_group *src = &groups->entries[i];
169 struct bch_disk_group_cpu *dst = &cpu_g->entries[i];
171 dst->deleted = BCH_GROUP_DELETED(src);
172 dst->parent = BCH_GROUP_PARENT(src);
173 memcpy(dst->label, src->label, sizeof(dst->label));
176 for (i = 0; i < c->disk_sb.sb->nr_devices; i++) {
177 struct bch_member m = bch2_sb_member_get(c->disk_sb.sb, i);
178 struct bch_disk_group_cpu *dst;
180 if (!bch2_member_exists(&m))
183 g = BCH_MEMBER_GROUP(&m);
185 dst = &cpu_g->entries[g - 1];
186 __set_bit(i, dst->devs.d);
191 old_g = rcu_dereference_protected(c->disk_groups,
192 lockdep_is_held(&c->sb_lock));
193 rcu_assign_pointer(c->disk_groups, cpu_g);
195 kfree_rcu(old_g, rcu);
200 const struct bch_devs_mask *bch2_target_to_mask(struct bch_fs *c, unsigned target)
202 struct target t = target_decode(target);
203 struct bch_devs_mask *devs;
212 struct bch_dev *ca = t.dev < c->sb.nr_devices
213 ? rcu_dereference(c->devs[t.dev])
215 devs = ca ? &ca->self : NULL;
219 struct bch_disk_groups_cpu *g = rcu_dereference(c->disk_groups);
221 devs = g && t.group < g->nr && !g->entries[t.group].deleted
222 ? &g->entries[t.group].devs
235 bool bch2_dev_in_target(struct bch_fs *c, unsigned dev, unsigned target)
237 struct target t = target_decode(target);
245 struct bch_disk_groups_cpu *g;
246 const struct bch_devs_mask *m;
250 g = rcu_dereference(c->disk_groups);
251 m = g && t.group < g->nr && !g->entries[t.group].deleted
252 ? &g->entries[t.group].devs
255 ret = m ? test_bit(dev, m->d) : false;
265 static int __bch2_disk_group_find(struct bch_sb_field_disk_groups *groups,
267 const char *name, unsigned namelen)
269 unsigned i, nr_groups = disk_groups_nr(groups);
271 if (!namelen || namelen > BCH_SB_LABEL_SIZE)
274 for (i = 0; i < nr_groups; i++) {
275 struct bch_disk_group *g = groups->entries + i;
277 if (BCH_GROUP_DELETED(g))
280 if (!BCH_GROUP_DELETED(g) &&
281 BCH_GROUP_PARENT(g) == parent &&
282 strnlen(g->label, sizeof(g->label)) == namelen &&
283 !memcmp(name, g->label, namelen))
290 static int __bch2_disk_group_add(struct bch_sb_handle *sb, unsigned parent,
291 const char *name, unsigned namelen)
293 struct bch_sb_field_disk_groups *groups =
294 bch2_sb_field_get(sb->sb, disk_groups);
295 unsigned i, nr_groups = disk_groups_nr(groups);
296 struct bch_disk_group *g;
298 if (!namelen || namelen > BCH_SB_LABEL_SIZE)
302 i < nr_groups && !BCH_GROUP_DELETED(&groups->entries[i]);
306 if (i == nr_groups) {
308 (sizeof(struct bch_sb_field_disk_groups) +
309 sizeof(struct bch_disk_group) * (nr_groups + 1)) /
312 groups = bch2_sb_field_resize(sb, disk_groups, u64s);
314 return -BCH_ERR_ENOSPC_disk_label_add;
316 nr_groups = disk_groups_nr(groups);
319 BUG_ON(i >= nr_groups);
321 g = &groups->entries[i];
323 memcpy(g->label, name, namelen);
324 if (namelen < sizeof(g->label))
325 g->label[namelen] = '\0';
326 SET_BCH_GROUP_DELETED(g, 0);
327 SET_BCH_GROUP_PARENT(g, parent);
328 SET_BCH_GROUP_DATA_ALLOWED(g, ~0);
333 int bch2_disk_path_find(struct bch_sb_handle *sb, const char *name)
335 struct bch_sb_field_disk_groups *groups =
336 bch2_sb_field_get(sb->sb, disk_groups);
340 const char *next = strchrnul(name, '.');
341 unsigned len = next - name;
346 v = __bch2_disk_group_find(groups, v + 1, name, len);
348 } while (*name && v >= 0);
353 int bch2_disk_path_find_or_create(struct bch_sb_handle *sb, const char *name)
355 struct bch_sb_field_disk_groups *groups;
360 const char *next = strchrnul(name, '.');
361 unsigned len = next - name;
366 groups = bch2_sb_field_get(sb->sb, disk_groups);
368 v = __bch2_disk_group_find(groups, parent, name, len);
370 v = __bch2_disk_group_add(sb, parent, name, len);
376 } while (*name && v >= 0);
381 void bch2_disk_path_to_text(struct printbuf *out, struct bch_fs *c, unsigned v)
383 struct bch_disk_groups_cpu *groups;
384 struct bch_disk_group_cpu *g;
390 groups = rcu_dereference(c->disk_groups);
395 if (nr == ARRAY_SIZE(path))
401 g = groups->entries + v;
416 g = groups->entries + v;
418 prt_printf(out, "%.*s", (int) sizeof(g->label), g->label);
420 prt_printf(out, ".");
427 prt_printf(out, "invalid label %u", v);
431 void bch2_disk_path_to_text_sb(struct printbuf *out, struct bch_sb *sb, unsigned v)
433 struct bch_sb_field_disk_groups *groups =
434 bch2_sb_field_get(sb, disk_groups);
435 struct bch_disk_group *g;
440 if (nr == ARRAY_SIZE(path))
443 if (v >= disk_groups_nr(groups))
446 g = groups->entries + v;
448 if (BCH_GROUP_DELETED(g))
453 if (!BCH_GROUP_PARENT(g))
456 v = BCH_GROUP_PARENT(g) - 1;
461 g = groups->entries + v;
463 prt_printf(out, "%.*s", (int) sizeof(g->label), g->label);
465 prt_printf(out, ".");
469 prt_printf(out, "invalid label %u", v);
472 int __bch2_dev_group_set(struct bch_fs *c, struct bch_dev *ca, const char *name)
474 struct bch_member *mi;
477 if (!strlen(name) || !strcmp(name, "none"))
480 v = bch2_disk_path_find_or_create(&c->disk_sb, name);
484 ret = bch2_sb_disk_groups_to_cpu(c);
488 mi = bch2_members_v2_get_mut(c->disk_sb.sb, ca->dev_idx);
489 SET_BCH_MEMBER_GROUP(mi, v + 1);
493 int bch2_dev_group_set(struct bch_fs *c, struct bch_dev *ca, const char *name)
497 mutex_lock(&c->sb_lock);
498 ret = __bch2_dev_group_set(c, ca, name) ?:
500 mutex_unlock(&c->sb_lock);
505 int bch2_opt_target_parse(struct bch_fs *c, const char *val, u64 *res,
506 struct printbuf *err)
517 if (!strlen(val) || !strcmp(val, "none")) {
522 /* Is it a device? */
523 ca = bch2_dev_lookup(c, val);
525 *res = dev_to_target(ca->dev_idx);
526 percpu_ref_put(&ca->ref);
530 mutex_lock(&c->sb_lock);
531 g = bch2_disk_path_find(&c->disk_sb, val);
532 mutex_unlock(&c->sb_lock);
535 *res = group_to_target(g);
542 void bch2_target_to_text(struct printbuf *out, struct bch_fs *c, unsigned v)
544 struct target t = target_decode(v);
548 prt_printf(out, "none");
555 ca = t.dev < c->sb.nr_devices
556 ? rcu_dereference(c->devs[t.dev])
559 if (ca && percpu_ref_tryget(&ca->io_ref)) {
560 prt_printf(out, "/dev/%s", ca->name);
561 percpu_ref_put(&ca->io_ref);
563 prt_printf(out, "offline device %u", t.dev);
565 prt_printf(out, "invalid device %u", t.dev);
573 bch2_disk_path_to_text(out, c, t.group);
580 static void bch2_target_to_text_sb(struct printbuf *out, struct bch_sb *sb, unsigned v)
582 struct target t = target_decode(v);
586 prt_printf(out, "none");
589 struct bch_member m = bch2_sb_member_get(sb, t.dev);
591 if (bch2_dev_exists(sb, t.dev)) {
592 prt_printf(out, "Device ");
593 pr_uuid(out, m.uuid.b);
594 prt_printf(out, " (%u)", t.dev);
596 prt_printf(out, "Bad device %u", t.dev);
601 bch2_disk_path_to_text_sb(out, sb, t.group);
608 void bch2_opt_target_to_text(struct printbuf *out,
614 bch2_target_to_text(out, c, v);
616 bch2_target_to_text_sb(out, sb, v);