1e100f6a2SEnji Cooper
2e100f6a2SEnji Cooper #include <sys/types.h>
3e100f6a2SEnji Cooper #include <sys/event.h>
4e100f6a2SEnji Cooper #include <sys/select.h>
5e100f6a2SEnji Cooper #include <sys/wait.h>
6e100f6a2SEnji Cooper #include <err.h>
7e100f6a2SEnji Cooper #include <fcntl.h>
8e100f6a2SEnji Cooper #include <mqueue.h>
9e100f6a2SEnji Cooper #include <signal.h>
10e100f6a2SEnji Cooper #include <stdio.h>
11e100f6a2SEnji Cooper #include <stdlib.h>
12e100f6a2SEnji Cooper #include <unistd.h>
13e100f6a2SEnji Cooper
14e8395c48SEnji Cooper #include "freebsd_test_suite/macros.h"
15e8395c48SEnji Cooper
16e100f6a2SEnji Cooper #define MQNAME "/mytstqueue4"
17e100f6a2SEnji Cooper #define LOOPS 1000
18e100f6a2SEnji Cooper #define PRIO 10
19e100f6a2SEnji Cooper
20e100f6a2SEnji Cooper static void
sighandler(int sig __unused)21e100f6a2SEnji Cooper sighandler(int sig __unused)
22e100f6a2SEnji Cooper {
23e100f6a2SEnji Cooper write(1, "timeout\n", 8);
24e100f6a2SEnji Cooper _exit(1);
25e100f6a2SEnji Cooper }
26e100f6a2SEnji Cooper
27e100f6a2SEnji Cooper int
main(void)28e100f6a2SEnji Cooper main(void)
29e100f6a2SEnji Cooper {
30e100f6a2SEnji Cooper struct kevent kev;
31e100f6a2SEnji Cooper struct mq_attr attr;
32e100f6a2SEnji Cooper mqd_t mq;
33e100f6a2SEnji Cooper int kq, status;
34e100f6a2SEnji Cooper pid_t pid;
35e100f6a2SEnji Cooper
36e8395c48SEnji Cooper PLAIN_REQUIRE_KERNEL_MODULE("mqueuefs", 0);
37e8395c48SEnji Cooper
38e100f6a2SEnji Cooper mq_unlink(MQNAME);
39e100f6a2SEnji Cooper
40e100f6a2SEnji Cooper attr.mq_maxmsg = 5;
41e100f6a2SEnji Cooper attr.mq_msgsize = 128;
42e100f6a2SEnji Cooper mq = mq_open(MQNAME, O_CREAT | O_RDWR | O_EXCL, 0666, &attr);
43e100f6a2SEnji Cooper if (mq == (mqd_t) -1)
44e100f6a2SEnji Cooper err(1, "mq_open()");
45e100f6a2SEnji Cooper status = mq_getattr(mq, &attr);
46e100f6a2SEnji Cooper if (status)
47e100f6a2SEnji Cooper err(1, "mq_getattr()");
48e100f6a2SEnji Cooper pid = fork();
49e100f6a2SEnji Cooper if (pid == 0) { /* child */
50e100f6a2SEnji Cooper char *buf;
51e100f6a2SEnji Cooper int j, i;
52e100f6a2SEnji Cooper unsigned int prio;
53e100f6a2SEnji Cooper
54e100f6a2SEnji Cooper mq_close(mq);
55e100f6a2SEnji Cooper kq = kqueue();
56e100f6a2SEnji Cooper mq = mq_open(MQNAME, O_RDWR);
57e100f6a2SEnji Cooper if (mq == (mqd_t)-1)
58e100f6a2SEnji Cooper err(1, "child: mq_open");
59ddce1c3dSKonstantin Belousov EV_SET(&kev, mq_getfd_np(mq), EVFILT_READ, EV_ADD, 0, 0, 0);
60e100f6a2SEnji Cooper status = kevent(kq, &kev, 1, NULL, 0, NULL);
61e100f6a2SEnji Cooper if (status == -1)
62e100f6a2SEnji Cooper err(1, "child: kevent");
63e100f6a2SEnji Cooper buf = malloc(attr.mq_msgsize);
64e100f6a2SEnji Cooper for (j = 0; j < LOOPS; ++j) {
65e100f6a2SEnji Cooper alarm(3);
66e100f6a2SEnji Cooper status = kevent(kq, NULL, 0, &kev, 1, NULL);
67e100f6a2SEnji Cooper if (status != 1)
68e100f6a2SEnji Cooper err(1, "child: kevent 2");
69e100f6a2SEnji Cooper status = mq_receive(mq, buf, attr.mq_msgsize, &prio);
70e100f6a2SEnji Cooper if (status == -1)
71e100f6a2SEnji Cooper err(2, "child: mq_receive");
72e100f6a2SEnji Cooper for (i = 0; i < attr.mq_msgsize; ++i)
73e100f6a2SEnji Cooper if (buf[i] != i)
74e100f6a2SEnji Cooper err(3, "child: message data corrupted");
75e100f6a2SEnji Cooper if (prio != PRIO)
76e100f6a2SEnji Cooper err(4, "child: priority is incorrect: %d",
77e100f6a2SEnji Cooper prio);
78e100f6a2SEnji Cooper }
79e100f6a2SEnji Cooper alarm(0);
80e100f6a2SEnji Cooper free(buf);
81e100f6a2SEnji Cooper mq_close(mq);
82e100f6a2SEnji Cooper return (0);
83e100f6a2SEnji Cooper } else if (pid == -1) {
84e100f6a2SEnji Cooper err(1, "fork()");
85e100f6a2SEnji Cooper } else {
86e100f6a2SEnji Cooper char *buf;
87e100f6a2SEnji Cooper int i, j;
88e100f6a2SEnji Cooper
89e100f6a2SEnji Cooper signal(SIGALRM, sighandler);
90e100f6a2SEnji Cooper kq = kqueue();
91ddce1c3dSKonstantin Belousov EV_SET(&kev, mq_getfd_np(mq), EVFILT_WRITE, EV_ADD, 0, 0, 0);
92e100f6a2SEnji Cooper status = kevent(kq, &kev, 1, NULL, 0, NULL);
93e100f6a2SEnji Cooper if (status == -1)
94e100f6a2SEnji Cooper err(1, "kevent");
95e100f6a2SEnji Cooper buf = malloc(attr.mq_msgsize);
96e100f6a2SEnji Cooper for (j = 0; j < LOOPS; ++j) {
97e100f6a2SEnji Cooper for (i = 0; i < attr.mq_msgsize; ++i) {
98e100f6a2SEnji Cooper buf[i] = i;
99e100f6a2SEnji Cooper }
100e100f6a2SEnji Cooper alarm(3);
101e100f6a2SEnji Cooper status = kevent(kq, NULL, 0, &kev, 1, NULL);
102e100f6a2SEnji Cooper if (status != 1)
103e100f6a2SEnji Cooper err(1, "child: kevent 2");
104e100f6a2SEnji Cooper status = mq_send(mq, buf, attr.mq_msgsize, PRIO);
105e100f6a2SEnji Cooper if (status) {
106e100f6a2SEnji Cooper err(2, "mq_send()");
107e100f6a2SEnji Cooper }
108e100f6a2SEnji Cooper }
109e100f6a2SEnji Cooper free(buf);
110e100f6a2SEnji Cooper alarm(3);
111e100f6a2SEnji Cooper wait(&status);
112e100f6a2SEnji Cooper alarm(0);
113e100f6a2SEnji Cooper }
114e100f6a2SEnji Cooper status = mq_close(mq);
115e100f6a2SEnji Cooper if (status)
116e100f6a2SEnji Cooper err(1, "mq_close");
117e100f6a2SEnji Cooper mq_unlink(MQNAME);
118e100f6a2SEnji Cooper return (0);
119e100f6a2SEnji Cooper }
120