187091063SKumar Kartikeya Dwivedi // SPDX-License-Identifier: GPL-2.0 287091063SKumar Kartikeya Dwivedi #include <test_progs.h> 387091063SKumar Kartikeya Dwivedi #include <network_helpers.h> 487091063SKumar Kartikeya Dwivedi #include "test_bpf_nf.skel.h" 587091063SKumar Kartikeya Dwivedi 687091063SKumar Kartikeya Dwivedi enum { 787091063SKumar Kartikeya Dwivedi TEST_XDP, 887091063SKumar Kartikeya Dwivedi TEST_TC_BPF, 987091063SKumar Kartikeya Dwivedi }; 1087091063SKumar Kartikeya Dwivedi 1187091063SKumar Kartikeya Dwivedi void test_bpf_nf_ct(int mode) 1287091063SKumar Kartikeya Dwivedi { 1387091063SKumar Kartikeya Dwivedi struct test_bpf_nf *skel; 1404fcb5f9SDelyan Kratunov int prog_fd, err; 1504fcb5f9SDelyan Kratunov LIBBPF_OPTS(bpf_test_run_opts, topts, 1604fcb5f9SDelyan Kratunov .data_in = &pkt_v4, 1704fcb5f9SDelyan Kratunov .data_size_in = sizeof(pkt_v4), 1804fcb5f9SDelyan Kratunov .repeat = 1, 1904fcb5f9SDelyan Kratunov ); 2087091063SKumar Kartikeya Dwivedi 2187091063SKumar Kartikeya Dwivedi skel = test_bpf_nf__open_and_load(); 2287091063SKumar Kartikeya Dwivedi if (!ASSERT_OK_PTR(skel, "test_bpf_nf__open_and_load")) 2387091063SKumar Kartikeya Dwivedi return; 2487091063SKumar Kartikeya Dwivedi 2587091063SKumar Kartikeya Dwivedi if (mode == TEST_XDP) 2687091063SKumar Kartikeya Dwivedi prog_fd = bpf_program__fd(skel->progs.nf_xdp_ct_test); 2787091063SKumar Kartikeya Dwivedi else 2887091063SKumar Kartikeya Dwivedi prog_fd = bpf_program__fd(skel->progs.nf_skb_ct_test); 2987091063SKumar Kartikeya Dwivedi 3004fcb5f9SDelyan Kratunov err = bpf_prog_test_run_opts(prog_fd, &topts); 3187091063SKumar Kartikeya Dwivedi if (!ASSERT_OK(err, "bpf_prog_test_run")) 3287091063SKumar Kartikeya Dwivedi goto end; 3387091063SKumar Kartikeya Dwivedi 3487091063SKumar Kartikeya Dwivedi ASSERT_EQ(skel->bss->test_einval_bpf_tuple, -EINVAL, "Test EINVAL for NULL bpf_tuple"); 3587091063SKumar Kartikeya Dwivedi ASSERT_EQ(skel->bss->test_einval_reserved, -EINVAL, "Test EINVAL for reserved not set to 0"); 3687091063SKumar Kartikeya Dwivedi ASSERT_EQ(skel->bss->test_einval_netns_id, -EINVAL, "Test EINVAL for netns_id < -1"); 3787091063SKumar Kartikeya Dwivedi ASSERT_EQ(skel->bss->test_einval_len_opts, -EINVAL, "Test EINVAL for len__opts != NF_BPF_CT_OPTS_SZ"); 3887091063SKumar Kartikeya Dwivedi ASSERT_EQ(skel->bss->test_eproto_l4proto, -EPROTO, "Test EPROTO for l4proto != TCP or UDP"); 3987091063SKumar Kartikeya Dwivedi ASSERT_EQ(skel->bss->test_enonet_netns_id, -ENONET, "Test ENONET for bad but valid netns_id"); 4087091063SKumar Kartikeya Dwivedi ASSERT_EQ(skel->bss->test_enoent_lookup, -ENOENT, "Test ENOENT for failed lookup"); 4187091063SKumar Kartikeya Dwivedi ASSERT_EQ(skel->bss->test_eafnosupport, -EAFNOSUPPORT, "Test EAFNOSUPPORT for invalid len__tuple"); 42*6eb7fba0SLorenzo Bianconi ASSERT_EQ(skel->data->test_alloc_entry, 0, "Test for alloc new entry"); 43*6eb7fba0SLorenzo Bianconi ASSERT_EQ(skel->data->test_insert_entry, 0, "Test for insert new entry"); 44*6eb7fba0SLorenzo Bianconi ASSERT_EQ(skel->data->test_succ_lookup, 0, "Test for successful lookup"); 45*6eb7fba0SLorenzo Bianconi /* allow some tolerance for test_delta_timeout value to avoid races. */ 46*6eb7fba0SLorenzo Bianconi ASSERT_GT(skel->bss->test_delta_timeout, 8, "Test for min ct timeout update"); 47*6eb7fba0SLorenzo Bianconi ASSERT_LE(skel->bss->test_delta_timeout, 10, "Test for max ct timeout update"); 48*6eb7fba0SLorenzo Bianconi /* expected status is IPS_SEEN_REPLY */ 49*6eb7fba0SLorenzo Bianconi ASSERT_EQ(skel->bss->test_status, 2, "Test for ct status update "); 5087091063SKumar Kartikeya Dwivedi end: 5187091063SKumar Kartikeya Dwivedi test_bpf_nf__destroy(skel); 5287091063SKumar Kartikeya Dwivedi } 5387091063SKumar Kartikeya Dwivedi 5487091063SKumar Kartikeya Dwivedi void test_bpf_nf(void) 5587091063SKumar Kartikeya Dwivedi { 5687091063SKumar Kartikeya Dwivedi if (test__start_subtest("xdp-ct")) 5787091063SKumar Kartikeya Dwivedi test_bpf_nf_ct(TEST_XDP); 5887091063SKumar Kartikeya Dwivedi if (test__start_subtest("tc-bpf-ct")) 5987091063SKumar Kartikeya Dwivedi test_bpf_nf_ct(TEST_TC_BPF); 6087091063SKumar Kartikeya Dwivedi } 61