1 files changed, 6 insertions, 2 deletions

diff --git a/kernel/bpf/syscall.c b/kernel/bpf/syscall.c
index b0de49598341..ae141e745f92 100644
--- a/kernel/bpf/syscall.c
+++ b/kernel/bpf/syscall.c
@@ -1789,12 +1789,16 @@ static int bpf_raw_tracepoint_open(const union bpf_attr *attr)
         }
         raw_tp->btp = btp;
 
-        prog = bpf_prog_get_type(attr->raw_tracepoint.prog_fd,
-                                 BPF_PROG_TYPE_RAW_TRACEPOINT);
+        prog = bpf_prog_get(attr->raw_tracepoint.prog_fd);
         if (IS_ERR(prog)) {
                 err = PTR_ERR(prog);
                 goto out_free_tp;
         }
+        if (prog->type != BPF_PROG_TYPE_RAW_TRACEPOINT &&
+            prog->type != BPF_PROG_TYPE_RAW_TRACEPOINT_WRITABLE) {
+                err = -EINVAL;
+                goto out_put_prog;
+        }
 
         err = bpf_probe_register(raw_tp->btp, prog);
         if (err)