1 /* $NetBSD: t_mqueue.c,v 1.3 2012/11/06 19:35:38 pgoyette 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 char *tmpdir; 20 21 #define MQ_PRIO_BASE 24 22 23 static void 24 send_msgs(mqd_t mqfd) 25 { 26 char msg[2]; 27 28 msg[1] = '\0'; 29 30 msg[0] = 'a'; 31 ATF_REQUIRE_MSG(mq_send(mqfd, msg, sizeof(msg), MQ_PRIO_BASE) != -1, 32 "mq_send 1 failed: %d", errno); 33 34 msg[0] = 'b'; 35 ATF_REQUIRE_MSG(mq_send(mqfd, msg, sizeof(msg), MQ_PRIO_BASE + 1) != -1, 36 "mq_send 2 failed: %d", errno); 37 38 msg[0] = 'c'; 39 ATF_REQUIRE_MSG(mq_send(mqfd, msg, sizeof(msg), MQ_PRIO_BASE) != -1, 40 "mq_send 3 failed: %d", errno); 41 42 msg[0] = 'd'; 43 ATF_REQUIRE_MSG(mq_send(mqfd, msg, sizeof(msg), MQ_PRIO_BASE - 1) != -1, 44 "mq_send 4 failed: %d", errno); 45 46 msg[0] = 'e'; 47 ATF_REQUIRE_MSG(mq_send(mqfd, msg, sizeof(msg), 0) != -1, 48 "mq_send 5 failed: %d", errno); 49 50 msg[0] = 'f'; 51 ATF_REQUIRE_MSG(mq_send(mqfd, msg, sizeof(msg), MQ_PRIO_BASE + 1) != -1, 52 "mq_send 6 failed: %d", errno); 53 } 54 55 static void 56 receive_msgs(mqd_t mqfd) 57 { 58 struct mq_attr mqa; 59 char *m; 60 unsigned p; 61 int len; 62 63 ATF_REQUIRE_MSG(mq_getattr(mqfd, &mqa) != -1, "mq_getattr failed %d", 64 errno); 65 66 len = mqa.mq_msgsize; 67 m = calloc(1, len); 68 ATF_REQUIRE_MSG(m != NULL, "calloc failed"); 69 70 ATF_REQUIRE_MSG(mq_receive(mqfd, m, len, &p) != -1, 71 "mq_receive 1 failed: %d", errno); 72 ATF_REQUIRE_MSG(p == (MQ_PRIO_BASE + 1) && m[0] == 'b', 73 "mq_receive 1 prio/data mismatch"); 74 75 ATF_REQUIRE_MSG(mq_receive(mqfd, m, len, &p) != -1, 76 "mq_receive 2 failed: %d", errno); 77 ATF_REQUIRE_MSG(p == (MQ_PRIO_BASE + 1) && m[0] == 'f', 78 "mq_receive 2 prio/data mismatch"); 79 80 ATF_REQUIRE_MSG(mq_receive(mqfd, m, len, &p) != -1, 81 "mq_receive 3 failed: %d", errno); 82 ATF_REQUIRE_MSG(p == MQ_PRIO_BASE && m[0] == 'a', 83 "mq_receive 3 prio/data mismatch"); 84 85 ATF_REQUIRE_MSG(mq_receive(mqfd, m, len, &p) != -1, 86 "mq_receive 4 failed: %d", errno); 87 ATF_REQUIRE_MSG(p == MQ_PRIO_BASE && m[0] == 'c', 88 "mq_receive 4 prio/data mismatch"); 89 90 ATF_REQUIRE_MSG(mq_receive(mqfd, m, len, &p) != -1, 91 "mq_receive 5 failed: %d", errno); 92 ATF_REQUIRE_MSG(p == (MQ_PRIO_BASE - 1) && m[0] == 'd', 93 "mq_receive 5 prio/data mismatch"); 94 95 ATF_REQUIRE_MSG(mq_receive(mqfd, m, len, &p) != -1, 96 "mq_receive 6 failed: %d", errno); 97 ATF_REQUIRE_MSG(p == 0 && m[0] == 'e', 98 "mq_receive 6 prio/data mismatch"); 99 } 100 101 ATF_TC_WITH_CLEANUP(mqueue); 102 ATF_TC_HEAD(mqueue, tc) 103 { 104 105 atf_tc_set_md_var(tc, "timeout", "3"); 106 atf_tc_set_md_var(tc, "descr", "Checks mqueue send/receive"); 107 } 108 109 ATF_TC_BODY(mqueue, tc) 110 { 111 int status; 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 133 ATF_TC_CLEANUP(mqueue, tc) 134 { 135 int status; 136 137 if (tmpdir != NULL) { 138 status = rmdir(tmpdir); 139 ATF_REQUIRE_MSG(status == 0, "rmdir failed: %d", errno); 140 } 141 } 142 143 ATF_TP_ADD_TCS(tp) 144 { 145 ATF_TP_ADD_TC(tp, mqueue); 146 147 return atf_no_error(); 148 } 149