1// SPDX-License-Identifier: GPL-2.0
  2/* Copyright Amazon.com Inc. or its affiliates. */
  3#include <sys/socket.h>
  4#include <sys/un.h>
  5#include <test_progs.h>
  6#include "bpf_iter_setsockopt_unix.skel.h"
  7
  8#define NR_CASES 5
  9
 10static int create_unix_socket(struct bpf_iter_setsockopt_unix *skel)
 11{
 12	struct sockaddr_un addr = {
 13		.sun_family = AF_UNIX,
 14		.sun_path = "",
 15	};
 16	socklen_t len;
 17	int fd, err;
 18
 19	fd = socket(AF_UNIX, SOCK_STREAM, 0);
 20	if (!ASSERT_NEQ(fd, -1, "socket"))
 21		return -1;
 22
 23	len = offsetof(struct sockaddr_un, sun_path);
 24	err = bind(fd, (struct sockaddr *)&addr, len);
 25	if (!ASSERT_OK(err, "bind"))
 26		return -1;
 27
 28	len = sizeof(addr);
 29	err = getsockname(fd, (struct sockaddr *)&addr, &len);
 30	if (!ASSERT_OK(err, "getsockname"))
 31		return -1;
 32
 33	memcpy(&skel->bss->sun_path, &addr.sun_path,
 34	       len - offsetof(struct sockaddr_un, sun_path));
 35
 36	return fd;
 37}
 38
 39static void test_sndbuf(struct bpf_iter_setsockopt_unix *skel, int fd)
 40{
 41	socklen_t optlen;
 42	int i, err;
 43
 44	for (i = 0; i < NR_CASES; i++) {
 45		if (!ASSERT_NEQ(skel->data->sndbuf_getsockopt[i], -1,
 46				"bpf_(get|set)sockopt"))
 47			return;
 48
 49		err = setsockopt(fd, SOL_SOCKET, SO_SNDBUF,
 50				 &(skel->data->sndbuf_setsockopt[i]),
 51				 sizeof(skel->data->sndbuf_setsockopt[i]));
 52		if (!ASSERT_OK(err, "setsockopt"))
 53			return;
 54
 55		optlen = sizeof(skel->bss->sndbuf_getsockopt_expected[i]);
 56		err = getsockopt(fd, SOL_SOCKET, SO_SNDBUF,
 57				 &(skel->bss->sndbuf_getsockopt_expected[i]),
 58				 &optlen);
 59		if (!ASSERT_OK(err, "getsockopt"))
 60			return;
 61
 62		if (!ASSERT_EQ(skel->data->sndbuf_getsockopt[i],
 63			       skel->bss->sndbuf_getsockopt_expected[i],
 64			       "bpf_(get|set)sockopt"))
 65			return;
 66	}
 67}
 68
 69void test_bpf_iter_setsockopt_unix(void)
 70{
 71	struct bpf_iter_setsockopt_unix *skel;
 72	int err, unix_fd, iter_fd;
 73	char buf;
 74
 75	skel = bpf_iter_setsockopt_unix__open_and_load();
 76	if (!ASSERT_OK_PTR(skel, "open_and_load"))
 77		return;
 78
 79	unix_fd = create_unix_socket(skel);
 80	if (!ASSERT_NEQ(unix_fd, -1, "create_unix_server"))
 81		goto destroy;
 82
 83	skel->links.change_sndbuf = bpf_program__attach_iter(skel->progs.change_sndbuf, NULL);
 84	if (!ASSERT_OK_PTR(skel->links.change_sndbuf, "bpf_program__attach_iter"))
 85		goto destroy;
 86
 87	iter_fd = bpf_iter_create(bpf_link__fd(skel->links.change_sndbuf));
 88	if (!ASSERT_GE(iter_fd, 0, "bpf_iter_create"))
 89		goto destroy;
 90
 91	while ((err = read(iter_fd, &buf, sizeof(buf))) == -1 &&
 92	       errno == EAGAIN)
 93		;
 94	if (!ASSERT_OK(err, "read iter error"))
 95		goto destroy;
 96
 97	test_sndbuf(skel, unix_fd);
 98destroy:
 99	bpf_iter_setsockopt_unix__destroy(skel);
100}