mirror of
https://github.com/checkpoint-restore/criu.git
synced 2026-07-18 00:58:31 +00:00
93dd984ca0
Acked-by: Mike Rapoport <rppt@linux.ibm.com> Signed-off-by: Adrian Reber <areber@redhat.com>
115 lines
2.1 KiB
C
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, (struct sockaddr *)&addr, sizeof(struct sockaddr_un)) < 0) {
|
|
pr_perror("connect failed %s", iter);
|
|
return 1;
|
|
}
|
|
|
|
if (send(sk, MSG, sizeof(MSG), 0) != sizeof(MSG)) {
|
|
pr_perror("send failed %s", 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;
|
|
}
|