[CRIU] [PATCH 3/3] zdtm: Add sockets02 test case

[Cyrill Gorcunov]

Code to test unix dgarm sockets (extracted from
previous sockets00 test case).

Signed-off-by: Cyrill Gorcunov <gorcunov at openvz.org>
---
 test/zdtm.sh                      |    1 +
 test/zdtm/live/static/Makefile    |    1 +
 test/zdtm/live/static/sockets02.c |  184 +++++++++++++++++++++++++++++++++++++
 3 files changed, 186 insertions(+), 0 deletions(-)
 create mode 100644 test/zdtm/live/static/sockets02.c

diff --git a/test/zdtm.sh b/test/zdtm.sh
index 86834b1..e5f9673 100644
--- a/test/zdtm.sh
+++ b/test/zdtm.sh
@@ -29,6 +29,7 @@ transition/fork
 static/zombie00
 static/sockets00
 static/sockets01
+static/sockets02
 static/socket_queues
 static/pid00
 static/pstree
diff --git a/test/zdtm/live/static/Makefile b/test/zdtm/live/static/Makefile
index ce02438..961c390 100644
--- a/test/zdtm/live/static/Makefile
+++ b/test/zdtm/live/static/Makefile
@@ -36,6 +36,7 @@ TST_NOFILE	=				\
 		pstree				\
 		sockets00			\
 		sockets01			\
+		sockets02			\
 		socket_queues			\
 		ipc_namespace			\
 		selfexe00			\
diff --git a/test/zdtm/live/static/sockets02.c b/test/zdtm/live/static/sockets02.c
new file mode 100644
index 0000000..25be6f0
--- /dev/null
+++ b/test/zdtm/live/static/sockets02.c
@@ -0,0 +1,184 @@
+#define _GNU_SOURCE
+
+#include <stdio.h>
+#include <stdlib.h>
+#include <ctype.h>
+#include <errno.h>
+#include <unistd.h>
+#include <sys/types.h>
+#include <sys/socket.h>
+#include <sys/wait.h>
+#include <sys/un.h>
+#include <sys/stat.h>
+#include <limits.h>
+#include <fcntl.h>
+
+#include "zdtmtst.h"
+
+const char *test_doc	= "Test unix dgram sockets\n";
+const char *test_author	= "Cyrill Gorcunov <gorcunov at openvz.org";
+
+#define SK_DATA_BOUND		"data-packet-bound"
+#define SK_DATA_CONN		"data-packet-conn"
+#define SK_DATA_BOUND_CONN	"data-packet-bound-conn"
+
+int main(int argc, char *argv[])
+{
+	struct sockaddr_un addr;
+	struct sockaddr_un name_bound;
+	struct sockaddr_un name_conn;
+	struct sockaddr_un name_bound_conn;
+	int sk_dgram_bound_client;
+	int sk_dgram_bound_server;
+	int sk_dgram_conn_client;
+	int sk_dgram_conn_server;
+	int sk_dgram_bound_conn;
+	unsigned int addrlen;
+
+	char path[PATH_MAX];
+	char buf[64];
+	char *cwd;
+
+	int ret;
+
+	test_init(argc, argv);
+
+	cwd = get_current_dir_name();
+	if (!cwd) {
+		fail("getcwd\n");
+		exit(1);
+	}
+
+	snprintf(path, sizeof(path), "%s/test-socket", cwd);
+	unlink(path);
+
+	addr.sun_family = AF_UNIX;
+	strncpy(addr.sun_path, path, sizeof(addr.sun_path));
+	addrlen = sizeof(addr.sun_family) + strlen(path);
+
+	sk_dgram_bound_client	= socket(AF_UNIX, SOCK_DGRAM, 0);
+	sk_dgram_bound_server	= socket(AF_UNIX, SOCK_DGRAM, 0);
+	sk_dgram_conn_client	= socket(AF_UNIX, SOCK_DGRAM, 0);
+	sk_dgram_conn_server	= socket(AF_UNIX, SOCK_DGRAM, 0);
+	sk_dgram_bound_conn	= socket(AF_UNIX, SOCK_DGRAM, 0);
+
+	if (sk_dgram_conn_server < 0	||
+	    sk_dgram_bound_server < 0	||
+	    sk_dgram_conn_client < 0	||
+	    sk_dgram_conn_server < 0	||
+	    sk_dgram_bound_conn < 0) {
+		fail("socket");
+		exit(1);
+	}
+
+	snprintf(path, sizeof(path), "%s/test-socket-bound", cwd);
+	unlink(path);
+
+	name_bound.sun_family = AF_UNIX;
+	strncpy(name_bound.sun_path, path, sizeof(name_bound.sun_path));
+
+	snprintf(path, sizeof(path), "%s/test-socket-conn", cwd);
+	unlink(path);
+
+	name_conn.sun_family = AF_UNIX;
+	strncpy(name_conn.sun_path, path, sizeof(name_conn.sun_path));
+
+	snprintf(path, sizeof(path), "%s/test-socket-bound-conn", cwd);
+	unlink(path);
+
+	name_bound_conn.sun_family = AF_UNIX;
+	strncpy(name_bound_conn.sun_path, path, sizeof(name_bound_conn.sun_path));
+
+	ret = bind(sk_dgram_bound_server, &name_bound, sizeof(name_bound));
+	if (ret) {
+		fail("bind");
+		exit(1);
+	}
+
+	ret = bind(sk_dgram_conn_server, &name_conn, sizeof(name_conn));
+	if (ret) {
+		fail("bind");
+		exit(1);
+	}
+
+	ret = connect(sk_dgram_conn_client, &name_conn, sizeof(name_conn));
+	if (ret) {
+		fail("connect");
+		exit(1);
+	}
+
+	ret = bind(sk_dgram_bound_conn, &name_bound_conn, sizeof(name_bound_conn));
+	if (ret) {
+		fail("bind");
+		exit(1);
+	}
+
+	/* Note, it's already bound, so make it more idiotic! */
+	ret = connect(sk_dgram_bound_conn, &name_bound_conn, sizeof(name_bound_conn));
+	if (ret) {
+		fail("connect");
+		exit(1);
+	}
+
+	memset(buf, 0, sizeof(buf));
+	sendto(sk_dgram_bound_client, SK_DATA_BOUND, sizeof(SK_DATA_BOUND), 0,
+	       &name_bound, sizeof(name_bound));
+	read(sk_dgram_bound_server, &buf, sizeof(buf));
+	if (strcmp(buf, SK_DATA_BOUND)) {
+		fail("data corrupted\n");
+		exit(1);
+	}
+	test_msg("dgram-bound       : '%s'\n", buf);
+
+	memset(buf, 0, sizeof(buf));
+	write(sk_dgram_conn_client, SK_DATA_CONN, sizeof(SK_DATA_CONN));
+	read(sk_dgram_conn_server, &buf, sizeof(buf));
+	if (strcmp(buf, SK_DATA_CONN)) {
+		fail("data corrupted\n");
+		exit(1);
+	}
+	test_msg("dgram-conn        : '%s'\n", buf);
+
+	memset(buf, 0, sizeof(buf));
+	write(sk_dgram_bound_conn, SK_DATA_BOUND_CONN, sizeof(SK_DATA_BOUND_CONN));
+	read(sk_dgram_bound_conn, &buf, sizeof(buf));
+	if (strcmp(buf, SK_DATA_BOUND_CONN)) {
+		fail("data corrupted\n");
+		exit(1);
+	}
+	test_msg("dgram-bound-conn  : '%s'\n", buf);
+
+	test_daemon();
+	test_waitsig();
+
+	memset(buf, 0, sizeof(buf));
+	sendto(sk_dgram_bound_client, SK_DATA_BOUND, sizeof(SK_DATA_BOUND), 0,
+	       &name_bound, sizeof(name_bound));
+	read(sk_dgram_bound_server, &buf, sizeof(buf));
+	if (strcmp(buf, SK_DATA_BOUND)) {
+		fail("data corrupted\n");
+		exit(1);
+	}
+	test_msg("dgram-bound       : '%s'\n", buf);
+
+	memset(buf, 0, sizeof(buf));
+	write(sk_dgram_conn_client, SK_DATA_CONN, sizeof(SK_DATA_CONN));
+	read(sk_dgram_conn_server, &buf, sizeof(buf));
+	if (strcmp(buf, SK_DATA_CONN)) {
+		fail("data corrupted\n");
+		exit(1);
+	}
+	test_msg("dgram-conn        : '%s'\n", buf);
+
+	memset(buf, 0, sizeof(buf));
+	write(sk_dgram_bound_conn, SK_DATA_BOUND_CONN, sizeof(SK_DATA_BOUND_CONN));
+	read(sk_dgram_bound_conn, &buf, sizeof(buf));
+	if (strcmp(buf, SK_DATA_BOUND_CONN)) {
+		fail("data corrupted\n");
+		exit(1);
+	}
+	test_msg("dgram-bound-conn  : '%s'\n", buf);
+
+	pass();
+	return 0;
+}
-- 
1.7.7.6