Initial import of t_shmat.
[pcca-tests.git] / mqueue.h / t_mq_parent_child_multmsg.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 <string.h>
34 #include <stdlib.h>
35 #include <unistd.h>     /* fork() */
36
37 #include <sys/types.h>
38 #include <sys/wait.h>
39
40 #define MQNAME  "/t_mq_parent_child_multmsg"
41
42 int
43 main(void)
44 {
45         /* Parent recites a poem. */
46         const char *msg[] = { "But I, being poor, have only my dreams;",
47                               "I have spread my dreams under your feet;",
48                               "Tread softly because you tread on my",
49                               "dreams.",
50                               "W.B. Yeats" };
51
52         /* Create the message queue.  */
53         mqd_t md;
54         md = mq_open(MQNAME, O_CREAT | O_EXCL | O_WRONLY, 0700, NULL);
55         assert(md != (mqd_t)-1);
56
57         /* Send messages. */
58         size_t i, N;
59
60         N = sizeof(msg) / sizeof(msg[0]);
61         for (i = 0; i < N; i++) {
62                 assert(mq_send(md, msg[i], strlen(msg[i]) + 1,
63                         /* priority */ 0) != -1);
64         }
65
66         /* Disassociate with message queue. */
67         assert(mq_close(md) != -1);
68
69         /* Fork and have child read the message from parent. */
70         pid_t pid;
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 | O_NONBLOCK);
77                 assert(md != (mqd_t)-1);
78
79                 char msg_recvd[8192];   /* Implementation defined. */
80                 for (i = 0; i < N; i++) {
81                         assert(mq_receive(md, msg_recvd, sizeof(msg_recvd),
82                                 /* priority */ NULL) != -1);
83
84                         /*printf("%s\n", msg_recvd);*/
85                         assert(strcmp(msg_recvd, msg[i]) == 0);
86                 }
87
88                 /* Disassociate with message queue. */
89                 assert(mq_close(md) != -1);
90
91                 /* Remove the message queue from the system. */
92                 assert(mq_unlink(MQNAME) != -1);
93         } else {
94                 /* We are inside the parent. */
95
96                 /* Wait for child to complete. */
97                 int status;
98                 assert(wait(&status) == pid);
99
100                 printf("passed\n");
101         }
102
103         /* Only reached by child. */
104         return EXIT_SUCCESS;
105 }