tests/kernel/t_mqueue.c

   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 }