/*
 * Copyright (c) 2003, Intel Corporation. All rights reserved.
 * Created by:  julie.n.fleischer REMOVE-THIS AT intel DOT com
 * This file is licensed under the GPL license.  For the full content
 * of this license, see the COPYING file at the top level of this
 * source tree.
 */

/*
 * Test that if the access mode is O_RDWR, the message queue can
 * send and receive messages.
 *
 * Test for a message queue opened twice in two different processes.
 *
 * 3/13/03 - Added fix from Gregoire Pichon for specifying an attr
 *           with a mq_maxmsg >= BUFFER.
 */

#include <stdio.h>
#include <mqueue.h>
#include <fcntl.h>
#include <sys/stat.h>
#include <sys/types.h>
#include <unistd.h>
#include <string.h>
#include <sys/wait.h>
#include "posixtest.h"

#define NAMESIZE 50
#define MSGSTR "0123456789"
#define BUFFER 40

#define CHILDPASS 1
#define CHILDFAIL 0

void handler(int signo)
{
	return;
}

int main()
{
       	char qname[NAMESIZE];
        const char *msgptr = MSGSTR;
	int pid;

       	sprintf(qname, "/mq_open_9-2_%d", getpid());

	if ((pid = fork()) == 0) {
        	mqd_t rdwrqueuechild;
		sigset_t mask;
		struct mq_attr attr;
		struct sigaction act;
		char msgrcd[BUFFER];
		int sig;
		unsigned pri;

		/* child here */
		
		/* Set up handler for SIGUSR1 */
        	act.sa_handler = handler;
		act.sa_flags = 0;
        	sigaction(SIGUSR1, &act, NULL);

		/* wait for parent to finish with first queue */
		sigemptyset(&mask);
		sigaddset(&mask, SIGUSR1);
		sigprocmask(SIG_BLOCK,&mask,NULL);
		sigwait(&mask, &sig);

		/* once parent has finished, open next queue */
		attr.mq_msgsize = BUFFER;
		attr.mq_maxmsg = BUFFER;
        	rdwrqueuechild = mq_open(qname, O_RDWR, 
				S_IRUSR | S_IWUSR, &attr);
        	if (rdwrqueuechild == (mqd_t)-1) {
                	perror("mq_open() read only failed");
                	return CHILDFAIL;
        	}
#ifdef DEBUG
		printf("read-write message queue opened in child\n");
#endif

        	if (mq_send(rdwrqueuechild, msgptr, strlen(msgptr)+1, 1) == -1) {
			perror("mq_send() did not return success");
			mq_close(rdwrqueuechild);
			return CHILDFAIL;
		}
#ifdef DEBUG
		printf("Message %s sent in child\n", msgptr);
#endif

        	if (mq_receive(rdwrqueuechild, msgrcd, BUFFER, &pri) == -1) {
			perror("mq_receive() did not return success");
			mq_close(rdwrqueuechild);
                	return CHILDFAIL;
		}
#ifdef DEBUG
		printf("Message %s received in child\n", msgrcd);
#endif

		mq_close(rdwrqueuechild);

		return CHILDPASS;
	} else {
		/* parent here */
        	mqd_t rdwrqueue;
		char msgrcd[BUFFER];
		struct mq_attr attr;
		int i;
		unsigned pri;

		attr.mq_msgsize = BUFFER;
		attr.mq_maxmsg = BUFFER;
        	rdwrqueue = mq_open(qname, O_CREAT |O_RDWR, 
				S_IRUSR | S_IWUSR, &attr);
        	if (rdwrqueue == (mqd_t)-1) {
                	perror("mq_open() did not return success");
			printf("Test UNRESOLVED\n");
			/* kill child and exit */
			kill(pid, SIGABRT);
                	return PTS_UNRESOLVED;
        	}
#ifdef DEBUG
		printf("read-write message queue opened in parent\n");
#endif

        	if (mq_send(rdwrqueue, msgptr, strlen(msgptr), 1) == -1) {
                	perror("mq_send() did not return success");
			printf("Test FAILED\n");
			/* close queue, kill child and exit */
			mq_close(rdwrqueue);
			mq_unlink(qname);
			kill(pid, SIGABRT);
			return PTS_FAIL;
		}
#ifdef DEBUG
		printf("Message %s sent\n", msgptr);
#endif

        	if (mq_receive(rdwrqueue, msgrcd, BUFFER, &pri) == -1) {
                	perror("mq_receive() did not return success");
			printf("Test FAILED\n");
			/* close queue, kill child and exit */
			mq_close(rdwrqueue);
			mq_unlink(qname);
			kill(pid, SIGABRT);
			return PTS_FAIL;
		}
#ifdef DEBUG
		printf("Message %s received\n", msgrcd);
#endif

		sleep(1);
		kill(pid, SIGUSR1); //tell child mq_open and mq_send finished

		if (wait(&i) == -1) {
			perror("Error waiting for child to exit");
			printf("Test UNRESOLVED\n");
			/* close queue and exit */
			mq_close(rdwrqueue);
			mq_unlink(qname);
			return PTS_UNRESOLVED;
		}
#ifdef DEBUG
		printf("Child finished\n");
#endif

		mq_close(rdwrqueue);
		mq_unlink(qname);

                if (!WIFEXITED(i) || !WEXITSTATUS(i)) {
			printf("Test FAILED\n");
			return PTS_FAIL;
		}

        	printf("Test PASSED\n");
        	return PTS_PASS;
	}

	return PTS_UNRESOLVED;
}