mirror of
https://github.com/checkpoint-restore/criu
synced 2025-08-22 18:07:57 +00:00
cd1570b15e
v2: remove unnecessary elif and else after return in
wait_server_addr()
v3: use IOError instead of FileNotFoundError for python2
compatibility
Signed-off-by: Zeyad Yasser <zeyady98@gmail.com>
70 lines
1.2 KiB
C
70 lines
1.2 KiB
C
#include "zdtmtst.h"
|
|
|
|
#if defined(ZDTM_IPV6)
|
|
#define ZDTM_FAMILY AF_INET6
|
|
#else
|
|
#define ZDTM_FAMILY AF_INET
|
|
#endif
|
|
|
|
const char *test_doc = "Check that sockets are locked between dump and restore\n";
|
|
const char *test_author = "Zeyad Yasser <zeyady98@gmail.com>";
|
|
|
|
#include <errno.h>
|
|
#include <fcntl.h>
|
|
#include <sched.h>
|
|
#include <sys/socket.h>
|
|
|
|
static int port = 8880;
|
|
#define SYNCFILE_PATH "socket_lock.sync"
|
|
|
|
int main(int argc, char **argv)
|
|
{
|
|
char buf[5];
|
|
int fd_s, fd_sock, buf_len;
|
|
FILE *f_sync;
|
|
|
|
test_init(argc, argv);
|
|
|
|
if ((fd_s = tcp_init_server(ZDTM_FAMILY, &port)) < 0) {
|
|
pr_err("initializing server failed\n");
|
|
return 1;
|
|
}
|
|
|
|
// Server is ready to accept sockets
|
|
f_sync = fopen(SYNCFILE_PATH, "w");
|
|
if (f_sync == NULL) {
|
|
pr_perror("cannot create sync file");
|
|
return 1;
|
|
}
|
|
#if defined(ZDTM_IPV6)
|
|
if (fprintf(f_sync, "ipv6") < 0) {
|
|
#else
|
|
if (fprintf(f_sync, "ipv4") < 0) {
|
|
#endif
|
|
pr_perror("cannot write to sync file");
|
|
return 1;
|
|
}
|
|
fclose(f_sync);
|
|
|
|
fd_sock = tcp_accept_server(fd_s);
|
|
|
|
test_daemon();
|
|
test_waitsig();
|
|
|
|
buf_len = recv(fd_sock, buf, 3, MSG_WAITALL);
|
|
if (buf_len < 0) {
|
|
pr_perror("read less then have to");
|
|
return 1;
|
|
}
|
|
|
|
if (!strncmp(buf, "ABC", 3))
|
|
pass();
|
|
else
|
|
fail();
|
|
|
|
close(fd_sock);
|
|
close(fd_s);
|
|
|
|
return 0;
|
|
}
|