/*	$OpenBSD: sigio_socket.c,v 1.1 2020/09/16 14:02:23 mpi Exp $	*/

/*
 * Copyright (c) 2018 Visa Hankala
 *
 * Permission to use, copy, modify, and distribute this software for any
 * purpose with or without fee is hereby granted, provided that the above
 * copyright notice and this permission notice appear in all copies.
 *
 * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
 * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
 * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
 * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
 * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
 * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
 * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
 */

#include <sys/socket.h>
#include <netinet/in.h>
#include <netinet/tcp.h>
#include <assert.h>
#include <fcntl.h>
#include <signal.h>
#include <string.h>
#include <unistd.h>

#include "common.h"

int
test_socket_badpgid(void)
{
	int fds[2];

	assert(socketpair(AF_UNIX, SOCK_STREAM, 0, fds) == 0);
	return test_common_badpgid(fds[0]);
}

int
test_socket_badsession(void)
{
	int fds[2];

	assert(socketpair(AF_UNIX, SOCK_STREAM, 0, fds) == 0);
	return test_common_badsession(fds[0]);
}

int
test_socket_cansigio(void)
{
	int fds[2];

	assert(socketpair(AF_UNIX, SOCK_STREAM, 0, fds) == 0);
	return test_common_cansigio(fds);
}

int
test_socket_getown(void)
{
	int fds[2];

	assert(socketpair(AF_UNIX, SOCK_STREAM, 0, fds) == 0);
	return test_common_getown(fds[0]);
}

/*
 * Test that the parent socket's signal target gets assigned to the socket
 * of an accepted connection.
 */
int
test_socket_inherit(void)
{
	struct sockaddr_in inaddr;
	socklen_t inaddrlen;
	pid_t pid;
	int cli, flags, sfd, sock;

	sock = socket(AF_INET, SOCK_STREAM, 0);
	assert(sock != -1);

	memset(&inaddr, 0, sizeof(inaddr));
	inaddr.sin_len = sizeof(inaddr);
	inaddr.sin_family = AF_INET;
	inaddr.sin_addr.s_addr = htonl(INADDR_LOOPBACK);
	assert(bind(sock, (struct sockaddr *)&inaddr, sizeof(inaddr)) == 0);
	assert(listen(sock, 1) == 0);

	flags = fcntl(sock, F_GETFL);
	assert(fcntl(sock, F_SETFL, flags | O_ASYNC) == 0);

	if (test_fork(&pid, &sfd) == PARENT) {
		inaddrlen = sizeof(inaddr);
		cli = accept(sock, (struct sockaddr *)&inaddr, &inaddrlen);
		assert(cli != -1);
		assert(fcntl(cli, F_GETOWN) == 0);
		close(cli);

		assert(fcntl(sock, F_SETOWN, getpid()) == 0);
		test_barrier(sfd);

		inaddrlen = sizeof(inaddr);
		cli = accept(sock, (struct sockaddr *)&inaddr, &inaddrlen);
		assert(cli != -1);
		assert(fcntl(cli, F_GETOWN) == getpid());
		close(cli);
	} else {
		inaddrlen = sizeof(inaddr);
		assert(getsockname(sock, (struct sockaddr *)&inaddr,
		    &inaddrlen) == 0);

		cli = socket(AF_INET, SOCK_STREAM, 0);
		assert(cli != -1);
		assert(connect(cli, (struct sockaddr *)&inaddr, sizeof(inaddr))
		    == 0);
		close(cli);

		test_barrier(sfd);

		cli = socket(AF_INET, SOCK_STREAM, 0);
		assert(cli != -1);
		assert(connect(cli, (struct sockaddr *)&inaddr, sizeof(inaddr))
		    == 0);
		close(cli);
	}
	return test_wait(pid, sfd);
}

int
test_socket_read(void)
{
	int fds[2];

	assert(socketpair(AF_UNIX, SOCK_STREAM, 0, fds) == 0);
	return test_common_read(fds);
}

int
test_socket_write(void)
{
	int fds[2];

	assert(socketpair(AF_UNIX, SOCK_STREAM, 0, fds) == 0);
	return test_common_write(fds);
}