eventpoll: split ep_clear_and_put() into drain helpers

ep_clear_and_put()'s two-pass walk is the main way an ep file close
tears down its state, and the ordering between the passes is
load-bearing (see previous commit's docblock). Give each pass its
own function so the ordering is enforced by the call sequence in
ep_clear_and_put() rather than by convention inside one body.

ep_drain_pollwaits() carries out Pass 1: walk the rbtree and
ep_unregister_pollwait() each epi. The function-level comment names
it as Pass 1 and spells out the synchronization contract with
ep_poll_callback().

ep_drain_tree() carries out Pass 2: walk the rbtree and ep_remove()
each epi, capturing rb_next() before each erase. The comment names
it as Pass 2 and documents the hand-off with a concurrent
eventpoll_release_file() (removal path C).

ep_clear_and_put() keeps the poll-on-ep wakeup, ep->mtx bracketing,
and ep_put() + conditional ep_free(), and its docblock shrinks to
the high-level summary; the per-pass detail moved into the helpers.

No functional change.

Signed-off-by: Christian Brauner (Amutable) <brauner@kernel.org>
Link: https://patch.msgid.link/20260424-work-epoll-rework-v1-11-249ed00a20f3@kernel.org
Signed-off-by: Christian Brauner <brauner@kernel.org>
This commit is contained in:
Christian Brauner 2026-04-24 15:46:42 +02:00
parent e0e35f4cb9
commit 08fe3679a8
No known key found for this signature in database
GPG Key ID: 91C61BC06578DCA2

View File

@ -1105,62 +1105,77 @@ static void ep_remove(struct eventpoll *ep, struct epitem *epi)
}
/*
* Removal path B (see "Removal paths" in the top-of-file banner):
* close of the epoll fd itself, reached via ep_eventpoll_release().
*
* Under ep->mtx we walk the rbtree twice:
*
* Pass 1 drains each epi's pwqlist via ep_unregister_pollwait().
* This takes each watched waitqueue head's lock and so
* synchronizes with any in-flight ep_poll_callback(), so
* after the pass ends no callback can still be holding or
* about to dereference any epi on this ep.
*
* Pass 2 runs ep_remove() on each epi. The per-epi pwqlist is
* already empty, but the rest of ep_remove() still runs
* (epi_fget() pin, f_ep clear under f_lock, rbtree erase,
* rdllist unlink, kfree_rcu).
*
* Pass 1 must strictly precede Pass 2: fusing them would let a
* callback queued on epi_i still fire after epi_{i+k} was freed.
*
* A concurrent eventpoll_release_file() (path C) serializes against
* us on ep->mtx; in Pass 2, ep_remove() transparently hands off any
* epi whose watched file is in __fput() by bailing when epi_fget()
* returns NULL, and C will clean that epi up on its side.
*
* ep->refcount is held > 0 throughout by the ep file's own share;
* we drop that share after the walk and free the eventpoll if we
* were last.
* Pass 1 of ep_clear_and_put(): drain every epi's pwqlist.
* ep_unregister_pollwait() takes each watched wait-queue head's lock,
* which synchronizes with any in-flight ep_poll_callback(); after
* this returns no callback can still be about to dereference an epi
* on this ep. Must strictly precede ep_drain_tree() -- fusing the
* two walks would let a callback queued on epi_i still fire after
* epi_{i+k} had already been freed.
*/
static void ep_clear_and_put(struct eventpoll *ep)
static void ep_drain_pollwaits(struct eventpoll *ep)
{
struct rb_node *rbp, *next;
struct rb_node *rbp;
struct epitem *epi;
/* Release any threads blocked in poll-on-ep. */
if (waitqueue_active(&ep->poll_wait))
ep_poll_safewake(ep, NULL, 0);
lockdep_assert_held(&ep->mtx);
mutex_lock(&ep->mtx);
/* Pass 1: drain pwqlists; synchronizes with in-flight callbacks. */
for (rbp = rb_first_cached(&ep->rbr); rbp; rbp = rb_next(rbp)) {
epi = rb_entry(rbp, struct epitem, rbn);
ep_unregister_pollwait(ep, epi);
cond_resched();
}
}
/*
* Pass 2 of ep_clear_and_put(): ep_remove() every epi. The per-epi
* pwqlist is already empty (ep_drain_pollwaits ran), but the rest of
* ep_remove() still runs: epi_fget() pin, f_ep clear under f_lock,
* rbtree erase, rdllist unlink, kfree_rcu(epi). rb_next() is captured
* before each erase so the iteration is stable.
*
* A concurrent eventpoll_release_file() (removal path C) on a watched
* file serializes with us via ep->mtx; ep_remove() transparently
* hands off any epi whose file is in __fput() by bailing when
* epi_fget() returns NULL, and path C will clean that epi up.
*/
static void ep_drain_tree(struct eventpoll *ep)
{
struct rb_node *rbp, *next;
struct epitem *epi;
lockdep_assert_held(&ep->mtx);
/* Pass 2: remove each epi. rb_next() is captured before erase. */
for (rbp = rb_first_cached(&ep->rbr); rbp; rbp = next) {
next = rb_next(rbp);
epi = rb_entry(rbp, struct epitem, rbn);
ep_remove(ep, epi);
cond_resched();
}
}
/*
* Removal path B (see "Removal paths" in the top-of-file banner):
* close of the epoll fd itself, reached via ep_eventpoll_release().
*
* Two passes under ep->mtx: first ep_drain_pollwaits() quiesces
* in-flight callbacks, then ep_drain_tree() frees the epis. The
* ep->refcount is kept > 0 across the walk by the ep file's own
* share, which we drop below; ep_free() runs iff we were the last
* holder after the tree drained.
*/
static void ep_clear_and_put(struct eventpoll *ep)
{
/* Release any threads blocked in poll-on-ep. */
if (waitqueue_active(&ep->poll_wait))
ep_poll_safewake(ep, NULL, 0);
mutex_lock(&ep->mtx);
ep_drain_pollwaits(ep);
ep_drain_tree(ep);
mutex_unlock(&ep->mtx);
if (ep_put(ep))
ep_free(ep);
}