criu/test/zdtm/static/socket_close_data01.c
Pavel Emelyanov 2e13f1f029 test: Get rid of live directory
Move static and transition into zdtm top. We can't move all the micro
tests themselves, as we need to distinguish static from non static (zdtm.py
makes additional checks on static ones).

Signed-off-by: Pavel Emelyanov <xemul@virtuozzo.com>
2016-02-20 13:40:52 +03:00

115 lines
2.1 KiB
C

#include <errno.h>
#include <stdlib.h>
#include <string.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <sys/types.h>
#include <sys/wait.h>
#include <sys/un.h>
#include "zdtmtst.h"
const char *test_doc = "Check data of bound socket and possibility to connect";
const char *test_author = "Kirill Tkhai <ktkhai@virtuozzo";
#define MSG "hello"
char *filename;
TEST_OPTION(filename, string, "file name", 1);
static int client(const char *iter)
{
struct sockaddr_un addr;
int sk;
sk = socket(AF_UNIX, SOCK_DGRAM, 0);
if (sk < 0) {
pr_perror("open client %s", iter);
return 1;
}
addr.sun_family = AF_UNIX;
strcpy(addr.sun_path, filename);
if (connect(sk, (void *)&addr, sizeof(struct sockaddr_un)) < 0) {
pr_perror("connect failed %s\n", iter);
return 1;
}
if (send(sk, MSG, sizeof(MSG), 0) != sizeof(MSG)) {
pr_perror("send failed %s\n", iter);
return 1;
}
return 0;
}
int main(int argc, char **argv)
{
struct sockaddr_un addr;
int srv, status, ret;
char buf[1024];
test_init(argc, argv);
srv = socket(AF_UNIX, SOCK_DGRAM | SOCK_NONBLOCK, 0);
if (srv < 0) {
pr_perror("open srv");
exit(1);
}
addr.sun_family = AF_UNIX;
strcpy(addr.sun_path, filename);
if (bind(srv, (struct sockaddr *) &addr, sizeof(struct sockaddr_un))) {
pr_perror("bind srv");
exit(1);
}
if (fork() == 0) {
close(srv);
client("(iter1)");
exit(0);
}
ret = 1;
if (wait(&status) == -1) {
fail("wait failed");
goto unlink;
}
if (status) {
pr_err("A child exited with 0x%x\n", status);
goto unlink;
}
test_daemon();
test_waitsig();
/* Test1: check we can read client message: */
ret = read(srv, buf, sizeof(MSG));
buf[ret > 0 ? ret : 0] = 0;
if (ret != sizeof(MSG)) {
fail("%d: %s", ret, buf);
ret = 1;
goto unlink;
}
/* Test2: check it's still possible to connect to the bound socket */
if (fork() == 0) {
exit(client("(iter2)"));
}
if (wait(&status) < 0) {
fail("wait failed");
goto unlink;
}
if (WEXITSTATUS(status) != 0) {
fail("connect failed");
goto unlink;
}
ret = 0;
pass();
unlink:
unlink(filename);
return ret;
}