ng_ksocket: Add tests for IPv6lf/dev/D50111

Add udp6_connect and udp6_bind for ng_ksocket

Differential Revision:	https://reviews.freebsd.org/D50111

1 files changed, 62 insertions, 0 deletions

diff --git a/tests/sys/netgraph/ksocket.c b/tests/sys/netgraph/ksocket.c
index e97b9b3f0691..b1c48a7bcf53 100644
--- a/tests/sys/netgraph/ksocket.c
+++ b/tests/sys/netgraph/ksocket.c
@@ -106,10 +106,72 @@ ATF_TC_BODY(udp_bind, tc)
 	hellocheck(us, ds);
 }
 
+ATF_TC_WITHOUT_HEAD(udp6_connect);
+ATF_TC_BODY(udp6_connect, tc)
+{
+	struct sockaddr_in6 sa = {
+		.sin6_family = AF_INET6,
+	};
+	socklen_t slen = sizeof(sa);
+	int ds, cs, us;
+
+	ATF_REQUIRE((us = socket(PF_INET6, SOCK_DGRAM, 0)) > 0);
+	ATF_REQUIRE(bind(us, (struct sockaddr *)&sa, sizeof(sa)) == 0);
+	ATF_REQUIRE(getsockname(us, (struct sockaddr *)&sa, &slen) == 0);
+
+	struct ngm_mkpeer mkp = {
+		.type = NG_KSOCKET_NODE_TYPE,
+		.ourhook = OURHOOK,
+		.peerhook = "inet6/dgram/udp6",
+	};
+	ATF_REQUIRE(NgMkSockNode(NULL, &cs, &ds) == 0);
+	ATF_REQUIRE(NgSendMsg(cs, ".", NGM_GENERIC_COOKIE, NGM_MKPEER, &mkp,
+	    sizeof(mkp)) >= 0);
+	ATF_REQUIRE(NgSendMsg(cs, ".:" OURHOOK, NGM_KSOCKET_COOKIE,
+	    NGM_KSOCKET_CONNECT, &sa, sizeof(sa)) >= 0);
+
+	hellocheck(ds, us);
+}
+
+
+ATF_TC_WITHOUT_HEAD(udp6_bind);
+ATF_TC_BODY(udp6_bind, tc)
+{
+	struct sockaddr_in6 sin6 = {
+		.sin6_family = AF_INET6,
+		.sin6_len = sizeof(sin6),
+	};
+	struct ng_mesg *rep;
+	int ds, cs, us;
+
+	struct ngm_mkpeer mkp = {
+		.type = NG_KSOCKET_NODE_TYPE,
+		.ourhook = OURHOOK,
+		.peerhook = "inet6/dgram/udp6",
+	};
+	ATF_REQUIRE(NgMkSockNode(NULL, &cs, &ds) == 0);
+	ATF_REQUIRE(NgSendMsg(cs, ".", NGM_GENERIC_COOKIE, NGM_MKPEER, &mkp,
+	    sizeof(mkp)) >= 0);
+	ATF_REQUIRE(NgSendMsg(cs, ".:" OURHOOK, NGM_KSOCKET_COOKIE,
+	    NGM_KSOCKET_BIND, &sin6, sizeof(sin6)) >= 0);
+	ATF_REQUIRE(NgSendMsg(cs, ".:" OURHOOK, NGM_KSOCKET_COOKIE,
+	    NGM_KSOCKET_GETNAME, NULL, 0) >= 0);
+	ATF_REQUIRE(NgAllocRecvMsg(cs, &rep, NULL) == sizeof(struct ng_mesg) +
+	    sizeof(struct sockaddr_in6));
+
+	ATF_REQUIRE((us = socket(PF_INET6, SOCK_DGRAM, 0)) > 0);
+	ATF_REQUIRE(connect(us, (struct sockaddr *)rep->data,
+			sizeof(struct sockaddr_in6)) == 0);
+
+	hellocheck(us, ds);
+}
+
 ATF_TP_ADD_TCS(tp)
 {
 	ATF_TP_ADD_TC(tp, udp_connect);
 	ATF_TP_ADD_TC(tp, udp_bind);
+	ATF_TP_ADD_TC(tp, udp6_connect);
+	ATF_TP_ADD_TC(tp, udp6_bind);
 
 	return (atf_no_error());
 }