/* SPDX-License-Identifier: GPL-2.0-only */ /* * Tools for integrating with lru_gen, like parsing the lru_gen debugfs output. * * Copyright (C) 2025, Google LLC. */ #ifndef SELFTEST_KVM_LRU_GEN_UTIL_H #define SELFTEST_KVM_LRU_GEN_UTIL_H #include <inttypes.h> #include <limits.h> #include <stdlib.h> #include "test_util.h" #define MAX_NR_GENS 16 /* MAX_NR_GENS in include/linux/mmzone.h */ #define MAX_NR_NODES 4 /* Maximum number of nodes supported by the test */ #define LRU_GEN_DEBUGFS "/sys/kernel/debug/lru_gen" #define LRU_GEN_ENABLED_PATH "/sys/kernel/mm/lru_gen/enabled" #define LRU_GEN_ENABLED 1 #define LRU_GEN_MM_WALK 2 struct generation_stats { int gen; long age_ms; long nr_anon; long nr_file; }; struct node_stats { int node; int nr_gens; /* Number of populated gens entries. */ struct generation_stats gens[MAX_NR_GENS]; }; struct memcg_stats { unsigned long memcg_id; int nr_nodes; /* Number of populated nodes entries. */ struct node_stats nodes[MAX_NR_NODES]; }; void lru_gen_read_memcg_stats(struct memcg_stats *stats, const char *memcg); long lru_gen_sum_memcg_stats(const struct memcg_stats *stats); long lru_gen_sum_memcg_stats_for_gen(int gen, const struct memcg_stats *stats); void lru_gen_do_aging(struct memcg_stats *stats, const char *memcg); int lru_gen_find_generation(const struct memcg_stats *stats, unsigned long total_pages); bool lru_gen_usable(void); #endif /* SELFTEST_KVM_LRU_GEN_UTIL_H */