zdtm: Extend file_own test for sockets

Signed-off-by: Cyrill Gorcunov <gorcunov@openvz.org>
Signed-off-by: Pavel Emelyanov <xemul@parallels.com>

test/zdtm/live/static/file_fown.c

@@ -11,6 +11,7 @@
 #include <sys/types.h>
 #include <sys/wait.h>
 #include <sys/mman.h>
+#include <sys/socket.h>
 
 #include "zdtmtst.h"
 
@@ -26,7 +27,10 @@ static int received_io;
 
 #define MAP(map, i)		(((int *)map)[i])
 #define MAP_SYNC(map)		MAP(map, 0)
-#define MAP_PID(map)		MAP(map, 1)
+#define MAP_PID_PIPE(map)	MAP(map, 1)
+#define MAP_PID_SOK(map)	MAP(map, 2)
+
+#define SK_DATA "packet"
 
 static void signal_handler_io(int status)
 {
@@ -44,6 +48,9 @@ int main(int argc, char ** argv)
 	uid_t euid;
 	uid_t suid;
 
+	int ssk_pair[2];
+	char buf[64];
+
 	test_init(argc, argv);
 
 	if (getresuid(&ruid, &euid, &suid)) {
@@ -62,6 +69,11 @@ int main(int argc, char ** argv)
 		exit(1);
 	}
 
+	if (socketpair(AF_UNIX, SOCK_STREAM, 0, ssk_pair) == -1) {
+		fail("socketpair\n");
+		exit(1);
+	}
+
 	memset(&saio, 0, sizeof(saio));
 	saio.sa_handler = (sig_t)signal_handler_io;
 	sigaction(SIGIO, &saio, 0);
@@ -82,6 +94,19 @@ int main(int argc, char ** argv)
 	fcntl(pipes[0], F_SETFL, fcntl(pipes[0], F_GETFL) | O_NONBLOCK | O_ASYNC);
 	fcntl(pipes[1], F_SETFL, fcntl(pipes[1], F_GETFL) | O_NONBLOCK | O_ASYNC);
 
+	fcntl(ssk_pair[0], F_SETOWN, getpid());
+	fcntl(ssk_pair[0], F_SETSIG, SIGIO);
+	fcntl(ssk_pair[0], F_SETFL, fcntl(ssk_pair[0], F_GETFL) | O_NONBLOCK | O_ASYNC);
+	test_msg("main owner ssk_pair[0]: %d\n", fcntl(ssk_pair[0], F_GETOWN));
+
+	write(ssk_pair[0], SK_DATA, sizeof(SK_DATA));
+	read(ssk_pair[1], &buf, sizeof(buf));
+	if (strcmp(buf, SK_DATA)) {
+		fail("data corrupted\n");
+		exit(1);
+	}
+	test_msg("stream            : '%s'\n", buf);
+
 	if (setresuid(-1, euid, -1)) {
 		fail("setresuid failed");
 		exit(1);
@@ -112,7 +137,16 @@ int main(int argc, char ** argv)
 		write(pipes[1], &v, sizeof(v));
 		read(pipes[0], &v, sizeof(v));
 
-		MAP_PID(map) = fcntl(pipes[0], F_GETOWN);
+		write(ssk_pair[0], SK_DATA, sizeof(SK_DATA));
+		read(ssk_pair[1], &buf, sizeof(buf));
+		if (strcmp(buf, SK_DATA)) {
+			fail("data corrupted\n");
+			exit(1);
+		}
+		test_msg("stream            : '%s'\n", buf);
+
+		MAP_PID_PIPE(map) = fcntl(pipes[0], F_GETOWN);
+		MAP_PID_SOK(map) = fcntl(ssk_pair[0], F_GETOWN);
 
 		exit(0);
 	}
@@ -127,9 +161,9 @@ int main(int argc, char ** argv)
 
 	waitpid(pid, &status, P_ALL);
 
-	if (received_io < 1 || MAP_PID(map) != ppid) {
-		fail("received_io = %d ppid: %d  MAP_PID(map): %d\n",
-		     received_io, ppid, MAP_PID(map));
+	if (received_io < 1 || MAP_PID_PIPE(map) != ppid || MAP_PID_SOK(map) != ppid) {
+		fail("received_io = %d ppid: %d  MAP_PID_PIPE(map): %d MAP_PID_SOK(map): %d\n",
+		     received_io, ppid, MAP_PID_PIPE(map), MAP_PID_SOK(map));
 		exit(1);
 	}