t_mqueue.c revision 314817 
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
26static void
27send_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
58static void
59receive_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
104ATF_TC(mqueue);
105ATF_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
112ATF_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
149ATF_TP_ADD_TCS(tp)
150{
151	ATF_TP_ADD_TC(tp, mqueue);
152
153	return atf_no_error();
154}
155