1 /* $NetBSD: t_mqueue.c,v 1.6 2017/01/14 20:57:24 christos Exp $ */
2
3 /*
4 * Test for POSIX message queue priority handling.
5 *
6 * This file is in the Public Domain.
7 */
8
9 #include <sys/stat.h>
10
11 #include <atf-c.h>
12 #include <errno.h>
13 #include <fcntl.h>
14 #include <mqueue.h>
15 #include <stdio.h>
16 #include <stdlib.h>
17 #include <string.h>
18 #include <unistd.h>
19
20 #define MQ_PRIO_BASE 24
21
22 static void
send_msgs(mqd_t mqfd)23 send_msgs(mqd_t mqfd)
24 {
25 char msg[2];
26
27 msg[1] = '\0';
28
29 msg[0] = 'a';
30 ATF_REQUIRE_MSG(mq_send(mqfd, msg, sizeof(msg), MQ_PRIO_BASE) != -1,
31 "mq_send 1 failed: %d", errno);
32
33 msg[0] = 'b';
34 ATF_REQUIRE_MSG(mq_send(mqfd, msg, sizeof(msg), MQ_PRIO_BASE + 1) != -1,
35 "mq_send 2 failed: %d", errno);
36
37 msg[0] = 'c';
38 ATF_REQUIRE_MSG(mq_send(mqfd, msg, sizeof(msg), MQ_PRIO_BASE) != -1,
39 "mq_send 3 failed: %d", errno);
40
41 msg[0] = 'd';
42 ATF_REQUIRE_MSG(mq_send(mqfd, msg, sizeof(msg), MQ_PRIO_BASE - 1) != -1,
43 "mq_send 4 failed: %d", errno);
44
45 msg[0] = 'e';
46 ATF_REQUIRE_MSG(mq_send(mqfd, msg, sizeof(msg), 0) != -1,
47 "mq_send 5 failed: %d", errno);
48
49 msg[0] = 'f';
50 ATF_REQUIRE_MSG(mq_send(mqfd, msg, sizeof(msg), MQ_PRIO_BASE + 1) != -1,
51 "mq_send 6 failed: %d", errno);
52 }
53
54 static void
receive_msgs(mqd_t mqfd)55 receive_msgs(mqd_t mqfd)
56 {
57 struct mq_attr mqa;
58 char *m;
59 unsigned p;
60 int len;
61
62 ATF_REQUIRE_MSG(mq_getattr(mqfd, &mqa) != -1, "mq_getattr failed %d",
63 errno);
64
65 len = mqa.mq_msgsize;
66 m = calloc(1, len);
67 ATF_REQUIRE_MSG(m != NULL, "calloc failed");
68
69 ATF_REQUIRE_MSG(mq_receive(mqfd, m, len, &p) != -1,
70 "mq_receive 1 failed: %d", errno);
71 ATF_REQUIRE_MSG(p == (MQ_PRIO_BASE + 1) && m[0] == 'b',
72 "mq_receive 1 prio/data mismatch");
73
74 ATF_REQUIRE_MSG(mq_receive(mqfd, m, len, &p) != -1,
75 "mq_receive 2 failed: %d", errno);
76 ATF_REQUIRE_MSG(p == (MQ_PRIO_BASE + 1) && m[0] == 'f',
77 "mq_receive 2 prio/data mismatch");
78
79 ATF_REQUIRE_MSG(mq_receive(mqfd, m, len, &p) != -1,
80 "mq_receive 3 failed: %d", errno);
81 ATF_REQUIRE_MSG(p == MQ_PRIO_BASE && m[0] == 'a',
82 "mq_receive 3 prio/data mismatch");
83
84 ATF_REQUIRE_MSG(mq_receive(mqfd, m, len, &p) != -1,
85 "mq_receive 4 failed: %d", errno);
86 ATF_REQUIRE_MSG(p == MQ_PRIO_BASE && m[0] == 'c',
87 "mq_receive 4 prio/data mismatch");
88
89 ATF_REQUIRE_MSG(mq_receive(mqfd, m, len, &p) != -1,
90 "mq_receive 5 failed: %d", errno);
91 ATF_REQUIRE_MSG(p == (MQ_PRIO_BASE - 1) && m[0] == 'd',
92 "mq_receive 5 prio/data mismatch");
93
94 ATF_REQUIRE_MSG(mq_receive(mqfd, m, len, &p) != -1,
95 "mq_receive 6 failed: %d", errno);
96 ATF_REQUIRE_MSG(p == 0 && m[0] == 'e',
97 "mq_receive 6 prio/data mismatch");
98 }
99
100 ATF_TC(mqueue);
ATF_TC_HEAD(mqueue,tc)101 ATF_TC_HEAD(mqueue, tc)
102 {
103
104 atf_tc_set_md_var(tc, "timeout", "3");
105 atf_tc_set_md_var(tc, "descr", "Checks mqueue send/receive");
106 }
107
ATF_TC_BODY(mqueue,tc)108 ATF_TC_BODY(mqueue, tc)
109 {
110 int status;
111 char *tmpdir;
112 char template[32];
113 char mq_name[64];
114
115 strlcpy(template, "./t_mqueue.XXXXXX", sizeof(template));
116 tmpdir = mkdtemp(template);
117 ATF_REQUIRE_MSG(tmpdir != NULL, "mkdtemp failed: %d", errno);
118 snprintf(mq_name, sizeof(mq_name), "%s/mq", tmpdir);
119
120 mqd_t mqfd;
121
122 mqfd = mq_open(mq_name, O_RDWR | O_CREAT,
123 S_IRUSR | S_IRWXG | S_IROTH, NULL);
124 ATF_REQUIRE_MSG(mqfd != -1, "mq_open failed: %d", errno);
125
126 send_msgs(mqfd);
127 receive_msgs(mqfd);
128
129 status = mq_close(mqfd);
130 ATF_REQUIRE_MSG(status == 0, "mq_close failed: %d", errno);
131 }
132
ATF_TP_ADD_TCS(tp)133 ATF_TP_ADD_TCS(tp)
134 {
135 ATF_TP_ADD_TC(tp, mqueue);
136
137 return atf_no_error();
138 }
139