mqueue.h/t_mq_prio.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 <fcntl.h>
  30 #include <mqueue.h>
  31 #include <signal.h>
  32 #include <stdio.h>
  33 #include <string.h>
  34 #include <stdlib.h>
  35 #include <unistd.h>     /* fork() */
  36 #include <sys/wait.h>
  37
  38 #define MQNAME  "/t_mq_prio"
  39
  40 mqd_t md;
  41
  42 /* Function prototypes. */
  43 static void myhandler(int sig);
  44
  45 int main(void)
  46 {
  47         const char lowmsg[] = "Parent says hello";
  48         const char highmsg[] = "Parent asks for HELP";
  49         unsigned int prio;
  50         int rv;
  51         pid_t pid;
  52
  53         signal(SIGABRT, myhandler);
  54
  55         /* Create a message queue for write only with default parameters. */
  56         md = mq_open(MQNAME, O_CREAT | O_EXCL | O_WRONLY, 0700, NULL);
  57         assert (md != -1);
  58
  59         /* Send messages, low first, then high. */
  60         rv = mq_send(md, lowmsg, sizeof(lowmsg), /* priority */ 0);
  61         assert(rv != -1);
  62
  63         rv = mq_send(md, highmsg, sizeof(highmsg), /* priority */ 1);
  64         assert(rv != -1);
  65
  66         /* Disassociate with message queue. */
  67         rv = mq_close(md);
  68         assert(rv != -1);
  69
  70         /* Fork and have child read the message from parent. */
  71         pid = fork();
  72         assert(pid != -1);
  73
  74         if (pid == 0) {
  75                 /* We are inside the child. */
  76                 md = mq_open(MQNAME, O_RDONLY);
  77                 assert(md != -1);
  78
  79                 char msg_recvd[8192];   /* Implementation defined. */
  80                 rv = mq_receive(md, msg_recvd, sizeof(msg_recvd), &prio);
  81                 assert(rv != -1);
  82                 /*
  83                  * Make sure that the message with higher priority,
  84                  * is delivered first of all.
  85                  */
  86                 assert(strcmp(msg_recvd, highmsg) == 0 && prio == 1);
  87
  88                 rv = mq_receive(md, msg_recvd, sizeof(msg_recvd), &prio);
  89                 assert(rv != -1);
  90                 assert(strcmp(msg_recvd, lowmsg) == 0 && prio == 0);
  91
  92                 /* Disassociate with message queue. */
  93                 rv = mq_close(md);
  94                 assert(rv != -1);
  95
  96                 /* Remove the message queue from the system. */
  97                 rv = mq_unlink(MQNAME);
  98                 assert(rv != -1);
  99
 100                 printf("passed\n");
 101         } else {
 102                 /* We are inside the parent. */
 103                 int status;
 104                 assert(wait(&status) == pid);
 105         }
 106
 107         return EXIT_SUCCESS;
 108 }
 109
 110 static void
 111 myhandler(int sig)
 112 {
 113         /*
 114          * Message queues' name & resources are persistent, i.e., they live
 115          * even after the process dies. That's why, disassociate and destroy
 116          * the queue on failure, or else we might end up with zombie queues and
 117          * hit the limit of max open queues.
 118          * Also, we don't care about the return value of the following calls.
 119          */
 120         mq_close(md);
 121         mq_unlink(MQNAME);
 122
 123         /* After this, the program will abort. */
 124 }