Test

Signed-off-by: Paul Chaignon <[email protected]>

Author: pchaigno

tools/testing/selftests/bpf/test_loader.c

@@ -913,7 +913,7 @@ static bool is_unpriv_capable_map(struct bpf_map *map)
 	}
 }
 
-static int do_prog_test_run(int fd_prog, int *retval, bool empty_opts, int linear_sz)
+static int do_prog_test_run(int fd_prog, int program_type, int *retval, bool empty_opts, int linear_sz)
 {
 	__u8 tmp_out[TEST_DATA_LEN << 2] = {};
 	__u8 tmp_in[TEST_DATA_LEN] = {};
@@ -927,6 +927,9 @@ static int do_prog_test_run(int fd_prog, int *retval, bool empty_opts, int linea
 		.repeat = 1,
 	);
 
+	if (program_type == BPF_PROG_TYPE_SCHED_CLS || program_type == BPF_PROG_TYPE_SCHED_ACT)
+		linear_sz = topts.data_size_in - 1;
+
 	if (linear_sz) {
 		ctx.data_end = linear_sz;
 		topts.ctx_in = &ctx;
@@ -1181,7 +1184,7 @@ void run_subtest(struct test_loader *tester,
 			}
 		}
 
-		err = do_prog_test_run(bpf_program__fd(tprog), &retval,
+		err = do_prog_test_run(bpf_program__fd(tprog), bpf_program__type(tprog), &retval,
 				       bpf_program__type(tprog) == BPF_PROG_TYPE_SYSCALL ? true : false,
 				       spec->linear_sz);
 		if (!err && retval != subspec->retval && subspec->retval != POINTER_VALUE) {