2 files changed, 125 insertions, 0 deletions

diff --git a/tools/regression/mqueue/mqtest5/Makefile b/tools/regression/mqueue/mqtest5/Makefile
new file mode 100644
index 000000000000..b43c4153a1f0
--- /dev/null
+++ b/tools/regression/mqueue/mqtest5/Makefile
@@ -0,0 +1,8 @@
+# $FreeBSD$
+
+PROG=mqtest5
+LDADD+=
+NO_MAN=
+DEBUG_FLAGS=-g
+
+.include <bsd.prog.mk>
diff --git a/tools/regression/mqueue/mqtest5/mqtest5.c b/tools/regression/mqueue/mqtest5/mqtest5.c
new file mode 100644
index 000000000000..2bb2625f81ec
--- /dev/null
+++ b/tools/regression/mqueue/mqtest5/mqtest5.c
@@ -0,0 +1,117 @@
+/* $FreeBSD$ */
+#include <stdio.h>
+#include <mqueue.h>
+#include <fcntl.h>
+#include <signal.h>
+#include <unistd.h>
+#include <sys/select.h>
+#include <sys/event.h>
+#include <signal.h>
+
+#define MQNAME	"/mytstqueue5"
+#define LOOPS	1000
+#define PRIO	10
+
+void sighandler(int sig)
+{
+	write(1, "timeout\n", 8);
+	_exit(1);
+}
+
+int main()
+{
+	int mq, status;
+	struct mq_attr attr;
+	int pid;
+	sigset_t set;
+	struct sigaction sa;
+	siginfo_t info;
+
+	mq_unlink(MQNAME);
+
+	sigemptyset(&set);
+	sigaddset(&set, SIGRTMIN);
+	sigprocmask(SIG_BLOCK, &set, NULL);
+	sigemptyset(&sa.sa_mask);
+	sa.sa_flags = SA_SIGINFO;
+	sa.sa_sigaction = (void *) SIG_DFL;
+	sigaction(SIGRTMIN, &sa, NULL);
+
+	attr.mq_maxmsg  = 5;
+	attr.mq_msgsize = 128;
+	mq = mq_open(MQNAME, O_CREAT | O_RDWR | O_EXCL, 0666, &attr);
+	if (mq == -1)
+		err(1, "mq_open()");
+	status = mq_getattr(mq, &attr);
+	if (status)
+		err(1, "mq_getattr()");
+	pid = fork();
+	if (pid == 0) { /* child */
+		int prio, j, i;
+		char *buf;
+		struct sigevent sigev;
+
+		signal(SIGALRM, sighandler);
+
+		sigev.sigev_notify = SIGEV_SIGNAL;
+		sigev.sigev_signo = SIGRTMIN;
+		sigev.sigev_value.sival_int = 2;
+
+		mq_close(mq);
+		mq = mq_open(MQNAME, O_RDWR | O_NONBLOCK);
+		if (mq == -1)
+			err(1, "child: mq_open");
+		buf = malloc(attr.mq_msgsize);
+		for (j = 0; j < LOOPS; ++j) {
+			alarm(3);
+			status = mq_notify(mq, &sigev);
+			if (status)
+				err(1, "child: mq_notify");
+			status = sigwaitinfo(&set, &info);
+			if (status == -1)
+				err(1, "child: sigwaitinfo");
+			if (info.si_value.sival_int != 2)
+				err(1, "child: sival_int");
+			status = mq_receive(mq, buf, attr.mq_msgsize, &prio);
+			if (status == -1)
+				err(2, "child: mq_receive");
+			for (i = 0; i < attr.mq_msgsize; ++i)
+				if (buf[i] != i)
+					err(3, "child: message data corrupted");
+			if (prio != PRIO)
+				err(4, "child: priority is incorrect: %d",
+					 prio);
+		}
+		alarm(0);
+		free(buf);
+		mq_close(mq);
+		return (0);
+	} else if (pid == -1) {
+		err(1, "fork()");
+	} else {
+		char *buf;
+		int i, j, prio;
+
+		signal(SIGALRM, sighandler);
+		buf = malloc(attr.mq_msgsize);
+		for (j = 0; j < LOOPS; ++j) {
+			for (i = 0; i < attr.mq_msgsize; ++i) {
+				buf[i] = i;
+			}
+			alarm(3);
+			status = mq_send(mq, buf, attr.mq_msgsize, PRIO);
+			if (status) {
+				kill(pid, SIGKILL);
+				err(2, "mq_send()");
+			}
+		}
+		alarm(3);
+		wait(&status);
+		alarm(0);
+	}
+	status = mq_close(mq);
+	if (status)
+		err(1, "mq_close");
+	mq_unlink(MQNAME);
+	return (0);
+}