criu/test/zdtm/static/net_lock_socket_iptables.c

#include "zdtmtst.h"

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, fd_sync, buf_len;

	test_init(argc, argv);

	if ((fd_s = tcp_init_server(AF_INET, &port)) < 0) {
		pr_err("initializing server failed\n");
		return 1;
	}

	// Server is ready to accept sockets
	fd_sync = open(SYNCFILE_PATH, O_CREAT, 0444);
	if (fd_sync == -1) {
		pr_perror("cannot create sync file");
		return 1;
	}
	close(fd_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;
}