[src/trunk]: src/tests/net/net new udp test

[christos <christos%NetBSD.org@localhost>]

details:   https://anonhg.NetBSD.org/src/rev/499910a7523d
branches:  trunk
changeset: 783727:499910a7523d
user:      christos <christos%NetBSD.org@localhost>
date:      Sun Jan 06 00:35:22 2013 +0000

description:
new udp test

diffstat:

 tests/net/net/Makefile |    3 +-
 tests/net/net/t_udp.c  |  110 +++++++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 112 insertions(+), 1 deletions(-)

diffs (131 lines):

diff -r 6c76d55ac730 -r 499910a7523d tests/net/net/Makefile
--- a/tests/net/net/Makefile    Sun Jan 06 00:19:13 2013 +0000
+++ b/tests/net/net/Makefile    Sun Jan 06 00:35:22 2013 +0000
@@ -1,4 +1,4 @@
-# $NetBSD: Makefile,v 1.3 2011/09/28 23:11:12 christos Exp $
+# $NetBSD: Makefile,v 1.4 2013/01/06 00:35:22 christos Exp $
 #
 
 .include <bsd.own.mk>
@@ -7,6 +7,7 @@
 
 TESTS_C=       t_raw
 TESTS_C+=      t_unix
+TESTS_C+=      t_udp
 
 LDADD.t_raw+=          -lrumpnet_local -lrumpnet_netinet -lrumpnet_net
 LDADD.t_raw+=          -lrumpnet -lrumpvfs -lrump -lrumpuser -lpthread
diff -r 6c76d55ac730 -r 499910a7523d tests/net/net/t_udp.c
--- /dev/null   Thu Jan 01 00:00:00 1970 +0000
+++ b/tests/net/net/t_udp.c     Sun Jan 06 00:35:22 2013 +0000
@@ -0,0 +1,110 @@
+#include <sys/cdefs.h>
+__RCSID("$NetBSD: t_udp.c,v 1.1 2013/01/06 00:35:22 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(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));
+
+       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();
+}