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 #ifdef __FreeBSD__
21 #include "freebsd_test_suite/macros.h"
22 #endif
23
24 #define MQ_PRIO_BASE 24
25
26 static void
send_msgs(mqd_t mqfd)27 send_msgs(mqd_t mqfd)
28 {
29 char msg[2];
30
31 msg[1] = '\0';
32
33 msg[0] = 'a';
34 ATF_REQUIRE_MSG(mq_send(mqfd, msg, sizeof(msg), MQ_PRIO_BASE) != -1,
35 "mq_send 1 failed: %d", errno);
36
37 msg[0] = 'b';
38 ATF_REQUIRE_MSG(mq_send(mqfd, msg, sizeof(msg), MQ_PRIO_BASE + 1) != -1,
39 "mq_send 2 failed: %d", errno);
40
41 msg[0] = 'c';
42 ATF_REQUIRE_MSG(mq_send(mqfd, msg, sizeof(msg), MQ_PRIO_BASE) != -1,
43 "mq_send 3 failed: %d", errno);
44
45 msg[0] = 'd';
46 ATF_REQUIRE_MSG(mq_send(mqfd, msg, sizeof(msg), MQ_PRIO_BASE - 1) != -1,
47 "mq_send 4 failed: %d", errno);
48
49 msg[0] = 'e';
50 ATF_REQUIRE_MSG(mq_send(mqfd, msg, sizeof(msg), 0) != -1,
51 "mq_send 5 failed: %d", errno);
52
53 msg[0] = 'f';
54 ATF_REQUIRE_MSG(mq_send(mqfd, msg, sizeof(msg), MQ_PRIO_BASE + 1) != -1,
55 "mq_send 6 failed: %d", errno);
56 }
57
58 static void
receive_msgs(mqd_t mqfd)59 receive_msgs(mqd_t mqfd)
60 {
61 struct mq_attr mqa;
62 char *m;
63 unsigned p;
64 int len;
65
66 ATF_REQUIRE_MSG(mq_getattr(mqfd, &mqa) != -1, "mq_getattr failed %d",
67 errno);
68
69 len = mqa.mq_msgsize;
70 m = calloc(1, len);
71 ATF_REQUIRE_MSG(m != NULL, "calloc failed");
72
73 ATF_REQUIRE_MSG(mq_receive(mqfd, m, len, &p) != -1,
74 "mq_receive 1 failed: %d", errno);
75 ATF_REQUIRE_MSG(p == (MQ_PRIO_BASE + 1) && m[0] == 'b',
76 "mq_receive 1 prio/data mismatch");
77
78 ATF_REQUIRE_MSG(mq_receive(mqfd, m, len, &p) != -1,
79 "mq_receive 2 failed: %d", errno);
80 ATF_REQUIRE_MSG(p == (MQ_PRIO_BASE + 1) && m[0] == 'f',
81 "mq_receive 2 prio/data mismatch");
82
83 ATF_REQUIRE_MSG(mq_receive(mqfd, m, len, &p) != -1,
84 "mq_receive 3 failed: %d", errno);
85 ATF_REQUIRE_MSG(p == MQ_PRIO_BASE && m[0] == 'a',
86 "mq_receive 3 prio/data mismatch");
87
88 ATF_REQUIRE_MSG(mq_receive(mqfd, m, len, &p) != -1,
89 "mq_receive 4 failed: %d", errno);
90 ATF_REQUIRE_MSG(p == MQ_PRIO_BASE && m[0] == 'c',
91 "mq_receive 4 prio/data mismatch");
92
93 ATF_REQUIRE_MSG(mq_receive(mqfd, m, len, &p) != -1,
94 "mq_receive 5 failed: %d", errno);
95 ATF_REQUIRE_MSG(p == (MQ_PRIO_BASE - 1) && m[0] == 'd',
96 "mq_receive 5 prio/data mismatch");
97
98 ATF_REQUIRE_MSG(mq_receive(mqfd, m, len, &p) != -1,
99 "mq_receive 6 failed: %d", errno);
100 ATF_REQUIRE_MSG(p == 0 && m[0] == 'e',
101 "mq_receive 6 prio/data mismatch");
102 }
103
104 ATF_TC(mqueue);
ATF_TC_HEAD(mqueue,tc)105 ATF_TC_HEAD(mqueue, tc)
106 {
107
108 atf_tc_set_md_var(tc, "timeout", "3");
109 atf_tc_set_md_var(tc, "descr", "Checks mqueue send/receive");
110 }
111
ATF_TC_BODY(mqueue,tc)112 ATF_TC_BODY(mqueue, tc)
113 {
114 int status;
115 char *tmpdir;
116 char template[32];
117 char mq_name[64];
118
119 #ifdef __FreeBSD__
120 ATF_REQUIRE_KERNEL_MODULE("mqueuefs");
121 #endif
122
123 strlcpy(template, "./t_mqueue.XXXXXX", sizeof(template));
124 tmpdir = mkdtemp(template);
125 ATF_REQUIRE_MSG(tmpdir != NULL, "mkdtemp failed: %d", errno);
126 #ifdef __FreeBSD__
127 snprintf(mq_name, sizeof(mq_name), "/t_mqueue");
128 #else
129 snprintf(mq_name, sizeof(mq_name), "%s/mq", tmpdir);
130 #endif
131
132 mqd_t mqfd;
133
134 mqfd = mq_open(mq_name, O_RDWR | O_CREAT,
135 S_IRUSR | S_IRWXG | S_IROTH, NULL);
136 #ifdef __FreeBSD__
137 ATF_REQUIRE_MSG(mqfd != (mqd_t)-1, "mq_open failed: %d", errno);
138 #else
139 ATF_REQUIRE_MSG(mqfd != -1, "mq_open failed: %d", errno);
140 #endif
141
142 send_msgs(mqfd);
143 receive_msgs(mqfd);
144
145 status = mq_close(mqfd);
146 ATF_REQUIRE_MSG(status == 0, "mq_close failed: %d", errno);
147 }
148
ATF_TP_ADD_TCS(tp)149 ATF_TP_ADD_TCS(tp)
150 {
151 ATF_TP_ADD_TC(tp, mqueue);
152
153 return atf_no_error();
154 }
155