1 files changed, 143 insertions, 0 deletions
diff --git a/lib/libc/tests/stdio/eintr_test.c b/lib/libc/tests/stdio/eintr_test.c
new file mode 100644
index 000000000000..55354c8926c5
--- /dev/null
+++ b/lib/libc/tests/stdio/eintr_test.c
@@ -0,0 +1,143 @@
+/*
+ * 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());
+}