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 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 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); 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 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 149 ATF_TP_ADD_TCS(tp) 150 { 151 ATF_TP_ADD_TC(tp, mqueue); 152 153 return atf_no_error(); 154 } 155