[<prev] [next>] [thread-next>] [day] [month] [year] [list]
Message-ID: <20251006175848.1906912-1-sashal@kernel.org>
Date: Mon, 6 Oct 2025 13:58:47 -0400
From: Sasha Levin <sashal@...nel.org>
To: rostedt@...dmis.org,
mhiramat@...nel.org
Cc: mathieu.desnoyers@...icios.com,
linux-kernel@...r.kernel.org,
linux-trace-kernel@...r.kernel.org,
Sasha Levin <sashal@...nel.org>
Subject: [PATCH 1/2] tracing/wakeup: Balance prolog on fgraph failure
Commit a485ea9e3ef3 ("tracing: Fix irqsoff and wakeup latency tracers when
using function graph") added fgraph_reserve_data()/fgraph_retrieve_data() to
the wakeup tracer's function-graph hooks.
When func_prolog_preempt_disable() succeeded but those calls failed, the code
returned without undoing the prolog (no local_dec(&data->disabled), no
preempt_enable_notrace()). That leaked a preempt disable into later code and
could trigger "BUG: scheduling while atomic".
Balance the prolog on these failure paths in both entry and return hooks.
Fixes: a485ea9e3ef3 ("tracing: Fix irqsoff and wakeup latency tracers when using function graph")
Assisted-by: gpt-5-codex
Signed-off-by: Sasha Levin <sashal@...nel.org>
---
kernel/trace/trace_sched_wakeup.c | 10 ++++++++--
1 file changed, 8 insertions(+), 2 deletions(-)
diff --git a/kernel/trace/trace_sched_wakeup.c b/kernel/trace/trace_sched_wakeup.c
index bf1cb80742aed..fa48bbdf0851c 100644
--- a/kernel/trace/trace_sched_wakeup.c
+++ b/kernel/trace/trace_sched_wakeup.c
@@ -138,8 +138,11 @@ static int wakeup_graph_entry(struct ftrace_graph_ent *trace,
return 0;
calltime = fgraph_reserve_data(gops->idx, sizeof(*calltime));
- if (!calltime)
+ if (!calltime) {
+ local_dec(&data->disabled);
+ preempt_enable_notrace();
return 0;
+ }
*calltime = trace_clock_local();
@@ -169,8 +172,11 @@ static void wakeup_graph_return(struct ftrace_graph_ret *trace,
rettime = trace_clock_local();
calltime = fgraph_retrieve_data(gops->idx, &size);
- if (!calltime)
+ if (!calltime) {
+ local_dec(&data->disabled);
+ preempt_enable_notrace();
return;
+ }
__trace_graph_return(tr, trace, trace_ctx, *calltime, rettime);
local_dec(&data->disabled);
--
2.51.0
Powered by blists - more mailing lists