/* $OpenBSD: unixsock_test.c,v 1.3 2017/03/08 19:28:47 deraadt Exp $ */
/* Written by Claudio Jeker in 2011 */
/* Public domain */
#include <sys/types.h>
#include <sys/socket.h>
#include <sys/stat.h>
#include <sys/un.h>
#include <sys/wait.h>
#include <err.h>
#include <string.h>
#include <unistd.h>
#include <stdlib.h>
#include <stdio.h>
#include <errno.h>

char path[1024];
char *dir;

int
test_bind(struct sockaddr_un *sun, socklen_t slen)
{
	int s, e, r;

	s = socket(AF_UNIX, SOCK_STREAM, 0);
	if (s == -1)
		err(1, "socket");

	r = bind(s, (struct sockaddr *)sun, slen);
	e = errno;
	close(s);
	sun->sun_path[slen - 2] = '\0';
	unlink(sun->sun_path);
	errno = e;
	return r;
}

int
test_connect(struct sockaddr_un *sun, socklen_t slen, struct sockaddr_un *b)
{
	int s, s2, e, r;

	s = socket(AF_UNIX, SOCK_STREAM, 0);
	if (s == -1)
		err(1, "socket");

	s2 = socket(AF_UNIX, SOCK_STREAM, 0);
	if (s2 == -1)
		err(1, "socket");
	if (bind(s2, (struct sockaddr *)b, sizeof(*b)) == -1)
		err(1, "bind");
	if (listen(s2, 5) == -1)
		err(1, "listen");

	r = connect(s, (struct sockaddr *)sun, slen);
	e = errno;
	close(s);
	close(s2);
	unlink(b->sun_path);
	errno = e;
	return r;
}

struct test {
	socklen_t	len;
	int		r;
} t[] = {
	{30, 0},
	{50, 0},
	{100, 0},
	{102, 0},
	{103, 0},
	{104, -1},
	{105, -1},
	{110, -1},
	{200, -1},
	{0, 0}
};

int
main()
{
	struct sockaddr_storage ss;
	struct sockaddr_un *sun, sun2;
	char dir_template[] = "/tmp/peer.XXXXXXXXXX";
	char aaa[] = "aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa"
	    "aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa"
	    "aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa";
	int i, fail = 0;

	dir = mkdtemp(dir_template);
	if (!dir)
		err(1, "mkdtemp");
	if (chdir(dir) == -1)
		err(1, "chdir");
	snprintf(path, sizeof path, "%s/%s", dir, "socket");

	for (i = 0; t[i].len != 0; i++) {
		socklen_t slen = t[i].len;
		memset(&ss, 0xfe, sizeof(ss));
		sun = (struct sockaddr_un *)&ss;
		sun->sun_family = AF_UNIX;

		memset(&sun2, 0, sizeof(sun2));
		sun2.sun_family = AF_UNIX;

		snprintf(sun->sun_path, sizeof(ss) - 2, "%s.%.*s", path,
		    (int)(slen - strlen(path) - 1), aaa);
		snprintf(sun2.sun_path, sizeof(sun2) - 2, "%s.%.*s", "socket",
		    (int)(slen - strlen(path) - 1), aaa);

		if (test_bind(sun, slen + 2) != t[i].r) {
			warn("FAIL: bind(\"%s\") len %d", sun->sun_path,
			    slen + 2);
			fail = 1;
		}
		if (test_bind(sun, slen + 3) != t[i].r) {
			warn("FAIL2: bind(\"%s\") len %d", sun->sun_path,
			     slen + 3);
			fail = 1;
		}
		if (sizeof(*sun) >= slen + 2 &&
		    test_bind(sun, sizeof(*sun)) != t[i].r) {
			warn("FAIL3: bind(\"%s\") len %zd", sun->sun_path,
			    sizeof(*sun));
			fail = 1;
		}
		sun->sun_path[slen] = 'a';
		if (test_bind(sun, slen + 2) != t[i].r) {
			warn("FAIL4: bind(\"%.*s\") len %d no-NUL",
			     (int)(slen + 2), sun->sun_path, slen + 2);
			fail = 1;
		}
		sun->sun_path[slen] = '\0';

		if (test_connect(sun, slen + 2, &sun2) != t[i].r) {
			warn("FAIL: connect(\"%s\") len %d", sun->sun_path,
			    slen + 2);
			fail = 1;
		}
		if (test_connect(sun, slen + 3, &sun2) != t[i].r) {
			warn("FAIL2: connect(\"%s\") len %d", sun->sun_path,
			    slen + 3);
			fail = 1;
		}
		if (sizeof(*sun) >= slen + 2 &&
		    test_connect(sun, sizeof(*sun), &sun2) != t[i].r) {
			warn("FAIL3: connect(\"%s\") len %zd", sun->sun_path,
			    sizeof(*sun));
			fail = 1;
		}
		sun->sun_path[slen] = 'a';
		if (test_connect(sun, slen + 2, &sun2) != t[i].r) {
			warn("FAIL4: connect(\"%.*s\") len %d no-NUL",
			    slen + 2, sun->sun_path, slen + 2);
			fail = 1;
		}
	}
	return fail;
}