contrib/netbsd-tests/kernel/t_mqueue.c

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