From 436fdc82bdc43c2d4af6b80d65479b59b554b741 Mon Sep 17 00:00:00 2001
From: christos <christos@NetBSD.org>
Date: Sun, 6 Jan 2013 00:35:22 +0000
Subject: [PATCH] new udp test

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

diff --git a/tests/net/net/Makefile b/tests/net/net/Makefile
index 329ca97b5a80..e4124fcd987d 100644
--- a/tests/net/net/Makefile
+++ b/tests/net/net/Makefile
@@ -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 @@ TESTSDIR=	${TESTSBASE}/net/net
 
 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 --git a/tests/net/net/t_udp.c b/tests/net/net/t_udp.c
new file mode 100644
index 000000000000..7d5199824e22
--- /dev/null
+++ b/tests/net/net/t_udp.c
@@ -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();
+}