1 // SPDX-License-Identifier: GPL-2.0
2 #include <test_progs.h>
3 #include <network_helpers.h>
5 void test_skb_ctx(void)
7 struct __sk_buff skb = {
20 struct bpf_prog_test_run_attr tattr = {
22 .data_size_in = sizeof(pkt_v4),
24 .ctx_size_in = sizeof(skb),
26 .ctx_size_out = sizeof(skb),
28 struct bpf_object *obj;
32 err = bpf_prog_load("./test_skb_ctx.o", BPF_PROG_TYPE_SCHED_CLS, &obj,
34 if (CHECK_ATTR(err, "load", "err %d errno %d\n", err, errno))
37 /* ctx_in != NULL, ctx_size_in == 0 */
39 tattr.ctx_size_in = 0;
40 err = bpf_prog_test_run_xattr(&tattr);
41 CHECK_ATTR(err == 0, "ctx_size_in", "err %d errno %d\n", err, errno);
42 tattr.ctx_size_in = sizeof(skb);
44 /* ctx_out != NULL, ctx_size_out == 0 */
46 tattr.ctx_size_out = 0;
47 err = bpf_prog_test_run_xattr(&tattr);
48 CHECK_ATTR(err == 0, "ctx_size_out", "err %d errno %d\n", err, errno);
49 tattr.ctx_size_out = sizeof(skb);
51 /* non-zero [len, tc_index] fields should be rejected*/
54 err = bpf_prog_test_run_xattr(&tattr);
55 CHECK_ATTR(err == 0, "len", "err %d errno %d\n", err, errno);
59 err = bpf_prog_test_run_xattr(&tattr);
60 CHECK_ATTR(err == 0, "tc_index", "err %d errno %d\n", err, errno);
63 /* non-zero [hash, sk] fields should be rejected */
66 err = bpf_prog_test_run_xattr(&tattr);
67 CHECK_ATTR(err == 0, "hash", "err %d errno %d\n", err, errno);
70 skb.sk = (struct bpf_sock *)1;
71 err = bpf_prog_test_run_xattr(&tattr);
72 CHECK_ATTR(err == 0, "sk", "err %d errno %d\n", err, errno);
75 err = bpf_prog_test_run_xattr(&tattr);
76 CHECK_ATTR(err != 0 || tattr.retval,
78 "err %d errno %d retval %d\n",
79 err, errno, tattr.retval);
81 CHECK_ATTR(tattr.ctx_size_out != sizeof(skb),
83 "incorrect output size, want %lu have %u\n",
84 sizeof(skb), tattr.ctx_size_out);
86 for (i = 0; i < 5; i++)
87 CHECK_ATTR(skb.cb[i] != i + 2,
89 "skb->cb[i] == %d, expected %d\n",
91 CHECK_ATTR(skb.priority != 7,
93 "skb->priority == %d, expected %d\n",
95 CHECK_ATTR(skb.tstamp != 8,
97 "skb->tstamp == %lld, expected %d\n",
99 CHECK_ATTR(skb.mark != 10,
101 "skb->mark == %u, expected %d\n",