bpf: Add test for SO_REUSEPORT_DETACH_BPF

This patch adds a test for the new sockopt SO_REUSEPORT_DETACH_BPF.

Signed-off-by: Martin KaFai Lau <kafai@fb.com>
Reviewed-by: Stanislav Fomichev <sdf@google.com>
Signed-off-by: Daniel Borkmann <daniel@iogearbox.net>

1 files changed, 54 insertions, 0 deletions

diff --git a/tools/testing/selftests/bpf/test_select_reuseport.c b/tools/testing/selftests/bpf/test_select_reuseport.c
index 75646d9b34aa..7566c13eb51a 100644
--- a/tools/testing/selftests/bpf/test_select_reuseport.c
+++ b/tools/testing/selftests/bpf/test_select_reuseport.c
@@ -523,6 +523,58 @@ static void test_pass_on_err(int type, sa_family_t family)
         printf("OK\n");
 }
 
+static void test_detach_bpf(int type, sa_family_t family)
+{
+#ifdef SO_DETACH_REUSEPORT_BPF
+        __u32 nr_run_before = 0, nr_run_after = 0, tmp, i;
+        struct epoll_event ev;
+        int cli_fd, err, nev;
+        struct cmd cmd = {};
+        int optvalue = 0;
+
+        printf("%s: ", __func__);
+        err = setsockopt(sk_fds[0], SOL_SOCKET, SO_DETACH_REUSEPORT_BPF,
+                         &optvalue, sizeof(optvalue));
+        CHECK(err == -1, "setsockopt(SO_DETACH_REUSEPORT_BPF)",
+              "err:%d errno:%d\n", err, errno);
+
+        err = setsockopt(sk_fds[1], SOL_SOCKET, SO_DETACH_REUSEPORT_BPF,
+                         &optvalue, sizeof(optvalue));
+        CHECK(err == 0 || errno != ENOENT, "setsockopt(SO_DETACH_REUSEPORT_BPF)",
+              "err:%d errno:%d\n", err, errno);
+
+        for (i = 0; i < NR_RESULTS; i++) {
+                err = bpf_map_lookup_elem(result_map, &i, &tmp);
+                CHECK(err == -1, "lookup_elem(result_map)",
+                      "i:%u err:%d errno:%d\n", i, err, errno);
+                nr_run_before += tmp;
+        }
+
+        cli_fd = send_data(type, family, &cmd, sizeof(cmd), PASS);
+        nev = epoll_wait(epfd, &ev, 1, 5);
+        CHECK(nev <= 0, "nev <= 0",
+              "nev:%d expected:1 type:%d family:%d data:(0, 0)\n",
+              nev,  type, family);
+
+        for (i = 0; i < NR_RESULTS; i++) {
+                err = bpf_map_lookup_elem(result_map, &i, &tmp);
+                CHECK(err == -1, "lookup_elem(result_map)",
+                      "i:%u err:%d errno:%d\n", i, err, errno);
+                nr_run_after += tmp;
+        }
+
+        CHECK(nr_run_before != nr_run_after,
+              "nr_run_before != nr_run_after",
+              "nr_run_before:%u nr_run_after:%u\n",
+              nr_run_before, nr_run_after);
+
+        printf("OK\n");
+        close(cli_fd);
+#else
+        printf("%s: SKIP\n", __func__);
+#endif
+}
+
 static void prepare_sk_fds(int type, sa_family_t family, bool inany)
 {
         const int first = REUSEPORT_ARRAY_SIZE - 1;
@@ -664,6 +716,8 @@ static void test_all(void)
                         test_pass(type, family);
                         test_syncookie(type, family);
                         test_pass_on_err(type, family);
+                        /* Must be the last test */
+                        test_detach_bpf(type, family);
 
                         cleanup_per_test();
                         printf("\n");