1// SPDX-License-Identifier: GPL-2.0
  2#include "bcachefs.h"
  3#include "super-io.h"
  4#include "sb-counters.h"
  5
  6/* BCH_SB_FIELD_counters */
  7
  8static const char * const bch2_counter_names[] = {
  9#define x(t, n, ...) (#t),
 10	BCH_PERSISTENT_COUNTERS()
 11#undef x
 12	NULL
 13};
 14
 15static size_t bch2_sb_counter_nr_entries(struct bch_sb_field_counters *ctrs)
 16{
 17	if (!ctrs)
 18		return 0;
 19
 20	return (__le64 *) vstruct_end(&ctrs->field) - &ctrs->d[0];
 21};
 22
 23static int bch2_sb_counters_validate(struct bch_sb *sb,
 24				     struct bch_sb_field *f,
 25				     struct printbuf *err)
 26{
 27	return 0;
 28};
 29
 30static void bch2_sb_counters_to_text(struct printbuf *out, struct bch_sb *sb,
 31			      struct bch_sb_field *f)
 32{
 33	struct bch_sb_field_counters *ctrs = field_to_type(f, counters);
 34	unsigned int i;
 35	unsigned int nr = bch2_sb_counter_nr_entries(ctrs);
 36
 37	for (i = 0; i < nr; i++) {
 38		if (i < BCH_COUNTER_NR)
 39			prt_printf(out, "%s ", bch2_counter_names[i]);
 40		else
 41			prt_printf(out, "(unknown)");
 42
 43		prt_tab(out);
 44		prt_printf(out, "%llu", le64_to_cpu(ctrs->d[i]));
 45		prt_newline(out);
 46	}
 47};
 48
 49int bch2_sb_counters_to_cpu(struct bch_fs *c)
 50{
 51	struct bch_sb_field_counters *ctrs = bch2_sb_field_get(c->disk_sb.sb, counters);
 52	unsigned int i;
 53	unsigned int nr = bch2_sb_counter_nr_entries(ctrs);
 54	u64 val = 0;
 55
 56	for (i = 0; i < BCH_COUNTER_NR; i++)
 57		c->counters_on_mount[i] = 0;
 58
 59	for (i = 0; i < min_t(unsigned int, nr, BCH_COUNTER_NR); i++) {
 60		val = le64_to_cpu(ctrs->d[i]);
 61		percpu_u64_set(&c->counters[i], val);
 62		c->counters_on_mount[i] = val;
 63	}
 64	return 0;
 65};
 66
 67int bch2_sb_counters_from_cpu(struct bch_fs *c)
 68{
 69	struct bch_sb_field_counters *ctrs = bch2_sb_field_get(c->disk_sb.sb, counters);
 70	struct bch_sb_field_counters *ret;
 71	unsigned int i;
 72	unsigned int nr = bch2_sb_counter_nr_entries(ctrs);
 73
 74	if (nr < BCH_COUNTER_NR) {
 75		ret = bch2_sb_field_resize(&c->disk_sb, counters,
 76					       sizeof(*ctrs) / sizeof(u64) + BCH_COUNTER_NR);
 77
 78		if (ret) {
 79			ctrs = ret;
 80			nr = bch2_sb_counter_nr_entries(ctrs);
 81		}
 82	}
 83
 84
 85	for (i = 0; i < min_t(unsigned int, nr, BCH_COUNTER_NR); i++)
 86		ctrs->d[i] = cpu_to_le64(percpu_u64_get(&c->counters[i]));
 87	return 0;
 88}
 89
 90void bch2_fs_counters_exit(struct bch_fs *c)
 91{
 92	free_percpu(c->counters);
 93}
 94
 95int bch2_fs_counters_init(struct bch_fs *c)
 96{
 97	c->counters = __alloc_percpu(sizeof(u64) * BCH_COUNTER_NR, sizeof(u64));
 98	if (!c->counters)
 99		return -BCH_ERR_ENOMEM_fs_counters_init;
100
101	return bch2_sb_counters_to_cpu(c);
102}
103
104const struct bch_sb_field_ops bch_sb_field_ops_counters = {
105	.validate	= bch2_sb_counters_validate,
106	.to_text	= bch2_sb_counters_to_text,
107};