diff options
Diffstat (limited to 'contrib/netbsd-tests/net/net/t_udp.c')
-rw-r--r-- | contrib/netbsd-tests/net/net/t_udp.c | 110 |
1 files changed, 110 insertions, 0 deletions
diff --git a/contrib/netbsd-tests/net/net/t_udp.c b/contrib/netbsd-tests/net/net/t_udp.c new file mode 100644 index 0000000..db59342 --- /dev/null +++ b/contrib/netbsd-tests/net/net/t_udp.c @@ -0,0 +1,110 @@ +#include <sys/cdefs.h> +__RCSID("$NetBSD: t_udp.c,v 1.2 2013/01/06 02:22:50 christos Exp $"); + +#include <sys/socket.h> +#include <netinet/in.h> + +#include <netdb.h> +#include <string.h> +#include <errno.h> +#include <unistd.h> + +#include <atf-c.h> + +static const char msg[] = "sendto test"; + +static void +sendit(int family) +{ + struct addrinfo hints; + struct addrinfo *res; + int S, s; + int e; + + /* lookup localhost addr, depending on argv[1] */ + memset(&hints, 0, sizeof(hints)); + hints.ai_family = family; + hints.ai_socktype = SOCK_DGRAM; + hints.ai_protocol = IPPROTO_UDP; + hints.ai_flags = 0; + + e = getaddrinfo("localhost", "9999", &hints, &res); + ATF_REQUIRE_MSG(e == 0, "getaddrinfo AF=%d: %s", family, + gai_strerror(e)); + + /* server socket */ + S = socket(res->ai_family, res->ai_socktype, res->ai_protocol); + ATF_REQUIRE_MSG(S >= 0, "server-socket AF=%d: %s", family, + strerror(errno)); + + e = bind(S, res->ai_addr, res->ai_addrlen); + ATF_REQUIRE_MSG(e == 0, "bind AF=%d: %s", family, + strerror(errno)); + + /* client socket */ + s = socket(res->ai_family, res->ai_socktype, res->ai_protocol); + ATF_REQUIRE_MSG(s >= 0, "client-socket AF=%d: %s", family, + strerror(errno)); + + /* sendto */ + e = sendto(s, msg, sizeof(msg), 0, res->ai_addr, res->ai_addrlen); + ATF_REQUIRE_MSG(e == sizeof(msg), "sendto(1) AF=%d: %s", family, + strerror(errno)); + + e = sendto(s, msg, sizeof(msg), 0, res->ai_addr, res->ai_addrlen); + ATF_REQUIRE_MSG(e == sizeof(msg), "sendto(2) AF=%d: %s", family, + strerror(errno)); + + /* connect + send */ + e = connect(s, res->ai_addr, res->ai_addrlen); + ATF_REQUIRE_MSG(e == 0, "connect(1) AF=%d: %s", family, + strerror(errno)); + + e = send(s, msg, sizeof(msg), 0); + ATF_REQUIRE_MSG(e == sizeof(msg), "send(1) AF=%d: %s", family, + strerror(errno)); + + e = connect(s, res->ai_addr, res->ai_addrlen); + ATF_REQUIRE_MSG(e == 0, "connect(2) AF=%d: %s", family, + strerror(errno)); + + e = send(s, msg, sizeof(msg), 0); + ATF_REQUIRE_MSG(e == sizeof(msg), "send(2) AF=%d: %s", family, + strerror(errno)); + + close(s); +} + +ATF_TC(udp4_send); +ATF_TC_HEAD(udp4_send, tc) +{ + + atf_tc_set_md_var(tc, "descr", "Check that inet4 udp send works both" + " for connected and unconnected sockets"); +} + +ATF_TC_BODY(udp4_send, tc) +{ + sendit(AF_INET); +} + +ATF_TC(udp6_send); +ATF_TC_HEAD(udp6_send, tc) +{ + + atf_tc_set_md_var(tc, "descr", "Check that inet6 udp send works both" + " for connected and unconnected sockets"); +} + +ATF_TC_BODY(udp6_send, tc) +{ + sendit(AF_INET6); +} + +ATF_TP_ADD_TCS(tp) +{ + + ATF_TP_ADD_TC(tp, udp4_send); + ATF_TP_ADD_TC(tp, udp6_send); + return atf_no_error(); +} |