[<prev] [next>] [<thread-prev] [day] [month] [year] [list]
Message-ID: <9483528f-83dd-4a30-9489-cf0fac4de5f7@linux.dev>
Date: Thu, 29 Jan 2026 13:00:11 -0800
From: Martin KaFai Lau <martin.lau@...ux.dev>
To: Roman Gushchin <roman.gushchin@...ux.dev>
Cc: Michal Hocko <mhocko@...e.com>, Alexei Starovoitov <ast@...nel.org>,
Matt Bobrowski <mattbobrowski@...gle.com>,
Shakeel Butt <shakeel.butt@...ux.dev>, JP Kobryn <inwardvessel@...il.com>,
linux-kernel@...r.kernel.org, linux-mm@...ck.org,
Suren Baghdasaryan <surenb@...gle.com>, Johannes Weiner
<hannes@...xchg.org>, Andrew Morton <akpm@...ux-foundation.org>,
bpf@...r.kernel.org
Subject: Re: [PATCH bpf-next v3 07/17] mm: introduce BPF OOM struct ops
On 1/26/26 6:44 PM, Roman Gushchin wrote:
> +bool bpf_handle_oom(struct oom_control *oc)
> +{
> + struct bpf_struct_ops_link *st_link;
> + struct bpf_oom_ops *bpf_oom_ops;
> + struct mem_cgroup *memcg;
> + struct bpf_map *map;
> + int ret = 0;
> +
> + /*
> + * System-wide OOMs are handled by the struct ops attached
> + * to the root memory cgroup
> + */
> + memcg = oc->memcg ? oc->memcg : root_mem_cgroup;
> +
> + rcu_read_lock_trace();
> +
> + /* Find the nearest bpf_oom_ops traversing the cgroup tree upwards */
> + for (; memcg; memcg = parent_mem_cgroup(memcg)) {
> + st_link = rcu_dereference_check(memcg->css.cgroup->bpf.bpf_oom_link,
> + rcu_read_lock_trace_held());
> + if (!st_link)
> + continue;
> +
> + map = rcu_dereference_check((st_link->map),
> + rcu_read_lock_trace_held());
> + if (!map)
> + continue;
> +
> + /* Call BPF OOM handler */
> + bpf_oom_ops = bpf_struct_ops_data(map);
> + ret = bpf_ops_handle_oom(bpf_oom_ops, st_link, oc);
> + if (ret && oc->bpf_memory_freed)
> + break;
> + ret = 0;
> + }
> +
> + rcu_read_unlock_trace();
> +
> + return ret && oc->bpf_memory_freed;
> +}
> +
[ ... ]
> +static int bpf_oom_ops_reg(void *kdata, struct bpf_link *link)
> +{
> + struct bpf_struct_ops_link *st_link = (struct bpf_struct_ops_link *)link;
> + struct cgroup *cgrp;
> +
> + /* The link is not yet fully initialized, but cgroup should be set */
> + if (!link)
> + return -EOPNOTSUPP;
> +
> + cgrp = st_link->cgroup;
> + if (!cgrp)
> + return -EINVAL;
> +
> + if (cmpxchg(&cgrp->bpf.bpf_oom_link, NULL, st_link))
> + return -EEXIST;
iiuc, this will allow only one oom_ops to be attached to a cgroup.
Considering oom_ops is the only user of the cgrp->bpf.struct_ops_links
(added in patch 2), the list should have only one element for now.
Copy some context from the patch 2 commit log.
> This change doesn't answer the question how bpf programs belonging
> to these struct ops'es will be executed. It will be done individually
> for every bpf struct ops which supports this.
>
> Please, note that unlike "normal" bpf programs, struct ops'es
> are not propagated to cgroup sub-trees.
There are NONE, BPF_F_ALLOW_OVERRIDE, and BPF_F_ALLOW_MULTI, which one
may be closer to the bpf_handle_oom() semantic. If it needs to change
the ordering (or allow multi) in the future, does it need a new flag or
the existing BPF_F_xxx flags can be used.
Powered by blists - more mailing lists