move test utilities to a separate directory

2 files changed, 140 insertions, 0 deletions

diff --git a/test/testclient.c b/test/testclient.c
new file mode 100644
index 0000000..19db1c1
--- /dev/null
+++ b/test/testclient.c
@@ -0,0 +1,44 @@
+#include <stdio.h>
+#include <errno.h>
+#include <sys/socket.h>
+#include <sys/un.h>
+#include <sys/stat.h>
+#include <string.h>
+#include <unistd.h>
+#include <assert.h>
+#include "utils.h"
+
+#define ERROR_EXIT(f_, ...) { \
+		fprintf(stderr, (f_), ##__VA_ARGS__); \
+		return 1; \
+	}
+
+int main()
+{
+	struct sockaddr_un addr = {0};
+	int sock;
+
+	// Create and connect a unix domain socket
+	sock = socket(AF_UNIX, SOCK_STREAM, 0);
+	if (sock == -1)
+		ERROR_EXIT("socket: %s\n", strerror(errno));
+
+	addr.sun_family  = AF_UNIX;
+	strcpy(&addr.sun_path[1], "/tmp/voidnsrun-test.sock");
+
+	if (connect(sock, (struct sockaddr *)&addr, sizeof(addr)) == -1)
+		ERROR_EXIT("connect: %s\n", strerror(errno));
+
+	int fd = recv_fd(sock);
+	close(sock);
+
+	assert(fd != 0);
+
+	struct stat st;
+	if (fstat(fd, &st) == -1)
+		ERROR_EXIT("stat: %s\n", strerror(errno));
+
+	printf("st_ino: %lu\n", st.st_ino);
+
+	return 0;
+}
\ No newline at end of file
diff --git a/test/testserver.c b/test/testserver.c
new file mode 100644
index 0000000..091d580
--- /dev/null
+++ b/test/testserver.c
@@ -0,0 +1,96 @@
+#define _GNU_SOURCE
+#include <stdio.h>
+#include <string.h>
+#include <fcntl.h>
+#include <unistd.h>
+#include <errno.h>
+#include <signal.h>
+#include <sys/socket.h>
+#include <sys/un.h>
+#include <sys/prctl.h>
+#include "utils.h"
+
+#define ERROR(f_, ...) fprintf(stderr, (f_), ##__VA_ARGS__)
+#define UNUSED(x)      (void)(x)
+#define ERROR_EXIT(f_, ...) { \
+		fprintf(stderr, (f_), ##__VA_ARGS__); \
+		return 1; \
+	}
+
+volatile sig_atomic_t term_caught = 0;
+void onterm(int sig)
+{
+	printf("sigterm caught\n");
+	UNUSED(sig);
+	term_caught = 1;
+}
+
+int main()
+{
+	int result;
+	int sock_fd, sock_conn;
+	int nsfd;
+
+	/* Get current namespace's file descriptor. */
+	nsfd = open("/proc/self/ns/mnt", O_RDONLY);
+	if (nsfd == -1)
+		ERROR_EXIT("error: failed to acquire mount namespace's fd.%s\n",
+				   strerror(errno));
+
+	/* Fork. */
+	pid_t ppid_before_fork = getpid();
+	pid_t pid = fork();
+	if (pid == -1)
+		ERROR_EXIT("fork: %s\n", strerror(errno));
+
+	if (pid == 0) {
+		/* Catch SIGTERM. */
+		struct sigaction sa = {0};
+		sa.sa_handler = onterm;
+		sigaction(SIGTERM, &sa, NULL);
+
+        /* Ignore SIGINT. */
+        signal(SIGINT, SIG_IGN);
+
+		/* Set the child to die when parent thread dies. */
+		int r = prctl(PR_SET_PDEATHSIG, SIGTERM);
+		if (r == -1)
+			ERROR_EXIT("prctl: %s\n", strerror(errno));
+		if (getppid() != ppid_before_fork)
+			ERROR_EXIT("error: parent has died already.\n");
+
+		/* Create unix socket. */
+		sock_fd = socket(AF_UNIX, SOCK_STREAM, 0);
+		if (sock_fd == -1)
+			ERROR_EXIT("socket: %s.\n", strerror(errno));
+
+		struct sockaddr_un sock_addr = {0};
+		sock_addr.sun_family = AF_UNIX;
+		strcpy(&sock_addr.sun_path[1], "/tmp/voidnsrun-test.sock");
+
+		if (bind(sock_fd, (struct sockaddr *)&sock_addr, sizeof(sock_addr)) == -1)
+			ERROR_EXIT("bind: %s\n", strerror(errno));
+
+		listen(sock_fd, 1);
+
+		while (!term_caught) {
+			sock_conn = accept(sock_fd, NULL, 0);
+			if (sock_conn == -1) {
+				ERROR("accept: %s\n", strerror(errno));
+				continue;
+			}
+			printf("accepted\n");
+			send_fd(sock_conn, nsfd);
+		}
+		printf("exiting\n");
+	} else {
+		/* This is parent. Launch a program. */
+		char *argv[2] = {"/bin/sh", NULL};
+
+		result = execvp(argv[0], (char *const *)argv);
+		if (result == -1)
+			ERROR_EXIT("execvp: %s\n", strerror(errno));
+	}
+
+	return 0;
+}
\ No newline at end of file