[CRIU] [PATCH v3 15/15] zdtm: Add socket_snd_addr01 test

[Andrew Vagin]

On Wed, Jun 01, 2016 at 06:13:37PM +0300, Kirill Tkhai wrote:
> Send messages from client to server, and check their msg_names.
> Client sockets stay alive during suspend/restore, so we check
> that criu properly sends them to server, and it restores its
> promiscous queue right.
> 
> Signed-off-by: Kirill Tkhai <ktkhai at virtuozzo.com>
> ---
>  test/zdtm/static/Makefile            |    1 
>  test/zdtm/static/socket_snd_addr01.c |  166 ++++++++++++++++++++++++++++++++++
>  2 files changed, 167 insertions(+)
>  create mode 100644 test/zdtm/static/socket_snd_addr01.c
> 
> diff --git a/test/zdtm/static/Makefile b/test/zdtm/static/Makefile
> index b6ed8f3..d6041e8 100644
> --- a/test/zdtm/static/Makefile
> +++ b/test/zdtm/static/Makefile
> @@ -37,6 +37,7 @@ TST_NOFILE	=				\
>  		socket_aio			\
>  		socket_close_data		\
>  		socket_snd_addr			\
> +		socket_snd_addr01		\
>  		socket_dgram_data		\
>  		packet_sock			\
>  		packet_sock_mmap		\
> diff --git a/test/zdtm/static/socket_snd_addr01.c b/test/zdtm/static/socket_snd_addr01.c
> new file mode 100644
> index 0000000..f9065d5
> --- /dev/null
> +++ b/test/zdtm/static/socket_snd_addr01.c
> @@ -0,0 +1,166 @@
> +#define _GNU_SOURCE
> +#include <errno.h>
> +#include <unistd.h>
> +#include <fcntl.h>
> +#include <stdlib.h>
> +#include <string.h>
> +#include <sys/types.h>
> +#include <sys/socket.h>
> +#include <sys/un.h>
> +#include <sys/wait.h>
> +#include <limits.h>
> +#include <sys/time.h>
> +#include <sys/resource.h>
> +
> +#include "zdtmtst.h"
> +
> +/* Based on socket_snd_addr test */
> +const char *test_doc	= "Check that sender addresses are restored correctly";
> +const char *test_author	= "Kirill Tkhai <ktkhai at virtuozzo.com";
> +
> +#define SK_SRV "\0socket_snd_srv"
> +#define SK_NAMEA "\0A-socket_snd_clnt"
> +#define SK_NAMEB "\0B-socket_snd_clnt"
> +
> +char *sk_names[3] = {
> +		SK_NAMEA,
> +		SK_NAMEB,
> +		NULL,
> +	};
> +
> +#define MSG "hello"
> +
> +static inline int sk_name_len(const char *name)
> +{
> +	if (!name)
> +		return 0;
> +	/* '\0' prefix and '\0' postfix */
> +	return 1 + strlen(name + 1) + 1;
> +}
> +
> +int do_client(int clnt[])
> +{
> +	int i;
> +
> +	for (i = 0; i < 6; i++) {
> +		if (send(clnt[i%3], MSG, sizeof(MSG), 0) != sizeof(MSG)) {
> +			pr_perror("write");
> +			return 1;
> +		}
> +	}
> +

You need to signal a parent from here that it can be daemonized. You can
look at zdtm/static/fd for example.

> +	test_waitsig();
> +	return 0;
> +}
> +
> +int main(int argc, char **argv)
> +{
> +	struct sockaddr_un addr;
> +	unsigned int addrlen;
> +	int srv, clnt[3] = {-1, -1, -1}, ret, i, status;
> +	char buf[1024];
> +	struct iovec iov = {
> +			.iov_base = &buf,
> +			.iov_len = sizeof(buf),
> +		};
> +	struct msghdr hdr = {
> +			.msg_name = &addr,
> +			.msg_namelen = sizeof(addr),
> +			.msg_iov = &iov,
> +			.msg_iovlen = 1,
> +		};
> +	pid_t pid;
> +
> +	test_init(argc, argv);
> +
> +	srv = socket(PF_UNIX, SOCK_DGRAM | SOCK_NONBLOCK, 0);
> +	if (srv < 0) {
> +		fail("socket");
> +		exit(1);
> +	}
> +
> +	addr.sun_family = AF_UNIX;
> +	memcpy(addr.sun_path, SK_SRV, sizeof(SK_SRV));
> +	addrlen = sizeof(addr.sun_family) + sizeof(SK_SRV);
> +
> +	if (bind(srv, &addr, addrlen)) {
> +		fail("bind\n");
> +		exit(1);
> +	}
> +
> +	for (i = 0; i < 3; i++) {
> +		clnt[i] = socket(PF_UNIX, SOCK_DGRAM, 0);
> +
> +		if (sk_names[i]) {
> +			addr.sun_family = AF_UNIX;
> +			memcpy(addr.sun_path, sk_names[i], sk_name_len(sk_names[i]));
> +			addrlen = sizeof(addr.sun_family) + sk_name_len(sk_names[i]);;
> +
> +			if (bind(clnt[i], &addr, addrlen)) {
> +				fail("bind\n");
> +				exit(1);
> +			}
> +		}
> +
> +		memcpy(addr.sun_path, SK_SRV, sizeof(SK_SRV));
> +		addrlen = sizeof(addr.sun_family) + sizeof(SK_SRV);
> +		if (connect(clnt[i], &addr, addrlen)) {
> +			fail("connect\n");
> +			exit(1);
> +		}
> +	}
> +
> +	pid = fork();
> +	if (pid == 0) {
> +		close(srv);
> +		exit(do_client(clnt));
> +	} else if (pid != -1) {
> +		for (i = 0; i < 2; i++)
> +			close(clnt[i]);
> +	} else {
> +		fail("fork");
> +		exit(1);
> +	}
> +
> +	test_daemon();
> +	test_waitsig();
> +
> +	kill(pid, SIGTERM);
> +
> +	if (wait(&status) != pid) {
> +		fail("wait");
> +		exit(1);
> +	}
> +
> +	if (status != 0) {
> +		fail("%d:%d:%d:%d", WIFEXITED(status), WEXITSTATUS(status),
> +				    WIFSIGNALED(status), WTERMSIG(status));
> +		return 1;
> +	}
> +
> +	for (i = 0; i < 6; i++) {
> +		const char *sk_name = sk_names[i % 3];
> +		hdr.msg_namelen = sizeof(addr);
> +		memset(addr.sun_path, 0, sizeof(addr.sun_path));
> +		ret = recvmsg(srv, &hdr, MSG_DONTWAIT);
> +		buf[ret > 0 ? ret : 0] = 0;
> +		if (ret != sizeof(MSG)) {
> +			fail("%d: %s", ret, buf);
> +			return 1;
> +		}
> +
> +		if (!sk_name && !hdr.msg_namelen)
> +			continue;
> +		if (hdr.msg_namelen != sk_name_len(sk_name) + sizeof(addr.sun_family)) {
> +			fail("Name len is mismatch: iter=%d, %d %d", i, hdr.msg_namelen, sk_name_len(sk_name));
> +			return 1;
> +		}
> +		if (sk_name_len(sk_name) && memcmp(addr.sun_path, sk_name, sk_name_len(sk_name))) {
> +			fail("A sender address is mismatch");
> +			return 1;
> +		}
> +	}
> +
> +	pass();
> +	return 0;
> +}
> 
> _______________________________________________
> CRIU mailing list
> CRIU at openvz.org
> https://lists.openvz.org/mailman/listinfo/criu