mqueue.h/t_mq_select.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>
  35 #include <unistd.h>     /* fork() */
  36
  37 #define MQNAME  "/t_mq_select"
  38
  39 int main(void)
  40 {
  41         const char msg[] = "Child says hello to parent";
  42         struct timeval abs_timeout;
  43         mqd_t md;
  44         fd_set ms;
  45         pid_t pid;
  46
  47         /* Create a message queue for write only with default parameters. */
  48         md = mq_open(MQNAME, O_CREAT | O_EXCL | O_RDWR, 0777, NULL);
  49         assert(md != -1);
  50
  51         /* Initialize descriptor set to null set. */
  52         FD_ZERO(&ms);
  53
  54         /* Add message queue descriptor to set. */
  55         FD_SET(md, &ms);
  56         assert(FD_ISSET(md, &ms));      /* Just a sanity check. */
  57
  58         /* Wait for 2 seconds. */
  59         abs_timeout.tv_sec = 2;
  60         abs_timeout.tv_usec = 0;
  61
  62         /* This should time out. */
  63         assert(select(md + 1, &ms, NULL, NULL, &abs_timeout) == 0);
  64
  65         FD_SET(md, &ms);
  66         assert(FD_ISSET(md, &ms));      /* Just a sanity check. */
  67
  68         /* Fork. */
  69         pid = fork();
  70         assert(pid != -1);
  71
  72         if (pid == 0) {
  73                 /* We are inside the child. */
  74                 /* Wait for parent to block on select(). */
  75                 sleep(1);
  76
  77                 /* Send message. */
  78                 assert(mq_send(md, msg, sizeof(msg), /* priority */ 0) != -1);
  79         } else {
  80                 /* We are inside the parent. */
  81                 assert(select(md + 1, &ms, NULL, NULL, &abs_timeout) > 0);
  82                 assert(FD_ISSET(md, &ms));
  83
  84                 char msg_recvd[8192];   /* Implementation defined. */
  85                 assert(mq_receive(md, msg_recvd, sizeof(msg_recvd), NULL) != -1);
  86                 assert(strcmp(msg_recvd, msg) == 0);
  87
  88                 /* Remove message queue descriptor from the set. */
  89                 FD_CLR(md, &ms);
  90
  91                 /*
  92                  * At this point we know for sure that the child has completed,
  93                  * otherwise we would still be blocked by select().
  94                  */
  95
  96                 /* Disassociate with message queue. */
  97                 assert(mq_close(md) != -1);
  98
  99                 /* Remove the message queue from the system. */
 100                 assert(mq_unlink(MQNAME) != -1);
 101
 102                 printf("passed\n");
 103         }
 104
 105         return (EXIT_SUCCESS);
 106 }