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