mirror of
https://git.kernel.org/pub/scm/linux/kernel/git/torvalds/linux.git
synced 2026-08-31 07:03:28 -04:00
Replace hardcoded enum values with bpf_core_enum_value() calls in cgroup_iter_memcg test to improve portability across different kernel versions. The change adds runtime enum value resolution for: - node_stat_item: NR_ANON_MAPPED, NR_SHMEM, NR_FILE_PAGES, NR_FILE_MAPPED - vm_event_item: PGFAULT This ensures the BPF program can adapt to enum value changes between kernel versions. Reviewed-by: Emil Tsalapatis <emil@etsalapatis.com> Reviewed-by: JP Kobryn <jp.kobryn@linux.dev> Signed-off-by: Hui Zhu <zhuhui@kylinos.cn> Link: https://lore.kernel.org/r/ca6eb1a1a4fd7a17ffe995acf52c9a4ceb7bac13.1772505399.git.zhuhui@kylinos.cn Signed-off-by: Alexei Starovoitov <ast@kernel.org>
46 lines
1.2 KiB
C
46 lines
1.2 KiB
C
// SPDX-License-Identifier: GPL-2.0
|
|
/* Copyright (c) 2025 Meta Platforms, Inc. and affiliates. */
|
|
#include <vmlinux.h>
|
|
#include <bpf/bpf_core_read.h>
|
|
#include "cgroup_iter_memcg.h"
|
|
|
|
char _license[] SEC("license") = "GPL";
|
|
|
|
/* The latest values read are stored here. */
|
|
struct memcg_query memcg_query SEC(".data.query");
|
|
|
|
SEC("iter.s/cgroup")
|
|
int cgroup_memcg_query(struct bpf_iter__cgroup *ctx)
|
|
{
|
|
struct cgroup *cgrp = ctx->cgroup;
|
|
struct cgroup_subsys_state *css;
|
|
struct mem_cgroup *memcg;
|
|
|
|
if (!cgrp)
|
|
return 1;
|
|
|
|
css = &cgrp->self;
|
|
memcg = bpf_get_mem_cgroup(css);
|
|
if (!memcg)
|
|
return 1;
|
|
|
|
bpf_mem_cgroup_flush_stats(memcg);
|
|
|
|
memcg_query.nr_anon_mapped = bpf_mem_cgroup_page_state(
|
|
memcg,
|
|
bpf_core_enum_value(enum node_stat_item, NR_ANON_MAPPED));
|
|
memcg_query.nr_shmem = bpf_mem_cgroup_page_state(
|
|
memcg, bpf_core_enum_value(enum node_stat_item, NR_SHMEM));
|
|
memcg_query.nr_file_pages = bpf_mem_cgroup_page_state(
|
|
memcg, bpf_core_enum_value(enum node_stat_item, NR_FILE_PAGES));
|
|
memcg_query.nr_file_mapped = bpf_mem_cgroup_page_state(
|
|
memcg,
|
|
bpf_core_enum_value(enum node_stat_item, NR_FILE_MAPPED));
|
|
memcg_query.pgfault = bpf_mem_cgroup_vm_events(
|
|
memcg, bpf_core_enum_value(enum vm_event_item, PGFAULT));
|
|
|
|
bpf_put_mem_cgroup(memcg);
|
|
|
|
return 0;
|
|
}
|