/*
 * This file and its contents are supplied under the terms of the
 * Common Development and Distribution License ("CDDL"), version 1.0.
 * You may only use this file in accordance with the terms of version
 * 1.0 of the CDDL.
 *
 * A full copy of the text of the CDDL should have accompanied this
 * source.  A copy of the CDDL is also available via the Internet at
 * http://www.illumos.org/license/CDDL.
 */

/*
 * Copyright 2017 Sebastian Wiedenroth
 */

/*
 * Test for MSG_NOSIGNAL flag.
 */


#include <unistd.h>
#include <signal.h>
#include <stdio.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <errno.h>
#include <assert.h>

/* allow the test to build and fail */
#ifndef MSG_NOSIGNAL
#define	MSG_NOSIGNAL 0
#endif

volatile sig_atomic_t sigcount = 0;

void
sigpipe_h(int sig0)
{
	sigcount++;
	signal(SIGPIPE, sigpipe_h);
}

int
main()
{
	signal(SIGPIPE, sigpipe_h);

	int len = 0;
	const char *msg = "hello illumos";

	struct sockaddr_in sin;
	sin.sin_family = AF_INET;
	sin.sin_addr.s_addr = inet_addr("127.0.0.1");
	sin.sin_port = htons(4242);

	int s = socket(PF_INET, SOCK_STREAM, 0);
	int c = socket(PF_INET, SOCK_STREAM, 0);
	assert(s >= 0 && c >= 0);

	assert(bind(s, (struct sockaddr *)&sin, sizeof (sin)) >= 0);
	assert(listen(s, 3) >= 0);
	assert(connect(c, (struct sockaddr *)&sin, sizeof (sin)) >= 0);
	assert(close(s) == 0);

	assert(MSG_NOSIGNAL > 0);
	assert(sigcount == 0);

	/* test failure with signal */
	len = send(c, msg, strlen(msg), 0);
	assert(len == -1 && errno == EPIPE);
	sleep(1);
	assert(sigcount == 1);

	/* test failure without signal */
	len = send(c, msg, strlen(msg), MSG_NOSIGNAL);
	assert(len == -1 && errno == EPIPE);
	sleep(1);
	assert(sigcount == 1);

	assert(close(c) == 0);
	return (0);
}