126 lines
2.9 KiB
C
126 lines
2.9 KiB
C
// SPDX-License-Identifier: GPL-2.0
|
|
/* Copyright (c) Meta Platforms, Inc. and affiliates. */
|
|
|
|
#define _GNU_SOURCE
|
|
#include <sched.h>
|
|
#include <linux/socket.h>
|
|
#include <net/if.h>
|
|
|
|
#include "test_progs.h"
|
|
#include "cgroup_helpers.h"
|
|
#include "network_helpers.h"
|
|
|
|
#include "setget_sockopt.skel.h"
|
|
|
|
#define CG_NAME "/setget-sockopt-test"
|
|
|
|
static const char addr4_str[] = "127.0.0.1";
|
|
static const char addr6_str[] = "::1";
|
|
static struct setget_sockopt *skel;
|
|
static int cg_fd;
|
|
|
|
static int create_netns(void)
|
|
{
|
|
if (!ASSERT_OK(unshare(CLONE_NEWNET), "create netns"))
|
|
return -1;
|
|
|
|
if (!ASSERT_OK(system("ip link set dev lo up"), "set lo up"))
|
|
return -1;
|
|
|
|
if (!ASSERT_OK(system("ip link add dev binddevtest1 type veth peer name binddevtest2"),
|
|
"add veth"))
|
|
return -1;
|
|
|
|
if (!ASSERT_OK(system("ip link set dev binddevtest1 up"),
|
|
"bring veth up"))
|
|
return -1;
|
|
|
|
return 0;
|
|
}
|
|
|
|
static void test_tcp(int family)
|
|
{
|
|
struct setget_sockopt__bss *bss = skel->bss;
|
|
int sfd, cfd;
|
|
|
|
memset(bss, 0, sizeof(*bss));
|
|
|
|
sfd = start_server(family, SOCK_STREAM,
|
|
family == AF_INET6 ? addr6_str : addr4_str, 0, 0);
|
|
if (!ASSERT_GE(sfd, 0, "start_server"))
|
|
return;
|
|
|
|
cfd = connect_to_fd(sfd, 0);
|
|
if (!ASSERT_GE(cfd, 0, "connect_to_fd_server")) {
|
|
close(sfd);
|
|
return;
|
|
}
|
|
close(sfd);
|
|
close(cfd);
|
|
|
|
ASSERT_EQ(bss->nr_listen, 1, "nr_listen");
|
|
ASSERT_EQ(bss->nr_connect, 1, "nr_connect");
|
|
ASSERT_EQ(bss->nr_active, 1, "nr_active");
|
|
ASSERT_EQ(bss->nr_passive, 1, "nr_passive");
|
|
ASSERT_EQ(bss->nr_socket_post_create, 2, "nr_socket_post_create");
|
|
ASSERT_EQ(bss->nr_binddev, 2, "nr_bind");
|
|
}
|
|
|
|
static void test_udp(int family)
|
|
{
|
|
struct setget_sockopt__bss *bss = skel->bss;
|
|
int sfd;
|
|
|
|
memset(bss, 0, sizeof(*bss));
|
|
|
|
sfd = start_server(family, SOCK_DGRAM,
|
|
family == AF_INET6 ? addr6_str : addr4_str, 0, 0);
|
|
if (!ASSERT_GE(sfd, 0, "start_server"))
|
|
return;
|
|
close(sfd);
|
|
|
|
ASSERT_GE(bss->nr_socket_post_create, 1, "nr_socket_post_create");
|
|
ASSERT_EQ(bss->nr_binddev, 1, "nr_bind");
|
|
}
|
|
|
|
void test_setget_sockopt(void)
|
|
{
|
|
cg_fd = test__join_cgroup(CG_NAME);
|
|
if (cg_fd < 0)
|
|
return;
|
|
|
|
if (create_netns())
|
|
goto done;
|
|
|
|
skel = setget_sockopt__open();
|
|
if (!ASSERT_OK_PTR(skel, "open skel"))
|
|
goto done;
|
|
|
|
strcpy(skel->rodata->veth, "binddevtest1");
|
|
skel->rodata->veth_ifindex = if_nametoindex("binddevtest1");
|
|
if (!ASSERT_GT(skel->rodata->veth_ifindex, 0, "if_nametoindex"))
|
|
goto done;
|
|
|
|
if (!ASSERT_OK(setget_sockopt__load(skel), "load skel"))
|
|
goto done;
|
|
|
|
skel->links.skops_sockopt =
|
|
bpf_program__attach_cgroup(skel->progs.skops_sockopt, cg_fd);
|
|
if (!ASSERT_OK_PTR(skel->links.skops_sockopt, "attach cgroup"))
|
|
goto done;
|
|
|
|
skel->links.socket_post_create =
|
|
bpf_program__attach_cgroup(skel->progs.socket_post_create, cg_fd);
|
|
if (!ASSERT_OK_PTR(skel->links.socket_post_create, "attach_cgroup"))
|
|
goto done;
|
|
|
|
test_tcp(AF_INET6);
|
|
test_tcp(AF_INET);
|
|
test_udp(AF_INET6);
|
|
test_udp(AF_INET);
|
|
|
|
done:
|
|
setget_sockopt__destroy(skel);
|
|
close(cg_fd);
|
|
}
|