tracepoints: Use new static branch API

The old static key API is deprecated.  Switch to the new one.

Cc: Masami Hiramatsu <mhiramat@kernel.org>
Cc: Mathieu Desnoyers <mathieu.desnoyers@efficios.com>
Cc: Alice Ryhl <aliceryhl@google.com>
Link: https://lore.kernel.org/7a08dae3c5eddb14b13864923c1b58ac1f4af83c.1728414936.git.jpoimboe@kernel.org
Signed-off-by: Josh Poimboeuf <jpoimboe@kernel.org>
Signed-off-by: Steven Rostedt (Google) <rostedt@goodmis.org>
This commit is contained in:
Josh Poimboeuf 2024-10-08 12:17:19 -07:00 committed by Steven Rostedt (Google)
parent 49e4154f4b
commit 4a8840af5f
5 changed files with 11 additions and 11 deletions

View File

@ -31,7 +31,7 @@ struct tracepoint_func {
struct tracepoint { struct tracepoint {
const char *name; /* Tracepoint name */ const char *name; /* Tracepoint name */
struct static_key key; struct static_key_false key;
struct static_call_key *static_call_key; struct static_call_key *static_call_key;
void *static_call_tramp; void *static_call_tramp;
void *iterator; void *iterator;
@ -83,7 +83,7 @@ struct bpf_raw_event_map {
#ifdef CONFIG_TRACEPOINTS #ifdef CONFIG_TRACEPOINTS
# define tracepoint_enabled(tp) \ # define tracepoint_enabled(tp) \
static_key_false(&(__tracepoint_##tp).key) static_branch_unlikely(&(__tracepoint_##tp).key)
#else #else
# define tracepoint_enabled(tracepoint) false # define tracepoint_enabled(tracepoint) false
#endif #endif

View File

@ -248,7 +248,7 @@ static inline struct tracepoint *tracepoint_ptr_deref(tracepoint_ptr_t *p)
#define __DECLARE_TRACE_RCU(name, proto, args, cond) \ #define __DECLARE_TRACE_RCU(name, proto, args, cond) \
static inline void trace_##name##_rcuidle(proto) \ static inline void trace_##name##_rcuidle(proto) \
{ \ { \
if (static_key_false(&__tracepoint_##name.key)) \ if (static_branch_unlikely(&__tracepoint_##name.key)) \
__DO_TRACE(name, \ __DO_TRACE(name, \
TP_ARGS(args), \ TP_ARGS(args), \
TP_CONDITION(cond), 1); \ TP_CONDITION(cond), 1); \
@ -274,7 +274,7 @@ static inline struct tracepoint *tracepoint_ptr_deref(tracepoint_ptr_t *p)
extern struct tracepoint __tracepoint_##name; \ extern struct tracepoint __tracepoint_##name; \
static inline void trace_##name(proto) \ static inline void trace_##name(proto) \
{ \ { \
if (static_key_false(&__tracepoint_##name.key)) \ if (static_branch_unlikely(&__tracepoint_##name.key)) \
__DO_TRACE(name, \ __DO_TRACE(name, \
TP_ARGS(args), \ TP_ARGS(args), \
TP_CONDITION(cond), 0); \ TP_CONDITION(cond), 0); \
@ -311,7 +311,7 @@ static inline struct tracepoint *tracepoint_ptr_deref(tracepoint_ptr_t *p)
static inline bool \ static inline bool \
trace_##name##_enabled(void) \ trace_##name##_enabled(void) \
{ \ { \
return static_key_false(&__tracepoint_##name.key); \ return static_branch_unlikely(&__tracepoint_##name.key);\
} }
/* /*
@ -328,7 +328,7 @@ static inline struct tracepoint *tracepoint_ptr_deref(tracepoint_ptr_t *p)
struct tracepoint __tracepoint_##_name __used \ struct tracepoint __tracepoint_##_name __used \
__section("__tracepoints") = { \ __section("__tracepoints") = { \
.name = __tpstrtab_##_name, \ .name = __tpstrtab_##_name, \
.key = STATIC_KEY_INIT_FALSE, \ .key = STATIC_KEY_FALSE_INIT, \
.static_call_key = &STATIC_CALL_KEY(tp_func_##_name), \ .static_call_key = &STATIC_CALL_KEY(tp_func_##_name), \
.static_call_tramp = STATIC_CALL_TRAMP_ADDR(tp_func_##_name), \ .static_call_tramp = STATIC_CALL_TRAMP_ADDR(tp_func_##_name), \
.iterator = &__traceiter_##_name, \ .iterator = &__traceiter_##_name, \

View File

@ -822,7 +822,7 @@ static inline void trace_synth(struct synth_event *event, u64 *var_ref_vals,
{ {
struct tracepoint *tp = event->tp; struct tracepoint *tp = event->tp;
if (unlikely(atomic_read(&tp->key.enabled) > 0)) { if (unlikely(static_key_enabled(&tp->key))) {
struct tracepoint_func *probe_func_ptr; struct tracepoint_func *probe_func_ptr;
synth_probe_func_t probe_func; synth_probe_func_t probe_func;
void *__data; void *__data;

View File

@ -1676,7 +1676,7 @@ static void update_enable_bit_for(struct user_event *user)
struct tracepoint *tp = &user->tracepoint; struct tracepoint *tp = &user->tracepoint;
char status = 0; char status = 0;
if (atomic_read(&tp->key.enabled) > 0) { if (static_key_enabled(&tp->key)) {
struct tracepoint_func *probe_func_ptr; struct tracepoint_func *probe_func_ptr;
user_event_func_t probe_func; user_event_func_t probe_func;
@ -2280,7 +2280,7 @@ static ssize_t user_events_write_core(struct file *file, struct iov_iter *i)
* It's possible key.enabled disables after this check, however * It's possible key.enabled disables after this check, however
* we don't mind if a few events are included in this condition. * we don't mind if a few events are included in this condition.
*/ */
if (likely(atomic_read(&tp->key.enabled) > 0)) { if (likely(static_key_enabled(&tp->key))) {
struct tracepoint_func *probe_func_ptr; struct tracepoint_func *probe_func_ptr;
user_event_func_t probe_func; user_event_func_t probe_func;
struct iov_iter copy; struct iov_iter copy;

View File

@ -358,7 +358,7 @@ static int tracepoint_add_func(struct tracepoint *tp,
tracepoint_update_call(tp, tp_funcs); tracepoint_update_call(tp, tp_funcs);
/* Both iterator and static call handle NULL tp->funcs */ /* Both iterator and static call handle NULL tp->funcs */
rcu_assign_pointer(tp->funcs, tp_funcs); rcu_assign_pointer(tp->funcs, tp_funcs);
static_key_enable(&tp->key); static_branch_enable(&tp->key);
break; break;
case TP_FUNC_2: /* 1->2 */ case TP_FUNC_2: /* 1->2 */
/* Set iterator static call */ /* Set iterator static call */
@ -414,7 +414,7 @@ static int tracepoint_remove_func(struct tracepoint *tp,
if (tp->unregfunc && static_key_enabled(&tp->key)) if (tp->unregfunc && static_key_enabled(&tp->key))
tp->unregfunc(); tp->unregfunc();
static_key_disable(&tp->key); static_branch_disable(&tp->key);
/* Set iterator static call */ /* Set iterator static call */
tracepoint_update_call(tp, tp_funcs); tracepoint_update_call(tp, tp_funcs);
/* Both iterator and static call handle NULL tp->funcs */ /* Both iterator and static call handle NULL tp->funcs */