1 // SPDX-License-Identifier: GPL-2.0
3 #include "disk_groups.h"
6 #include <linux/sort.h>
8 static int group_cmp(const void *_l, const void *_r)
10 const struct bch_disk_group *l = _l;
11 const struct bch_disk_group *r = _r;
13 return ((BCH_GROUP_DELETED(l) > BCH_GROUP_DELETED(r)) -
14 (BCH_GROUP_DELETED(l) < BCH_GROUP_DELETED(r))) ?:
15 ((BCH_GROUP_PARENT(l) > BCH_GROUP_PARENT(r)) -
16 (BCH_GROUP_PARENT(l) < BCH_GROUP_PARENT(r))) ?:
17 strncmp(l->label, r->label, sizeof(l->label));
20 static int bch2_sb_disk_groups_validate(struct bch_sb *sb,
21 struct bch_sb_field *f,
24 struct bch_sb_field_disk_groups *groups =
25 field_to_type(f, disk_groups);
26 struct bch_disk_group *g, *sorted = NULL;
27 struct bch_sb_field_members *mi = bch2_sb_get_members(sb);
28 unsigned nr_groups = disk_groups_nr(groups);
32 for (i = 0; i < sb->nr_devices; i++) {
33 struct bch_member *m = mi->members + i;
36 if (!BCH_MEMBER_GROUP(m))
39 g = BCH_MEMBER_GROUP(m) - 1;
42 pr_buf(err, "disk %u has invalid label %u (have %u)",
47 if (BCH_GROUP_DELETED(&groups->entries[g])) {
48 pr_buf(err, "disk %u has deleted label %u", i, g);
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 pr_buf(err, "label %u empty", i);
69 sorted = kmalloc_array(nr_groups, sizeof(*sorted), GFP_KERNEL);
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 pr_buf(err, "duplicate label %llu.%.*s",
81 (int) sizeof(g->label), g->label);
91 static void bch2_sb_disk_groups_to_text(struct printbuf *out,
93 struct bch_sb_field *f)
95 struct bch_sb_field_disk_groups *groups =
96 field_to_type(f, disk_groups);
97 struct bch_disk_group *g;
98 unsigned nr_groups = disk_groups_nr(groups);
100 for (g = groups->entries;
101 g < groups->entries + nr_groups;
103 if (g != groups->entries)
106 if (BCH_GROUP_DELETED(g))
107 pr_buf(out, "[deleted]");
109 pr_buf(out, "[parent %llu name %s]",
110 BCH_GROUP_PARENT(g), g->label);
114 const struct bch_sb_field_ops bch_sb_field_ops_disk_groups = {
115 .validate = bch2_sb_disk_groups_validate,
116 .to_text = bch2_sb_disk_groups_to_text
119 int bch2_sb_disk_groups_to_cpu(struct bch_fs *c)
121 struct bch_sb_field_members *mi;
122 struct bch_sb_field_disk_groups *groups;
123 struct bch_disk_groups_cpu *cpu_g, *old_g;
124 unsigned i, g, nr_groups;
126 lockdep_assert_held(&c->sb_lock);
128 mi = bch2_sb_get_members(c->disk_sb.sb);
129 groups = bch2_sb_get_disk_groups(c->disk_sb.sb);
130 nr_groups = disk_groups_nr(groups);
135 cpu_g = kzalloc(sizeof(*cpu_g) +
136 sizeof(cpu_g->entries[0]) * nr_groups, GFP_KERNEL);
140 cpu_g->nr = nr_groups;
142 for (i = 0; i < nr_groups; i++) {
143 struct bch_disk_group *src = &groups->entries[i];
144 struct bch_disk_group_cpu *dst = &cpu_g->entries[i];
146 dst->deleted = BCH_GROUP_DELETED(src);
147 dst->parent = BCH_GROUP_PARENT(src);
150 for (i = 0; i < c->disk_sb.sb->nr_devices; i++) {
151 struct bch_member *m = mi->members + i;
152 struct bch_disk_group_cpu *dst =
153 &cpu_g->entries[BCH_MEMBER_GROUP(m)];
155 if (!bch2_member_exists(m))
158 g = BCH_MEMBER_GROUP(m);
160 dst = &cpu_g->entries[g - 1];
161 __set_bit(i, dst->devs.d);
166 old_g = rcu_dereference_protected(c->disk_groups,
167 lockdep_is_held(&c->sb_lock));
168 rcu_assign_pointer(c->disk_groups, cpu_g);
170 kfree_rcu(old_g, rcu);
175 const struct bch_devs_mask *bch2_target_to_mask(struct bch_fs *c, unsigned target)
177 struct target t = target_decode(target);
183 struct bch_dev *ca = t.dev < c->sb.nr_devices
184 ? rcu_dereference(c->devs[t.dev])
186 return ca ? &ca->self : NULL;
189 struct bch_disk_groups_cpu *g = rcu_dereference(c->disk_groups);
191 return g && t.group < g->nr && !g->entries[t.group].deleted
192 ? &g->entries[t.group].devs
200 bool bch2_dev_in_target(struct bch_fs *c, unsigned dev, unsigned target)
202 struct target t = target_decode(target);
210 struct bch_disk_groups_cpu *g;
211 const struct bch_devs_mask *m;
215 g = rcu_dereference(c->disk_groups);
216 m = g && t.group < g->nr && !g->entries[t.group].deleted
217 ? &g->entries[t.group].devs
220 ret = m ? test_bit(dev, m->d) : false;
230 static int __bch2_disk_group_find(struct bch_sb_field_disk_groups *groups,
232 const char *name, unsigned namelen)
234 unsigned i, nr_groups = disk_groups_nr(groups);
236 if (!namelen || namelen > BCH_SB_LABEL_SIZE)
239 for (i = 0; i < nr_groups; i++) {
240 struct bch_disk_group *g = groups->entries + i;
242 if (BCH_GROUP_DELETED(g))
245 if (!BCH_GROUP_DELETED(g) &&
246 BCH_GROUP_PARENT(g) == parent &&
247 strnlen(g->label, sizeof(g->label)) == namelen &&
248 !memcmp(name, g->label, namelen))
255 static int __bch2_disk_group_add(struct bch_sb_handle *sb, unsigned parent,
256 const char *name, unsigned namelen)
258 struct bch_sb_field_disk_groups *groups =
259 bch2_sb_get_disk_groups(sb->sb);
260 unsigned i, nr_groups = disk_groups_nr(groups);
261 struct bch_disk_group *g;
263 if (!namelen || namelen > BCH_SB_LABEL_SIZE)
267 i < nr_groups && !BCH_GROUP_DELETED(&groups->entries[i]);
271 if (i == nr_groups) {
273 (sizeof(struct bch_sb_field_disk_groups) +
274 sizeof(struct bch_disk_group) * (nr_groups + 1)) /
277 groups = bch2_sb_resize_disk_groups(sb, u64s);
281 nr_groups = disk_groups_nr(groups);
284 BUG_ON(i >= nr_groups);
286 g = &groups->entries[i];
288 memcpy(g->label, name, namelen);
289 if (namelen < sizeof(g->label))
290 g->label[namelen] = '\0';
291 SET_BCH_GROUP_DELETED(g, 0);
292 SET_BCH_GROUP_PARENT(g, parent);
293 SET_BCH_GROUP_DATA_ALLOWED(g, ~0);
298 int bch2_disk_path_find(struct bch_sb_handle *sb, const char *name)
300 struct bch_sb_field_disk_groups *groups =
301 bch2_sb_get_disk_groups(sb->sb);
305 const char *next = strchrnul(name, '.');
306 unsigned len = next - name;
311 v = __bch2_disk_group_find(groups, v + 1, name, len);
313 } while (*name && v >= 0);
318 int bch2_disk_path_find_or_create(struct bch_sb_handle *sb, const char *name)
320 struct bch_sb_field_disk_groups *groups;
325 const char *next = strchrnul(name, '.');
326 unsigned len = next - name;
331 groups = bch2_sb_get_disk_groups(sb->sb);
333 v = __bch2_disk_group_find(groups, parent, name, len);
335 v = __bch2_disk_group_add(sb, parent, name, len);
341 } while (*name && v >= 0);
346 void bch2_disk_path_to_text(struct printbuf *out, struct bch_sb *sb, unsigned v)
348 struct bch_sb_field_disk_groups *groups =
349 bch2_sb_get_disk_groups(sb);
350 struct bch_disk_group *g;
355 if (nr == ARRAY_SIZE(path))
358 if (v >= disk_groups_nr(groups))
361 g = groups->entries + v;
363 if (BCH_GROUP_DELETED(g))
368 if (!BCH_GROUP_PARENT(g))
371 v = BCH_GROUP_PARENT(g) - 1;
376 g = groups->entries + v;
378 pr_buf(out, "%.*s", (int) sizeof(g->label), g->label);
384 pr_buf(out, "invalid label %u", v);
387 int bch2_dev_group_set(struct bch_fs *c, struct bch_dev *ca, const char *name)
389 struct bch_member *mi;
393 mutex_lock(&c->sb_lock);
395 if (!strlen(name) || !strcmp(name, "none"))
398 v = bch2_disk_path_find_or_create(&c->disk_sb, name);
400 mutex_unlock(&c->sb_lock);
404 ret = bch2_sb_disk_groups_to_cpu(c);
408 mi = &bch2_sb_get_members(c->disk_sb.sb)->members[ca->dev_idx];
409 SET_BCH_MEMBER_GROUP(mi, v + 1);
413 mutex_unlock(&c->sb_lock);
418 int bch2_opt_target_parse(struct bch_fs *c, const char *buf, u64 *v)
423 if (!strlen(buf) || !strcmp(buf, "none")) {
428 /* Is it a device? */
429 ca = bch2_dev_lookup(c, buf);
431 *v = dev_to_target(ca->dev_idx);
432 percpu_ref_put(&ca->ref);
436 mutex_lock(&c->sb_lock);
437 g = bch2_disk_path_find(&c->disk_sb, buf);
438 mutex_unlock(&c->sb_lock);
441 *v = group_to_target(g);
448 void bch2_opt_target_to_text(struct printbuf *out,
453 struct target t = target_decode(v);
464 ca = t.dev < c->sb.nr_devices
465 ? rcu_dereference(c->devs[t.dev])
468 if (ca && percpu_ref_tryget(&ca->io_ref)) {
469 char b[BDEVNAME_SIZE];
471 pr_buf(out, "/dev/%s",
472 bdevname(ca->disk_sb.bdev, b));
473 percpu_ref_put(&ca->io_ref);
475 pr_buf(out, "offline device %u", t.dev);
477 pr_buf(out, "invalid device %u", t.dev);
482 struct bch_sb_field_members *mi = bch2_sb_get_members(sb);
483 struct bch_member *m = mi->members + t.dev;
485 if (bch2_dev_exists(sb, mi, t.dev)) {
486 pr_buf(out, "Device ");
487 pr_uuid(out, m->uuid.b);
488 pr_buf(out, " (%u)", t.dev);
490 pr_buf(out, "Bad device %u", t.dev);
496 mutex_lock(&c->sb_lock);
497 bch2_disk_path_to_text(out, c->disk_sb.sb, t.group);
498 mutex_unlock(&c->sb_lock);
500 bch2_disk_path_to_text(out, sb, t.group);