1
0
mirror of https://git.FreeBSD.org/src.git synced 2026-06-02 11:24:32 +00:00

tests/netinet: add few tests for unconnected UDP socket

1) Test that getsockname() on a socket that experienced a sendto(2)
   returns allocated port.
2) Same as 1), but jailed.  Make sure IP address was also selected.
3) Test that IP_SENDSRCADDR/IP_RECVDSTADDR work.
4) Regression test for bug 274009.
This commit is contained in:
Gleb Smirnoff
2025-02-17 15:28:51 -08:00
parent 1f1963bd32
commit 4f274f849b
2 changed files with 251 additions and 0 deletions
+2
View File
@@ -13,6 +13,7 @@ ATF_TESTS_C= fibs_multibind_test \
tcp_connect_port_test \
tcp_implied_connect \
tcp_md5_getsockopt \
udp_bindings \
udp_io
ATF_TESTS_SH= arp \
@@ -29,6 +30,7 @@ ATF_TESTS_PYTEST+= carp.py
ATF_TESTS_PYTEST+= igmp.py
LIBADD.so_reuseport_lb_test= pthread
LIBADD.udp_bindings= pthread
# Some of the arp tests look for log messages in the dmesg buffer, so run them
# serially to avoid problems with interleaved output.
+249
View File
@@ -0,0 +1,249 @@
/*-
* SPDX-License-Identifier: BSD-2-Clause
*
* Copyright (c) 2025 Gleb Smirnoff <glebius@FreeBSD.org>
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions
* are met:
* 1. Redistributions of source code must retain the above copyright
* notice, this list of conditions and the following disclaimer.
* 2. Redistributions in binary form must reproduce the above copyright
* notice, this list of conditions and the following disclaimer in the
* documentation and/or other materials provided with the distribution.
*
* THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
* ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
* ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
* FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
* DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
* OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
* HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
* LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
* OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
* SUCH DAMAGE.
*/
#include <sys/param.h>
#include <sys/jail.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <errno.h>
#include <ifaddrs.h>
#include <pthread.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <atf-c.h>
static const char buf[] = "Hello";
static void
sendtolocalhost(int s)
{
struct sockaddr_in dst = {
.sin_family = AF_INET,
.sin_len = sizeof(struct sockaddr_in),
.sin_addr = { htonl(INADDR_LOOPBACK) },
.sin_port = htons(1638),
};
ATF_REQUIRE(sendto(s, buf, sizeof(buf), 0, (struct sockaddr *)&dst,
sizeof(dst)) == sizeof(buf));
}
/*
* Echo back to the sender its own address in payload.
*/
static void *
echo(void *arg)
{
int s = *(int *)arg;
struct sockaddr_in sin;
socklen_t slen = sizeof(sin);
char rbuf[sizeof(buf)];
ATF_REQUIRE(recvfrom(s, &rbuf, sizeof(rbuf), 0, (struct sockaddr *)&sin,
&slen) == sizeof(rbuf));
printf("Echo to %s:%u\n", inet_ntoa(sin.sin_addr), ntohs(sin.sin_port));
ATF_REQUIRE(sendto(s, &sin, sizeof(sin), 0, (struct sockaddr *)&sin,
sizeof(sin)) == sizeof(sin));
return (NULL);
}
/*
* Cycle through local addresses (normally there should be at least two
* different IPv4 ones), and communicate to the echo server checking both
* IP_SENDSRCADDR and IP_RECVDSTADDR. Use same cmsg buffer for both send
* and receive operation, this is a suggested in manual, given that
* IP_RECVDSTADDR == IP_SENDSRCADDR.
* At the setup phase check that IP_SENDSRCADDR doesn't work on unbound socket.
*/
ATF_TC_WITHOUT_HEAD(IP_SENDSRCADDR);
ATF_TC_BODY(IP_SENDSRCADDR, tc)
{
struct sockaddr_in srv = {
.sin_family = AF_INET,
.sin_len = sizeof(struct sockaddr_in),
}, dst;
char cbuf[CMSG_SPACE(sizeof(struct in_addr))];
struct iovec iov = {
.iov_base = __DECONST(char *, buf),
.iov_len = sizeof(buf),
};
struct iovec riov = {
.iov_base = &dst,
.iov_len = sizeof(dst),
};
struct msghdr msg = {
.msg_iov = &iov,
.msg_iovlen = 1,
.msg_name = &srv,
.msg_namelen = sizeof(srv),
.msg_control = cbuf,
.msg_controllen = sizeof(cbuf),
};
struct msghdr rmsg = {
.msg_iov = &riov,
.msg_iovlen = 1,
.msg_control = cbuf,
.msg_controllen = sizeof(cbuf),
};
struct cmsghdr *cmsg = CMSG_FIRSTHDR(&msg);
*cmsg = (struct cmsghdr) {
.cmsg_level = IPPROTO_IP,
.cmsg_type = IP_SENDSRCADDR,
.cmsg_len = CMSG_LEN(sizeof(struct in_addr)),
};
socklen_t slen = sizeof(struct sockaddr_in);
struct ifaddrs *ifa0, *ifa;
pthread_t tid;
int s, e;
/* First check that IP_SENDSRCADDR doesn't work on an unbound socket. */
ATF_REQUIRE((s = socket(PF_INET, SOCK_DGRAM, 0)) > 0);
ATF_REQUIRE_MSG(sendmsg(s, &msg, 0) == -1 && errno == EINVAL,
"sendmsg(.cmsg_type = IP_SENDSRCADDR), errno %d", errno);
/* Bind to random ports both sender and echo server. */
ATF_REQUIRE(bind(s, (struct sockaddr *)&srv, sizeof(srv)) == 0);
ATF_REQUIRE((e = socket(PF_INET, SOCK_DGRAM, 0)) > 0);
ATF_REQUIRE(bind(e, (struct sockaddr *)&srv, sizeof(srv)) == 0);
ATF_REQUIRE(getsockname(e, (struct sockaddr *)&srv, &slen) == 0);
srv.sin_addr.s_addr = htonl(INADDR_LOOPBACK);
ATF_REQUIRE(getifaddrs(&ifa0) == 0);
for (ifa = ifa0; ifa != NULL; ifa = ifa->ifa_next) {
struct sockaddr_in src;
struct in_addr vrf;
if (ifa->ifa_addr->sa_family != AF_INET)
continue;
memcpy(&src, ifa->ifa_addr, sizeof(src));
printf("Sending from %s\n", inet_ntoa(src.sin_addr));
ATF_REQUIRE(pthread_create(&tid, NULL, echo, &e) == 0);
memcpy(CMSG_DATA(cmsg), &src.sin_addr, sizeof(src.sin_addr));
ATF_REQUIRE(sendmsg(s, &msg, 0) == sizeof(buf));
ATF_REQUIRE(recvmsg(s, &rmsg, 0) == sizeof(struct sockaddr_in));
memcpy(&vrf, CMSG_DATA(cmsg), sizeof(vrf));
ATF_REQUIRE_MSG(dst.sin_addr.s_addr == src.sin_addr.s_addr,
"Sent from %s, but echo server reports %s",
inet_ntoa(src.sin_addr), inet_ntoa(dst.sin_addr));
ATF_REQUIRE_MSG(vrf.s_addr == src.sin_addr.s_addr,
"Sent from %s, but IP_RECVDSTADDR reports %s",
inet_ntoa(src.sin_addr), inet_ntoa(vrf));
ATF_REQUIRE(pthread_join(tid, NULL) == 0);
}
freeifaddrs(ifa0);
close(s);
close(e);
}
/*
* Check gethostname(2) on a newborn socket, and then on an unconnected, but
* used socket. The first shall return all-zeroes, and second one should
* return us our assigned port.
*/
ATF_TC_WITHOUT_HEAD(gethostname);
ATF_TC_BODY(gethostname, tc)
{
struct sockaddr_in sin;
socklen_t slen = sizeof(sin);
int s;
ATF_REQUIRE((s = socket(PF_INET, SOCK_DGRAM, 0)) > 0);
ATF_REQUIRE(getsockname(s, (struct sockaddr *)&sin, &slen) == 0);
ATF_REQUIRE_MSG(sin.sin_addr.s_addr == INADDR_ANY && sin.sin_port == 0,
"newborn socket name %s:%u", inet_ntoa(sin.sin_addr),
ntohs(sin.sin_port));
sendtolocalhost(s);
ATF_REQUIRE(getsockname(s, (struct sockaddr *)&sin, &slen) == 0);
ATF_REQUIRE_MSG(sin.sin_addr.s_addr == INADDR_ANY && sin.sin_port != 0,
"used unconnected socket name %s:%u", inet_ntoa(sin.sin_addr),
ntohs(sin.sin_port));
close(s);
}
ATF_TC_WITHOUT_HEAD(gethostname_jailed);
ATF_TC_BODY(gethostname_jailed, tc)
{
struct in_addr laddr = { htonl(INADDR_LOOPBACK) };
struct jail jconf = {
.version = JAIL_API_VERSION,
.path = __DECONST(char *, "/"),
.hostname = __DECONST(char *,"test"),
.ip4s = 1,
.ip4 = &laddr,
};
struct sockaddr_in sin;
socklen_t slen = sizeof(sin);
int s;
ATF_REQUIRE(jail(&jconf) > 0);
ATF_REQUIRE((s = socket(PF_INET, SOCK_DGRAM, 0)) > 0);
sendtolocalhost(s);
ATF_REQUIRE(getsockname(s, (struct sockaddr *)&sin, &slen) == 0);
ATF_REQUIRE_MSG(sin.sin_addr.s_addr == laddr.s_addr &&
sin.sin_port != 0,
"jailed unconnected socket name %s:%u", inet_ntoa(sin.sin_addr),
ntohs(sin.sin_port));
close(s);
}
/*
* See bug 274009.
*/
ATF_TC_WITHOUT_HEAD(v4mapped);
ATF_TC_BODY(v4mapped, tc)
{
struct sockaddr_in6 sa6 = {
.sin6_family = AF_INET6,
.sin6_len = sizeof(struct sockaddr_in6),
.sin6_port = htons(1),
};
int s;
ATF_REQUIRE((s = socket(PF_INET6, SOCK_DGRAM, 0)) > 0);
ATF_REQUIRE(setsockopt(s, IPPROTO_IPV6, IPV6_V6ONLY, &(int){0},
sizeof(int)) == 0);
ATF_REQUIRE(inet_pton(AF_INET6, "::ffff:127.0.0.1", &(sa6.sin6_addr))
== 1);
ATF_REQUIRE(sendto(s, buf, sizeof(buf), 0, (struct sockaddr *)&sa6,
sizeof(sa6)) == sizeof(buf));
close(s);
}
ATF_TP_ADD_TCS(tp)
{
ATF_TP_ADD_TC(tp, v4mapped);
ATF_TP_ADD_TC(tp, gethostname);
ATF_TP_ADD_TC(tp, gethostname_jailed);
ATF_TP_ADD_TC(tp, IP_SENDSRCADDR);
return (atf_no_error());
}