[PATCH bpf-next 2/2] selftests: bpf: add selftest for __sk_buff tstamp

[Stanislav Fomichev]

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

Signed-off-by: Stanislav Fomichev <s...@google.com>
---
 tools/testing/selftests/bpf/prog_tests/skb_ctx.c | 5 +++++
 tools/testing/selftests/bpf/progs/test_skb_ctx.c | 1 +
 2 files changed, 6 insertions(+)

diff --git a/tools/testing/selftests/bpf/prog_tests/skb_ctx.c 
b/tools/testing/selftests/bpf/prog_tests/skb_ctx.c
index e95baa32e277..a2eb8db8dafb 100644
--- 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 7a80960d7df1..2a9f4c736ebc 100644
--- 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;
 }
-- 
2.23.0.700.g56cf767bdb-goog