Add .test_run for tp_btf. Use the .test_run for raw_tp.

Signed-off-by: KaFai Wan <[email protected]>
---
 net/bpf/test_run.c | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/net/bpf/test_run.c b/net/bpf/test_run.c
index 8cb285187270..8c901ec92341 100644
--- a/net/bpf/test_run.c
+++ b/net/bpf/test_run.c
@@ -690,6 +690,9 @@ int bpf_prog_test_run_tracing(struct bpf_prog *prog,
        int b = 2, err = -EFAULT;
        u32 retval = 0;
 
+       if (prog->expected_attach_type == BPF_TRACE_RAW_TP)
+               return bpf_prog_test_run_raw_tp(prog, kattr, uattr);
+
        if (kattr->test.flags || kattr->test.cpu || kattr->test.batch_size)
                return -EINVAL;
 
-- 
2.43.0


Reply via email to