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