mqueue.h/t_mq_timedsend.c

   1 /*
   2  * Copyright (c) 2009, Stathis Kamperis
   3  * All rights reserved.
   4  *
   5  * Redistribution and use in source and binary forms, with or without
   6  * modification, are permitted provided that the following conditions
   7  * are met:
   8  * 1. Redistributions of source code must retain the above copyright
   9  *    notice, this list of conditions and the following disclaimer.
  10  * 2. Redistributions in binary form must reproduce the above copyright
  11  *    notice, this list of conditions and the following disclaimer in the
  12  *    documentation and/or other materials provided with the distribution.
  13  *
  14  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
  15  * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
  16  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
  17  * FOR A PARTICULAR PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE
  18  * COPYRIGHT HOLDERS OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
  19  * INCIDENTAL, SPECIAL, EXEMPLARY OR CONSEQUENTIAL DAMAGES (INCLUDING,
  20  * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
  21  * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
  22  * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
  23  * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT
  24  * OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  25  * SUCH DAMAGE.
  26  */
  27
  28 #include <assert.h>
  29 #include <errno.h>
  30 #include <fcntl.h>
  31 #include <mqueue.h>
  32 #include <stdio.h>
  33 #include <stdlib.h>
  34 #include <string.h>     /* memset() */
  35 #include <time.h>
  36 #include <unistd.h>     /* fork() */
  37 #include <sys/wait.h>
  38
  39 #define MQNAME  "/t_mq_timedsend"
  40
  41 int main(void)
  42 {
  43         mqd_t md2;
  44
  45         /* Create a message queue for with capacity `1'. */
  46         struct mq_attr attr;
  47         mqd_t md;
  48
  49         memset(&attr, 0, sizeof(attr));
  50         attr.mq_maxmsg = 1;     /* Maximum number of messages. */
  51         attr.mq_msgsize = 1024; /* Maximum message size. */
  52
  53         md = mq_open(MQNAME, O_CREAT | O_EXCL | O_WRONLY, 0700, &attr);
  54         assert(md != (mqd_t)-1);
  55
  56         /* Send a message to fill in the queue. */
  57         assert(mq_send(md, "foo", sizeof("foo"), /* priority */ 0) != -1);
  58
  59         /*
  60          * Fork and let the parent block on mq_timedsend().
  61          * Meanwhile, the child receives the old message from the queue, making
  62          * room for parent to send the new one!
  63          */
  64         pid_t pid = fork();
  65         assert(pid != -1);
  66
  67         if (pid == 0) {
  68                 /* We are inside the child. */
  69                 md2 = mq_open(MQNAME, O_RDONLY);
  70                 assert(md2 != (mqd_t)-1);
  71
  72                 char msg_recvd[8192];   /* Implementation defined. */
  73                 assert(mq_receive(md2, msg_recvd, sizeof(msg_recvd),
  74                         /* priority */ NULL) != -1);
  75
  76                 /* Disassociate from message queue. */
  77                 assert(mq_close(md2) != -1);
  78         } else {
  79                 /* We are inside the parent. */
  80                 /* Make sure the child has completed. */
  81                 int status;
  82                 assert(wait(&status) == pid);
  83
  84                 /*
  85                  * Determine if the child exited normally, or due to a SIGABRT
  86                  * signal being delivered to it by a failed assertion.
  87                  */
  88                 if (WIFSIGNALED(status)) {
  89                         assert(WTERMSIG(status) == SIGABRT);
  90                         return (EXIT_FAILURE);
  91                 }
  92
  93                 /*
  94                  * This shouldn't timeout now, since the queue is or is about to
  95                  * become empty. It may block though for a while.
  96                  */
  97                 struct timespec now, timeout;
  98
  99                 assert(clock_gettime(CLOCK_REALTIME, &now) == 0);
 100                 timeout.tv_sec  = now.tv_sec + 3;
 101                 timeout.tv_nsec = now.tv_nsec;
 102
 103                 assert(mq_timedsend(md, "foo", sizeof("foo"), /* priority */ 0,
 104                         &timeout) != -1);
 105
 106                 /* Disassociate from message queue. */
 107                 mq_close(md);
 108
 109                 /* Remove message queue from the system. */
 110                 mq_unlink(MQNAME);
 111
 112                 printf("passed\n");
 113
 114                 return (EXIT_SUCCESS);
 115         }
 116
 117         /* Only reached by child upon success. */
 118         return (EXIT_SUCCESS);
 119 }