1/* SPDX-License-Identifier: GPL-2.0 */
  2#ifndef _BCACHEFS_DISK_GROUPS_H
  3#define _BCACHEFS_DISK_GROUPS_H
  4
  5#include "disk_groups_types.h"
  6
  7extern const struct bch_sb_field_ops bch_sb_field_ops_disk_groups;
  8
  9static inline unsigned disk_groups_nr(struct bch_sb_field_disk_groups *groups)
 10{
 11	return groups
 12		? (vstruct_end(&groups->field) -
 13		   (void *) &groups->entries[0]) / sizeof(struct bch_disk_group)
 14		: 0;
 15}
 16
 17struct target {
 18	enum {
 19		TARGET_NULL,
 20		TARGET_DEV,
 21		TARGET_GROUP,
 22	}			type;
 23	union {
 24		unsigned	dev;
 25		unsigned	group;
 26	};
 27};
 28
 29#define TARGET_DEV_START	1
 30#define TARGET_GROUP_START	(256 + TARGET_DEV_START)
 31
 32static inline u16 dev_to_target(unsigned dev)
 33{
 34	return TARGET_DEV_START + dev;
 35}
 36
 37static inline u16 group_to_target(unsigned group)
 38{
 39	return TARGET_GROUP_START + group;
 40}
 41
 42static inline struct target target_decode(unsigned target)
 43{
 44	if (target >= TARGET_GROUP_START)
 45		return (struct target) {
 46			.type	= TARGET_GROUP,
 47			.group	= target - TARGET_GROUP_START
 48		};
 49
 50	if (target >= TARGET_DEV_START)
 51		return (struct target) {
 52			.type	= TARGET_DEV,
 53			.group	= target - TARGET_DEV_START
 54		};
 55
 56	return (struct target) { .type = TARGET_NULL };
 57}
 58
 59const struct bch_devs_mask *bch2_target_to_mask(struct bch_fs *, unsigned);
 60
 61static inline struct bch_devs_mask target_rw_devs(struct bch_fs *c,
 62						  enum bch_data_type data_type,
 63						  u16 target)
 64{
 65	struct bch_devs_mask devs = c->rw_devs[data_type];
 66	const struct bch_devs_mask *t = bch2_target_to_mask(c, target);
 67
 68	if (t)
 69		bitmap_and(devs.d, devs.d, t->d, BCH_SB_MEMBERS_MAX);
 70	return devs;
 71}
 72
 73static inline bool bch2_target_accepts_data(struct bch_fs *c,
 74					    enum bch_data_type data_type,
 75					    u16 target)
 76{
 77	struct bch_devs_mask rw_devs = target_rw_devs(c, data_type, target);
 78	return !bitmap_empty(rw_devs.d, BCH_SB_MEMBERS_MAX);
 79}
 80
 81bool bch2_dev_in_target(struct bch_fs *, unsigned, unsigned);
 82
 83int bch2_disk_path_find(struct bch_sb_handle *, const char *);
 84
 85/* Exported for userspace bcachefs-tools: */
 86int bch2_disk_path_find_or_create(struct bch_sb_handle *, const char *);
 87
 88void bch2_disk_path_to_text(struct printbuf *, struct bch_fs *, unsigned);
 89void bch2_disk_path_to_text_sb(struct printbuf *, struct bch_sb *, unsigned);
 90
 91void bch2_target_to_text(struct printbuf *out, struct bch_fs *, unsigned);
 92
 93int bch2_opt_target_parse(struct bch_fs *, const char *, u64 *, struct printbuf *);
 94void bch2_opt_target_to_text(struct printbuf *, struct bch_fs *, struct bch_sb *, u64);
 95
 96#define bch2_opt_target (struct bch_opt_fn) {		\
 97	.parse		= bch2_opt_target_parse,	\
 98	.to_text	= bch2_opt_target_to_text,	\
 99}
100
101int bch2_sb_disk_groups_to_cpu(struct bch_fs *);
102
103int __bch2_dev_group_set(struct bch_fs *, struct bch_dev *, const char *);
104int bch2_dev_group_set(struct bch_fs *, struct bch_dev *, const char *);
105
106const char *bch2_sb_validate_disk_groups(struct bch_sb *,
107					 struct bch_sb_field *);
108
109void bch2_disk_groups_to_text(struct printbuf *, struct bch_fs *);
110
111#endif /* _BCACHEFS_DISK_GROUPS_H */