[CRIU] [PATCH 8/8] zdtm: add a test case for netlink sockets

[Andrey Vagin]

* create two bound socket (for unicase and broadcase messages),
  a connected socket, an unbound and unconnected socket.

Signed-off-by: Andrey Vagin <avagin at openvz.org>
---
 test/zdtm.sh                       |   1 +
 test/zdtm/live/static/Makefile     |   1 +
 test/zdtm/live/static/sk-netlink.c | 160 +++++++++++++++++++++++++++++++++++++
 3 files changed, 162 insertions(+)
 create mode 100644 test/zdtm/live/static/sk-netlink.c

diff --git a/test/zdtm.sh b/test/zdtm.sh
index a02aaf8..4163282 100755
--- a/test/zdtm.sh
+++ b/test/zdtm.sh
@@ -84,6 +84,7 @@ static/sse20
 static/fdt_shared
 static/file_locks00
 static/file_locks01
+static/sk-netlink
 "
 # Duplicate list with ns/ prefix
 TEST_LIST=$TEST_LIST$(echo $TEST_LIST | tr ' ' '\n' | sed 's#^#ns/#')
diff --git a/test/zdtm/live/static/Makefile b/test/zdtm/live/static/Makefile
index 398d2a9..76b6c7c 100644
--- a/test/zdtm/live/static/Makefile
+++ b/test/zdtm/live/static/Makefile
@@ -89,6 +89,7 @@ TST_NOFILE	=				\
 		cow00				\
 		child_opened_proc		\
 		posix_timers			\
+		sk-netlink			\
 #		jobctl00			\
 
 TST_FILE	=				\
diff --git a/test/zdtm/live/static/sk-netlink.c b/test/zdtm/live/static/sk-netlink.c
new file mode 100644
index 0000000..7b384ad
--- /dev/null
+++ b/test/zdtm/live/static/sk-netlink.c
@@ -0,0 +1,160 @@
+#include <unistd.h>
+#include <linux/netlink.h>
+#include <sys/socket.h>
+#include <linux/socket.h>
+#include <string.h>
+
+#include "zdtmtst.h"
+
+#ifndef SOL_NETLINK
+#define SOL_NETLINK     270
+#endif
+
+#define	UDEV_MONITOR_TEST 32
+
+const char *test_doc	= "Support of netlink sockets";
+const char *test_author	= "Andrew Vagin <avagin at parallels.com>";
+
+int main(int argc, char ** argv)
+{
+	int ssk, bsk, csk, dsk;
+	struct sockaddr_nl addr;
+	struct msghdr msg;
+	struct {
+		struct nlmsghdr hdr;
+	} req;
+	struct iovec iov;
+	char buf[4096];
+
+	test_init(argc, argv);
+
+	ssk = socket(PF_NETLINK, SOCK_RAW, NETLINK_KOBJECT_UEVENT);
+	if (ssk < 0) {
+		err("Can't create sock diag socket");
+		return -1;
+	}
+	bsk = socket(PF_NETLINK, SOCK_RAW, NETLINK_KOBJECT_UEVENT);
+	if (bsk < 0) {
+		err("Can't create sock diag socket");
+		return -1;
+	}
+#if 0
+	int on, bbsk;
+
+	bbsk = socket(PF_NETLINK, SOCK_RAW, NETLINK_KOBJECT_UEVENT);
+	if (bbsk < 0) {
+		err("Can't create sock diag socket");
+		return -1;
+	}
+
+	on = UDEV_MONITOR_TEST;
+	setsockopt(bbsk, SOL_NETLINK, NETLINK_ADD_MEMBERSHIP, &on, sizeof(on));
+#endif
+	csk = socket(PF_NETLINK, SOCK_RAW, NETLINK_KOBJECT_UEVENT);
+	if (csk < 0) {
+		err("Can't create sock diag socket");
+		return -1;
+	}
+	dsk = socket(PF_NETLINK, SOCK_RAW, NETLINK_KOBJECT_UEVENT);
+	if (dsk < 0) {
+		err("Can't create sock diag socket");
+		return -1;
+	}
+
+	addr.nl_family = AF_NETLINK;
+	addr.nl_groups = 0;
+	addr.nl_pid = getpid();
+	if (bind(ssk, (struct sockaddr *) &addr, sizeof(struct sockaddr_nl))) {
+		err("bind");
+		return 1;
+	}
+
+	addr.nl_groups = 1 << (UDEV_MONITOR_TEST - 1);
+	addr.nl_pid = 0;
+	if (bind(bsk, (struct sockaddr *) &addr, sizeof(struct sockaddr_nl))) {
+		err("bind");
+		return 1;
+	}
+
+	addr.nl_pid = getpid();;
+	addr.nl_groups = 1 << (UDEV_MONITOR_TEST - 1);
+	if (connect(csk, (struct sockaddr *) &addr, sizeof(struct sockaddr_nl))) {
+		err("connect");
+		return 1;
+	}
+
+	test_daemon();
+
+	test_waitsig();
+
+	req.hdr.nlmsg_len       = sizeof(req);
+	req.hdr.nlmsg_type      = 0x1234;
+	req.hdr.nlmsg_flags     = NLM_F_DUMP | NLM_F_REQUEST;
+	req.hdr.nlmsg_seq       = 0xabcd;
+
+	memset(&msg, 0, sizeof(msg));
+	msg.msg_namelen = 0;
+	msg.msg_iov     = &iov;
+	msg.msg_iovlen  = 1;
+
+	iov.iov_base    = (void *) &req;
+	iov.iov_len     = sizeof(req);;
+
+	if (sendmsg(csk, &msg, 0) < 0) {
+		err("Can't send request message");
+		return 1;
+	}
+
+	memset(&msg, 0, sizeof(msg));
+	msg.msg_namelen = 0;
+	msg.msg_iov     = &iov;
+	msg.msg_iovlen  = 1;
+
+	iov.iov_base    = buf;
+	iov.iov_len     = sizeof(buf);
+
+	if (recvmsg(ssk, &msg, 0) < 0) {
+		err("Can't recv request message");
+		return 1;
+	}
+
+	if (recvmsg(bsk, &msg, 0) < 0) {
+		err("Can't recv request message");
+		return 1;
+	}
+
+	addr.nl_family = AF_NETLINK;
+	addr.nl_groups = 0;
+	addr.nl_pid = getpid();
+
+	memset(&msg, 0, sizeof(msg));
+	msg.msg_namelen = sizeof(addr);
+	msg.msg_name	= &addr;
+	msg.msg_iov     = &iov;
+	msg.msg_iovlen  = 1;
+
+	iov.iov_base    = (void *) &req;
+	iov.iov_len     = sizeof(req);;
+
+	if (sendmsg(dsk, &msg, 0) < 0) {
+		err("Can't send request message");
+		return 1;
+	}
+
+	memset(&msg, 0, sizeof(msg));
+	msg.msg_namelen = 0;
+	msg.msg_iov     = &iov;
+	msg.msg_iovlen  = 1;
+
+	iov.iov_base    = buf;
+	iov.iov_len     = sizeof(buf);
+
+	if (recvmsg(ssk, &msg, 0) < 0) {
+		err("Can't recv request message");
+		return 1;
+	}
+
+	pass();
+
+	return 0;
+}
-- 
1.7.11.7