1 files changed, 2 insertions, 0 deletions

diff --git a/kernel/bpf/syscall.c b/kernel/bpf/syscall.c
index f74ca17af64a..461eb1e66a0f 100644
--- a/kernel/bpf/syscall.c
+++ b/kernel/bpf/syscall.c
@@ -707,6 +707,7 @@ void bpf_prog_put(struct bpf_prog *prog)
 {
         if (atomic_dec_and_test(&prog->aux->refcnt)) {
                 trace_bpf_prog_put_rcu(prog);
+                bpf_prog_kallsyms_del(prog);
                 call_rcu(&prog->aux->rcu, __bpf_prog_put_rcu);
         }
 }
@@ -903,6 +904,7 @@ static int bpf_prog_load(union bpf_attr *attr)
                 /* failed to allocate fd */
                 goto free_used_maps;
 
+        bpf_prog_kallsyms_add(prog);
         trace_bpf_prog_load(prog, err);
         return err;