/* $OpenBSD: socket3.c,v 1.4 2005/10/30 23:59:43 fgsch Exp $ */
/* PUBLIC DOMAIN Oct 2002 <marc@snafu.org> */

/* Test blocking/non-blocking mode inheritance on accept */

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

#include <netinet/in.h>

#include <fcntl.h>
#include <poll.h>
#include <pthread.h>
#include <string.h>
#include <unistd.h>

#include "test.h"

/*
 * connect to the test port passed in arg, then close the connection
 * and return.
 */
static void *
sock_connect(void *arg)
{
        struct sockaddr_in sin;
        int port;
        int sock;

        SET_NAME("connect");
        port = *(int *)arg;
        CHECKe(sock = socket(AF_INET, SOCK_STREAM, 0));
        sin.sin_family = AF_INET;
        sin.sin_port = htons(port);
        sin.sin_addr.s_addr = htonl(INADDR_LOOPBACK);
        CHECKe(connect(sock, (struct sockaddr *)&sin, sizeof sin));
        CHECKe(close(sock));
        return NULL;
}

/*
 * listen for a connection, accept it using a non-blocking socket, and
 * verify that the blocking mode of the socket returned from accept is
 * also non-blocking
 */
static void *
sock_accept(void *arg)
{
        pthread_t connect_thread;
        struct pollfd fds;
        struct sockaddr_in sa;
        struct sockaddr accept_sa;
        int accept_fd;
        int accept_sa_size;
        int flags;
        int listen_fd;
        int port;

        SET_NAME("accept");

        /* listen for a connection */

        port = 6543;
        memset(&sa, 0, sizeof sa);
        sa.sin_family = AF_INET;
        sa.sin_addr.s_addr = htonl(INADDR_LOOPBACK);
        sa.sin_port = htons(port);
        CHECKe(listen_fd = socket(AF_INET, SOCK_STREAM, 0));
        printf("listen_fd = %d\n", listen_fd);
        while (1) {
                if (bind(listen_fd, (struct sockaddr *)&sa, sizeof(sa)) == 0)
                        break;
                if (errno == EADDRINUSE) {
                        sa.sin_port = htons(++port);
                        continue;
                }
                DIE(errno, "bind");
        }
        CHECKe(listen(listen_fd, 2));

        /* Create another thread to connect to the listening socket. */
        CHECKr(pthread_create(&connect_thread, NULL, sock_connect,
            (void *)&port));

        /*
         * Use poll to check for a pending connection as the socket
         * passed to accept will be in non-blocking mode.
         */
        fds.fd = listen_fd;
        fds.events = POLLIN;
        CHECKe(poll(&fds, 1, INFTIM));

        /*
         * set non blocking mode on the listening socket and close stdin
         * (fd 0) so the accept will use fd 0 (needed to test boundary
         * condition in the pthread accept code).
         */
        flags = fcntl(listen_fd, F_GETFL);
        CHECKr(fcntl(listen_fd, F_SETFL, flags |= O_NONBLOCK));
        CHECKe(close(STDIN_FILENO));
        accept_sa_size = sizeof accept_sa;
        CHECKe(accept_fd = accept(listen_fd, &accept_sa, &accept_sa_size));
        /* verify O_NONBLOCK on the accepted fd */
        flags = fcntl(accept_fd, F_GETFL);
        printf("accept_fd = %d, flags = %x\n", accept_fd, flags);
        ASSERT(flags & O_NONBLOCK);
        CHECKe(close(listen_fd));
        CHECKe(close(accept_fd));
        CHECKr(pthread_join(connect_thread, NULL));
        return NULL;
}

int
main(int argc, char * argv[])
{
        pthread_t accept_thread;

        CHECKr(pthread_create(&accept_thread, NULL, sock_accept, NULL));
        CHECKr(pthread_join(accept_thread, NULL));
        SUCCEED;
}