/*
 * Initially written by Yar Tikhiy <yar@freebsd.org> in PR 76398.
 * Bug fixes and instrumentation by kib@freebsd.org.
 */

#include <sys/types.h>
#include <sys/socket.h>
#include <sys/stat.h>
#include <sys/wait.h>
#include <err.h>
#include <errno.h>
#include <fcntl.h>
#include <md5.h>
#include <signal.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>

#include <atf-c.h>

#define NDATA	1000
#define	DELAY	2

static void
hup(int signo __unused)
{
}

static int ndata, seq;

static void
setdata(int n)
{
	ndata = n;
	seq = 0;
}

static char *
getdata(void)
{
	static char databuf[256];
	static char xeof[] = "#";

	if (seq > ndata)
		return (NULL);
	if (seq == ndata) {
		seq++;
		return (xeof);
	}
	sprintf(databuf, "%08d xxxxxxxxxxxxxxxxxxxxxxxxxxxxxx\n", seq++);
	return (databuf);
}

ATF_TC_WITHOUT_HEAD(eintr_test);
ATF_TC_BODY(eintr_test, tc)
{
	char c, digest0[33], digest[33], *p;
	FILE *fp;
	int i, s[2], total0, total;
	MD5_CTX md5;
	pid_t child;
	struct sigaction sa;

	MD5Init(&md5);
	setdata(NDATA);
	for (total0 = 0; (p = getdata()) != NULL; total0 += strlen(p))
		MD5Update(&md5, p, strlen(p));
	p = MD5End(&md5, digest0);

	sa.sa_handler = hup;
	sigemptyset(&sa.sa_mask);
	sa.sa_flags = 0;
	ATF_REQUIRE(sigaction(SIGHUP, &sa, NULL) == 0);

	ATF_REQUIRE(socketpair(PF_UNIX, SOCK_STREAM, 0, s) == 0);

	switch (child = fork()) {
	case -1:
		atf_tc_fail("fork failed %s", strerror(errno));
		break;

	case 0:
		ATF_REQUIRE((fp = fdopen(s[0], "w")) != NULL);
		close(s[1]);
		setdata(NDATA);
		while ((p = getdata())) {
			for (; *p;) {
				if (fputc(*p, fp) == EOF) {
					if (errno == EINTR) {
						clearerr(fp);
					} else {
						atf_tc_fail("fputc errno %s",
						    strerror(errno));
					}
				} else {
					p++;
				}
			}
		}
		fclose(fp);
		break;

	default:
		close(s[0]);
		ATF_REQUIRE((fp = fdopen(s[1], "r")) != NULL);
		sleep(DELAY);
		ATF_REQUIRE(kill(child, SIGHUP) != -1);
		sleep(DELAY);
		MD5Init(&md5);
		for (total = 0;;) {
			i = fgetc(fp);
			if (i == EOF) {
				if (errno == EINTR) {
					clearerr(fp);
				} else {
					atf_tc_fail("fgetc errno %s",
					    strerror(errno));
				}
				continue;
			}
			total++;
			c = i;
			MD5Update(&md5, &c, 1);
			if (i == '#')
				break;
		}
		MD5End(&md5, digest);
		fclose(fp);
		ATF_REQUIRE_MSG(total == total0,
		    "Total number of bytes read does not match: %d %d",
		    total, total0);
		ATF_REQUIRE_MSG(strcmp(digest, digest0) == 0,
		    "Digests do not match %s %s", digest, digest0);
		break;
	}
}

ATF_TP_ADD_TCS(tp)
{
	ATF_TP_ADD_TC(tp, eintr_test);
	return (atf_no_error());
}