tools/lib/bpf: add support for BPF_PROG_TEST_RUN command

add support for BPF_PROG_TEST_RUN command to libbpf.a

Signed-off-by: Alexei Starovoitov <ast@kernel.org>
Acked-by: Daniel Borkmann <daniel@iogearbox.net>
Acked-by: Martin KaFai Lau <kafai@fb.com>
Acked-by: Wang Nan <wangnan0@huawei.com>
Signed-off-by: David S. Miller <davem@davemloft.net>

1 files changed, 24 insertions, 0 deletions

diff --git a/tools/lib/bpf/bpf.c b/tools/lib/bpf/bpf.c
index 9b58d20e8c93..f84c398c11f4 100644
--- a/tools/lib/bpf/bpf.c
+++ b/tools/lib/bpf/bpf.c
@@ -209,3 +209,27 @@ int bpf_prog_detach(int target_fd, enum bpf_attach_type type)
 
         return sys_bpf(BPF_PROG_DETACH, &attr, sizeof(attr));
 }
+
+int bpf_prog_test_run(int prog_fd, int repeat, void *data, __u32 size,
+                      void *data_out, __u32 *size_out, __u32 *retval,
+                      __u32 *duration)
+{
+        union bpf_attr attr;
+        int ret;
+
+        bzero(&attr, sizeof(attr));
+        attr.test.prog_fd = prog_fd;
+        attr.test.data_in = ptr_to_u64(data);
+        attr.test.data_out = ptr_to_u64(data_out);
+        attr.test.data_size_in = size;
+        attr.test.repeat = repeat;
+
+        ret = sys_bpf(BPF_PROG_TEST_RUN, &attr, sizeof(attr));
+        if (size_out)
+                *size_out = attr.test.data_size_out;
+        if (retval)
+                *retval = attr.test.retval;
+        if (duration)
+                *duration = attr.test.duration;
+        return ret;
+}