1 files changed, 9 insertions, 1 deletions

diff --git a/kernel/bpf/syscall.c b/kernel/bpf/syscall.c
index 438199e2eca4..d995eedfdd16 100644
--- a/kernel/bpf/syscall.c
+++ b/kernel/bpf/syscall.c
@@ -2009,7 +2009,7 @@ static int bpf_prog_query(const union bpf_attr *attr,
         return cgroup_bpf_prog_query(attr, uattr);
 }
 
-#define BPF_PROG_TEST_RUN_LAST_FIELD test.duration
+#define BPF_PROG_TEST_RUN_LAST_FIELD test.ctx_out
 
 static int bpf_prog_test_run(const union bpf_attr *attr,
                              union bpf_attr __user *uattr)
@@ -2022,6 +2022,14 @@ static int bpf_prog_test_run(const union bpf_attr *attr,
         if (CHECK_ATTR(BPF_PROG_TEST_RUN))
                 return -EINVAL;
 
+        if ((attr->test.ctx_size_in && !attr->test.ctx_in) ||
+            (!attr->test.ctx_size_in && attr->test.ctx_in))
+                return -EINVAL;
+
+        if ((attr->test.ctx_size_out && !attr->test.ctx_out) ||
+            (!attr->test.ctx_size_out && attr->test.ctx_out))
+                return -EINVAL;
+
         prog = bpf_prog_get(attr->test.prog_fd);
         if (IS_ERR(prog))
                 return PTR_ERR(prog);