diff mbox series

[bpf-next,v3,2/2] selftests/bpf: Add bpf_program__attach_netfilter helper test

Message ID 20230627115839.1034-3-fw@strlen.de (mailing list archive)
State Superseded
Delegated to: BPF
Headers show
Series libbpf: add netfilter link attach helper | expand

Checks

Context Check Description
bpf/vmtest-bpf-next-VM_Test-1 success Logs for ShellCheck
bpf/vmtest-bpf-next-VM_Test-4 success Logs for build for x86_64 with gcc
bpf/vmtest-bpf-next-VM_Test-5 success Logs for build for x86_64 with llvm-16
bpf/vmtest-bpf-next-VM_Test-6 success Logs for set-matrix
bpf/vmtest-bpf-next-VM_Test-2 success Logs for build for aarch64 with gcc
bpf/vmtest-bpf-next-VM_Test-3 success Logs for build for s390x with gcc
bpf/vmtest-bpf-next-VM_Test-7 success Logs for test_maps on aarch64 with gcc
bpf/vmtest-bpf-next-VM_Test-9 success Logs for test_maps on x86_64 with gcc
bpf/vmtest-bpf-next-VM_Test-10 success Logs for test_maps on x86_64 with llvm-16
bpf/vmtest-bpf-next-VM_Test-17 success Logs for test_progs_no_alu32 on x86_64 with gcc
bpf/vmtest-bpf-next-VM_Test-18 success Logs for test_progs_no_alu32 on x86_64 with llvm-16
bpf/vmtest-bpf-next-VM_Test-19 success Logs for test_progs_no_alu32_parallel on aarch64 with gcc
bpf/vmtest-bpf-next-VM_Test-20 success Logs for test_progs_no_alu32_parallel on x86_64 with gcc
bpf/vmtest-bpf-next-VM_Test-21 success Logs for test_progs_no_alu32_parallel on x86_64 with llvm-16
bpf/vmtest-bpf-next-VM_Test-22 success Logs for test_progs_parallel on aarch64 with gcc
bpf/vmtest-bpf-next-VM_Test-23 success Logs for test_progs_parallel on x86_64 with gcc
bpf/vmtest-bpf-next-VM_Test-24 success Logs for test_progs_parallel on x86_64 with llvm-16
bpf/vmtest-bpf-next-VM_Test-25 success Logs for test_verifier on aarch64 with gcc
bpf/vmtest-bpf-next-VM_Test-27 success Logs for test_verifier on x86_64 with gcc
bpf/vmtest-bpf-next-VM_Test-28 success Logs for test_verifier on x86_64 with llvm-16
bpf/vmtest-bpf-next-VM_Test-29 success Logs for veristat
bpf/vmtest-bpf-next-VM_Test-11 success Logs for test_progs on aarch64 with gcc
bpf/vmtest-bpf-next-VM_Test-13 success Logs for test_progs on x86_64 with gcc
bpf/vmtest-bpf-next-VM_Test-14 success Logs for test_progs on x86_64 with llvm-16
bpf/vmtest-bpf-next-VM_Test-15 success Logs for test_progs_no_alu32 on aarch64 with gcc
bpf/vmtest-bpf-next-VM_Test-26 success Logs for test_verifier on s390x with gcc
netdev/series_format success Posting correctly formatted
netdev/tree_selection success Clearly marked for bpf-next
netdev/fixes_present success Fixes tag not required for -next series
netdev/header_inline success No static functions without inline keyword in header files
netdev/build_32bit success Errors and warnings before: 8 this patch: 8
netdev/cc_maintainers warning 14 maintainers not CCed: daniel@iogearbox.net yhs@fb.com kpsingh@kernel.org martin.lau@linux.dev john.fastabend@gmail.com song@kernel.org sdf@google.com andrii@kernel.org shuah@kernel.org mykolal@fb.com linux-kselftest@vger.kernel.org jolsa@kernel.org haoluo@google.com ast@kernel.org
netdev/build_clang success Errors and warnings before: 8 this patch: 8
netdev/verify_signedoff success Signed-off-by tag matches author and committer
netdev/deprecated_api success None detected
netdev/check_selftest success No net selftest shell script
netdev/verify_fixes success No Fixes tag
netdev/build_allmodconfig_warn success Errors and warnings before: 8 this patch: 8
netdev/checkpatch warning WARNING: Missing a blank line after declarations WARNING: added, moved or deleted file(s), does MAINTAINERS need updating? WARNING: line length of 81 exceeds 80 columns WARNING: line length of 91 exceeds 80 columns
netdev/kdoc success Errors and warnings before: 0 this patch: 0
netdev/source_inline success Was 0 now: 0
bpf/vmtest-bpf-next-VM_Test-16 success Logs for test_progs_no_alu32 on s390x with gcc
bpf/vmtest-bpf-next-VM_Test-12 success Logs for test_progs on s390x with gcc
bpf/vmtest-bpf-next-PR success PR summary
bpf/vmtest-bpf-next-VM_Test-8 success Logs for test_maps on s390x with gcc

Commit Message

Florian Westphal June 27, 2023, 11:58 a.m. UTC 
Call bpf_program__attach_netfilter() with different
protocol/hook/priority combinations.

Test fails if supposedly-illegal attachments work
(e.g., bogus protocol family, illegal priority and so on) or if a
should-work attachment fails.  Expected output:

 ./test_progs -t netfilter_link_attach
 #145/1   netfilter_link_attach/netfilter link attach 0:OK
 #145/2   netfilter_link_attach/netfilter link attach 1:OK
 #145/3   netfilter_link_attach/netfilter link attach 2:OK
 #145/4   netfilter_link_attach/netfilter link attach 3:OK
 #145/5   netfilter_link_attach/netfilter link attach 4:OK
 #145/6   netfilter_link_attach/netfilter link attach 5:OK
 #145/7   netfilter_link_attach/netfilter link attach 6:OK
 #145/8   netfilter_link_attach/netfilter link attach 7:OK
 #145/9   netfilter_link_attach/netfilter link attach 8:OK
 #145     netfilter_link_attach:OK

Signed-off-by: Florian Westphal <fw@strlen.de>
---
 .../bpf/prog_tests/netfilter_link_attach.c    | 88 +++++++++++++++++++
 .../bpf/progs/test_netfilter_link_attach.c    | 14 +++
 2 files changed, 102 insertions(+)
 create mode 100644 tools/testing/selftests/bpf/prog_tests/netfilter_link_attach.c
 create mode 100644 tools/testing/selftests/bpf/progs/test_netfilter_link_attach.c

Comments

Daniel Xu June 27, 2023, 2:40 p.m. UTC | #1 
Hi Florian,

On Tue, Jun 27, 2023 at 01:58:39PM +0200, Florian Westphal wrote:
> Call bpf_program__attach_netfilter() with different
> protocol/hook/priority combinations.
> 
> Test fails if supposedly-illegal attachments work
> (e.g., bogus protocol family, illegal priority and so on) or if a
> should-work attachment fails.  Expected output:
> 
>  ./test_progs -t netfilter_link_attach
>  #145/1   netfilter_link_attach/netfilter link attach 0:OK
>  #145/2   netfilter_link_attach/netfilter link attach 1:OK
>  #145/3   netfilter_link_attach/netfilter link attach 2:OK
>  #145/4   netfilter_link_attach/netfilter link attach 3:OK
>  #145/5   netfilter_link_attach/netfilter link attach 4:OK
>  #145/6   netfilter_link_attach/netfilter link attach 5:OK
>  #145/7   netfilter_link_attach/netfilter link attach 6:OK
>  #145/8   netfilter_link_attach/netfilter link attach 7:OK
>  #145/9   netfilter_link_attach/netfilter link attach 8:OK
>  #145     netfilter_link_attach:OK
> 
> Signed-off-by: Florian Westphal <fw@strlen.de>
> ---
>  .../bpf/prog_tests/netfilter_link_attach.c    | 88 +++++++++++++++++++
>  .../bpf/progs/test_netfilter_link_attach.c    | 14 +++
>  2 files changed, 102 insertions(+)
>  create mode 100644 tools/testing/selftests/bpf/prog_tests/netfilter_link_attach.c
>  create mode 100644 tools/testing/selftests/bpf/progs/test_netfilter_link_attach.c
> 
> diff --git a/tools/testing/selftests/bpf/prog_tests/netfilter_link_attach.c b/tools/testing/selftests/bpf/prog_tests/netfilter_link_attach.c
> new file mode 100644
> index 000000000000..dfec6c44f81d
> --- /dev/null
> +++ b/tools/testing/selftests/bpf/prog_tests/netfilter_link_attach.c
> @@ -0,0 +1,88 @@
> +// SPDX-License-Identifier: GPL-2.0-or-later
> +
> +#include <netinet/in.h>
> +#include <linux/netfilter.h>
> +
> +#include "test_progs.h"
> +#include "test_netfilter_link_attach.skel.h"
> +
> +struct nf_link_test {
> +	__u32 pf;
> +	__u32 hooknum;
> +	__s32 priority;
> +	__u32 flags;
> +
> +	bool expect_success;
> +};
> +
> +struct nf_link_test nf_hook_link_tests[] = {
> +	{  },
> +	{ .pf = NFPROTO_NUMPROTO, },
> +	{ .pf = NFPROTO_IPV4, .hooknum = 42, },
> +	{ .pf = NFPROTO_IPV4, .priority = INT_MIN },
> +	{ .pf = NFPROTO_IPV4, .priority = INT_MAX },
> +	{ .pf = NFPROTO_IPV4, .flags = UINT_MAX },
> +
> +	{ .pf = NFPROTO_INET, .priority = 1, },
> +
> +	{ .pf = NFPROTO_IPV4, .priority = -10000, .expect_success = true },
> +	{ .pf = NFPROTO_IPV6, .priority = 10001, .expect_success = true },
> +};
> +
> +void test_netfilter_link_attach(void)
> +{
> +	struct test_netfilter_link_attach *skel;
> +	struct bpf_program *prog;
> +	LIBBPF_OPTS(bpf_netfilter_opts, opts);
> +	int i;
> +
> +	skel = test_netfilter_link_attach__open_and_load();
> +	if (!ASSERT_OK_PTR(skel, "test_netfilter_link_attach__open_and_load"))
> +		goto out;
> +
> +	prog = skel->progs.nf_link_attach_test;
> +	if (!ASSERT_OK_PTR(prog, "attach program"))
> +		goto out;
> +
> +	for (i = 0; i < ARRAY_SIZE(nf_hook_link_tests); i++) {
> +		struct bpf_link *link;
> +		char name[128];
> +
> +		snprintf(name, sizeof(name), "netfilter link attach %u", i);
> +
> +		if (!test__start_subtest(name))
> +			continue;

Nit: naming by index makes it a little annoying to debug failures. How
about adding a name field to struct nf_link_test?

[...]

Thanks,
Daniel
diff mbox series

Patch

diff --git a/tools/testing/selftests/bpf/prog_tests/netfilter_link_attach.c b/tools/testing/selftests/bpf/prog_tests/netfilter_link_attach.c
new file mode 100644
index 000000000000..dfec6c44f81d
--- /dev/null
+++ b/tools/testing/selftests/bpf/prog_tests/netfilter_link_attach.c
@@ -0,0 +1,88 @@ 
+// SPDX-License-Identifier: GPL-2.0-or-later
+
+#include <netinet/in.h>
+#include <linux/netfilter.h>
+
+#include "test_progs.h"
+#include "test_netfilter_link_attach.skel.h"
+
+struct nf_link_test {
+	__u32 pf;
+	__u32 hooknum;
+	__s32 priority;
+	__u32 flags;
+
+	bool expect_success;
+};
+
+struct nf_link_test nf_hook_link_tests[] = {
+	{  },
+	{ .pf = NFPROTO_NUMPROTO, },
+	{ .pf = NFPROTO_IPV4, .hooknum = 42, },
+	{ .pf = NFPROTO_IPV4, .priority = INT_MIN },
+	{ .pf = NFPROTO_IPV4, .priority = INT_MAX },
+	{ .pf = NFPROTO_IPV4, .flags = UINT_MAX },
+
+	{ .pf = NFPROTO_INET, .priority = 1, },
+
+	{ .pf = NFPROTO_IPV4, .priority = -10000, .expect_success = true },
+	{ .pf = NFPROTO_IPV6, .priority = 10001, .expect_success = true },
+};
+
+void test_netfilter_link_attach(void)
+{
+	struct test_netfilter_link_attach *skel;
+	struct bpf_program *prog;
+	LIBBPF_OPTS(bpf_netfilter_opts, opts);
+	int i;
+
+	skel = test_netfilter_link_attach__open_and_load();
+	if (!ASSERT_OK_PTR(skel, "test_netfilter_link_attach__open_and_load"))
+		goto out;
+
+	prog = skel->progs.nf_link_attach_test;
+	if (!ASSERT_OK_PTR(prog, "attach program"))
+		goto out;
+
+	for (i = 0; i < ARRAY_SIZE(nf_hook_link_tests); i++) {
+		struct bpf_link *link;
+		char name[128];
+
+		snprintf(name, sizeof(name), "netfilter link attach %u", i);
+
+		if (!test__start_subtest(name))
+			continue;
+
+#define X(opts, m, i)	opts.m = nf_hook_link_tests[(i)].m
+		X(opts, pf, i);
+		X(opts, hooknum, i);
+		X(opts, priority, i);
+		X(opts, flags, i);
+#undef X
+		link = bpf_program__attach_netfilter(prog, &opts);
+		if (nf_hook_link_tests[i].expect_success) {
+			struct bpf_link *link2;
+
+			if (!ASSERT_OK_PTR(link, "program attach successful"))
+				continue;
+
+			link2 = bpf_program__attach_netfilter(prog, &opts);
+			ASSERT_ERR_PTR(link2, "attach program with same pf/hook/priority");
+
+			if (!ASSERT_OK(bpf_link__destroy(link), "link destroy"))
+				break;
+
+			link2 = bpf_program__attach_netfilter(prog, &opts);
+			if (!ASSERT_OK_PTR(link2, "program reattach successful"))
+				continue;
+			if (!ASSERT_OK(bpf_link__destroy(link2), "link destroy"))
+				break;
+		} else {
+			ASSERT_ERR_PTR(link, "program load failure");
+		}
+	}
+
+out:
+	test_netfilter_link_attach__destroy(skel);
+}
+
diff --git a/tools/testing/selftests/bpf/progs/test_netfilter_link_attach.c b/tools/testing/selftests/bpf/progs/test_netfilter_link_attach.c
new file mode 100644
index 000000000000..03a475160abe
--- /dev/null
+++ b/tools/testing/selftests/bpf/progs/test_netfilter_link_attach.c
@@ -0,0 +1,14 @@ 
+// SPDX-License-Identifier: GPL-2.0-or-later
+
+#include "vmlinux.h"
+#include <bpf/bpf_helpers.h>
+
+#define NF_ACCEPT 1
+
+SEC("netfilter")
+int nf_link_attach_test(struct bpf_nf_ctx *ctx)
+{
+	return NF_ACCEPT;
+}
+
+char _license[] SEC("license") = "GPL";