selftests: bpf: Add selftest for __sk_buff tstamp

Make sure BPF_PROG_TEST_RUN accepts tstamp and exports any
modifications that BPF program does.

Signed-off-by: Stanislav Fomichev <sdf@google.com>
Signed-off-by: Alexei Starovoitov <ast@kernel.org>
Acked-by: Martin KaFai Lau <kafai@fb.com>
Link: https://lore.kernel.org/bpf/20191015183125.124413-2-sdf@google.com

diff --git a/tools/testing/selftests/bpf/prog_tests/skb_ctx.c b/tools/testing/selftests/bpf/prog_tests/skb_ctx.c
index e95baa32e2778daba75b5af14eb09f8992289df7..a2eb8db8dafb65c76f4c427bd2083638e3a3136b 100644 (file)
--- a/tools/testing/selftests/bpf/prog_tests/skb_ctx.c
+++ b/tools/testing/selftests/bpf/prog_tests/skb_ctx.c
@@ -10,6 +10,7 @@ void test_skb_ctx(void)
                .cb[3] = 4,
                .cb[4] = 5,
                .priority = 6,
+               .tstamp = 7,
        };
        struct bpf_prog_test_run_attr tattr = {
                .data_in = &pkt_v4,
@@ -86,4 +87,8 @@ void test_skb_ctx(void)
                   "ctx_out_priority",
                   "skb->priority == %d, expected %d\n",
                   skb.priority, 7);
+       CHECK_ATTR(skb.tstamp != 8,
+                  "ctx_out_tstamp",
+                  "skb->tstamp == %lld, expected %d\n",
+                  skb.tstamp, 8);
 }

diff --git a/tools/testing/selftests/bpf/progs/test_skb_ctx.c b/tools/testing/selftests/bpf/progs/test_skb_ctx.c
index 7a80960d7df134602a8f27ac6e10b1d12347ae03..2a9f4c736ebc188e274868a8497e36798eb0fab3 100644 (file)
--- a/tools/testing/selftests/bpf/progs/test_skb_ctx.c
+++ b/tools/testing/selftests/bpf/progs/test_skb_ctx.c
@@ -16,6 +16,7 @@ int process(struct __sk_buff *skb)
                skb->cb[i]++;
        }
        skb->priority++;
+       skb->tstamp++;
 
        return 0;
 }