#include <sys/socket.h>
#include <sys/un.h>

#include <string.h>
#include <stdlib.h>
#include <unistd.h>

#include "unveil.h"

static int
test_bind_unix_socket(int do_uv)
{
	struct sockaddr_un sun1, sun2, sun3;
	char *path1, *path2, *path3;
	int c_fd1, c_fd2, fd1, fd2, fd3;

	char uv_dir3[] = "/tmp/uvdir3.XXXXXX";

	if (asprintf(&path1, "%s/1.sock", uv_dir1) == -1)
		err(1, NULL);
	if (asprintf(&path2, "%s/2.sock", uv_dir2) == -1)
		err(1, NULL);
	if (asprintf(&path3, "%s/3.sock", uv_dir3) == -1)
		err(1, NULL);

	memset(&sun1, 0, sizeof(sun1));
	sun1.sun_family = AF_UNIX;
	strlcpy(sun1.sun_path, path1, sizeof(sun1.sun_path));

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

	memset(&sun3, 0, sizeof(sun3));
	sun3.sun_family = AF_UNIX;
	strlcpy(sun3.sun_path, path3, sizeof(sun3.sun_path));

	if (unlink(path1) == -1)
		if (errno != ENOENT) {
			warn("%s: unlink %s", __func__, path1);
			return -1;
		}
	if (unlink(path2) == -1)
		if (errno != ENOENT) {
			warn("%s: unlink %s", __func__, path2);
			return -1;
		}
	if (unlink(path3) == -1)
		if (errno != ENOENT) {
			warn("%s: unlink %s", __func__, path3);
			return -1;
		}

	if (do_uv) {
		printf("testing bind and connect on unix socket\n");
		/* printf("testing bind on unix socket %s and %s\n", path1, path2); */
		if (unveil(uv_dir1, "wc") == -1) /* both bind and connect work */
			err(1, "unveil");
		if (unveil(uv_dir2, "c") == -1) /*  bind works, connect fails */
			err(1, "unveil");
		if (unveil(uv_dir3, "") == -1) /* no bind, dont test anything else */
			err(1, "unveil");
	}

	if ((fd1 = socket(AF_UNIX, SOCK_STREAM, 0)) == -1)
		err(1, "%s: socket", __func__);
	UV_SHOULD_SUCCEED(
	    (bind(fd1, (struct sockaddr *)&sun1, sizeof(sun1)) == -1), "bind");
	if (listen(fd1, 5) == -1)
		err(1, "%s: listen", __func__);

	if ((fd2 = socket(AF_UNIX, SOCK_STREAM, 0)) == -1)
		err(1, "%s: socket", __func__);
	UV_SHOULD_SUCCEED(
	    (bind(fd2, (struct sockaddr *)&sun2, sizeof(sun2)) == -1), "bind");
	if (listen(fd2, 5) == -1)
		err(1, "%s: listen", __func__);

	if ((fd3 = socket(AF_UNIX, SOCK_STREAM, 0)) == -1)
		err(1, "%s: socket", __func__);
	UV_SHOULD_ENOENT(
	    (bind(fd3, (struct sockaddr *)&sun3, sizeof(sun3)) == -1), "bind");

	/* Connect to control socket. */

	if ((c_fd1 = socket(AF_UNIX, SOCK_STREAM, 0)) == -1)
		err(1, "socket");
	UV_SHOULD_SUCCEED(
	    (connect(c_fd1, (struct sockaddr *)&sun1, sizeof(sun1)) == -1),
	    "connect");

	if ((c_fd2 = socket(AF_UNIX, SOCK_STREAM, 0)) == -1)
		err(1, "socket");
	UV_SHOULD_EACCES(
	    (connect(c_fd2, (struct sockaddr *)&sun2, sizeof(sun2)) == -1),
	    "connect");

	close(fd1);
	close(c_fd1);
	close(fd2);
	close(c_fd2);
	return 0;
}

int
main(void)
{
	int failures = 0;

	test_setup();

	failures += runcompare(test_bind_unix_socket);
	exit(failures);
}