UbiquitousOS
/
OpenCloudOS-Kernel
mirror of https://gitee.com/OpenCloudOS/OpenCloudOS-Kernel.git
11
0
Fork 0
Code Issues Projects Releases Wiki Activity

bpf: Fix bpf_put_raw_tracepoint()'s use of __module_address() 

__module_address() needs to be called with preemption disabled or with
module_mutex taken. preempt_disable() is enough for read-only uses, which is
what this fix does. Also, module_put() does internal check for NULL, so drop
it as well.

Fixes: a38d1107f9 ("bpf: support raw tracepoints in modules")
Signed-off-by: Andrii Nakryiko <andrii@kernel.org>
Signed-off-by: Alexei Starovoitov <ast@kernel.org>
Acked-by: Martin KaFai Lau <kafai@fb.com>
Link: https://lore.kernel.org/bpf/20201203204634.1325171-2-andrii@kernel.org
This commit is contained in:
Andrii Nakryiko 2020-12-03 12:46:21 -08:00 committed by Alexei Starovoitov
parent cadd64807c
commit 12cc126df8
1 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
kernel/trace/bpf_trace.c
View File

@ -2060,10 +2060,12 @@ struct bpf_raw_event_map *bpf_get_raw_tracepoint(const char *name)
void bpf_put_raw_tracepoint(struct bpf_raw_event_map *btp) void bpf_put_raw_tracepoint(struct bpf_raw_event_map *btp)
{ {
struct module *mod = __module_address((unsigned long)btp); struct module *mod;
if (mod) preempt_disable();
mod = __module_address((unsigned long)btp);
module_put(mod); module_put(mod);
preempt_enable();
} }
static __always_inline static __always_inline