/************************************************************************** * apps/examples/ostest/mqueue.c * * Copyright (C) 2007-2009, 2011 Gregory Nutt. All rights reserved. * Author: Gregory Nutt * * Redistribution and use in source and binary forms, with or without * modification, are permitted provided that the following conditions * are met: * * 1. Redistributions of source code must retain the above copyright * notice, this list of conditions and the following disclaimer. * 2. Redistributions in binary form must reproduce the above copyright * notice, this list of conditions and the following disclaimer in * the documentation and/or other materials provided with the * distribution. * 3. Neither the name NuttX nor the names of its contributors may be * used to endorse or promote products derived from this software * without specific prior written permission. * * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE * COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS * OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN * ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE * POSSIBILITY OF SUCH DAMAGE. * **************************************************************************/ /************************************************************************** * Included Files **************************************************************************/ #include #include #include #include #include #include #include #include #include #include #include "ostest.h" /************************************************************************** * Private Definitions **************************************************************************/ #define TEST_MESSAGE "This is a test and only a test" #if defined(SDCC) || defined(__ZILOG__) /* Cannot use strlen in array size */ # define TEST_MSGLEN (31) #else /* Message lenght is the size of the message plus the null terminator */ # define TEST_MSGLEN (strlen(TEST_MESSAGE)+1) #endif #define TEST_SEND_NMSGS (10) #define TEST_RECEIVE_NMSGS (10) #if CONFIG_NFILE_DESCRIPTORS > 0 && CONFIG_NFILE_STREAMS > 0 # define FFLUSH() fflush(stdout) #else # define FFLUSH() #endif /************************************************************************** * Private Types **************************************************************************/ /************************************************************************** * Private Function Prototypes **************************************************************************/ /************************************************************************** * Global Variables **************************************************************************/ /************************************************************************** * Private Variables **************************************************************************/ /************************************************************************** * Private Functions **************************************************************************/ /************************************************************************** * Public Functions **************************************************************************/ static void *sender_thread(void *arg) { mqd_t mqfd; char msg_buffer[TEST_MSGLEN]; struct mq_attr attr; int status = 0; int nerrors = 0; int i; printf("sender_thread: Starting\n"); /* Fill in attributes for message queue */ attr.mq_maxmsg = TEST_SEND_NMSGS-1; attr.mq_msgsize = TEST_MSGLEN; attr.mq_flags = 0; /* Set the flags for the open of the queue. * Make it a blocking open on the queue, meaning it will block if * this process tries to send to the queue and the queue is full. * * O_CREAT - the queue will get created if it does not already exist. * O_WRONLY - we are only planning to write to the queue. * * Open the queue, and create it if the receiving process hasn't * already created it. */ mqfd = mq_open("testmq", O_WRONLY|O_CREAT, 0666, &attr); if (mqfd < 0) { printf("sender_thread: ERROR mq_open failed\n"); pthread_exit((pthread_addr_t)1); } /* Fill in a test message buffer to send */ memcpy(msg_buffer, TEST_MESSAGE, TEST_MSGLEN); /* Perform the send TEST_SEND_NMSGS times */ for (i = 0; i < TEST_SEND_NMSGS; i++) { struct timespec time; status = clock_gettime(CLOCK_REALTIME, &time); if (status != 0) { printf("sender_thread: ERROR clock_gettime failed\n"); } time.tv_sec += 5; /* The first TEST_SEND_NMSGS-1 send should succeed. The last * one should fail with errno == ETIMEDOUT */ status = mq_timedsend(mqfd, msg_buffer, TEST_MSGLEN, 42, &time); if (status < 0) { if (i == TEST_SEND_NMSGS-1 && errno == ETIMEDOUT) { printf("sender_thread: mq_timedsend %d timed out as expected\n", i); } else { printf("sender_thread: ERROR mq_timedsend failure=%d on msg %d\n", errno, i); nerrors++; } } else { if (i == TEST_SEND_NMSGS-1) { printf("sender_thread: ERROR mq_timedsend of msg %d succeeded\n", i); nerrors++; } else { printf("sender_thread: mq_timedsend succeeded on msg %d\n", i); } } } /* Close the queue and return success */ if (mq_close(mqfd) < 0) { printf("sender_thread: ERROR mq_close failed\n"); } printf("sender_thread: returning nerrors=%d\n", nerrors); FFLUSH(); return (pthread_addr_t)nerrors; } static void *receiver_thread(void *arg) { mqd_t mqfd; char msg_buffer[TEST_MSGLEN]; struct mq_attr attr; int nbytes; int nerrors = 0; int i; printf("receiver_thread: Starting\n"); /* Fill in attributes for message queue */ attr.mq_maxmsg = TEST_SEND_NMSGS-1; attr.mq_msgsize = TEST_MSGLEN; attr.mq_flags = 0; /* Set the flags for the open of the queue. * Make it a blocking open on the queue, meaning it will block if * this process tries to* send to the queue and the queue is full. * * O_CREAT - the queue will get created if it does not already exist. * O_RDONLY - we are only planning to write to the queue. * * Open the queue, and create it if the sending process hasn't * already created it. */ mqfd = mq_open("testmq", O_RDONLY|O_CREAT, 0666, &attr); if (mqfd < 0) { printf("receiver_thread: ERROR mq_open failed\n"); pthread_exit((pthread_addr_t)1); } /* Perform the receive TEST_RECEIVE_NMSGS times */ for (i = 0; i < TEST_RECEIVE_NMSGS; i++) { struct timespec time; int status = clock_gettime(CLOCK_REALTIME, &time); if (status != 0) { printf("sender_thread: ERROR clock_gettime failed\n"); } time.tv_sec += 5; /* The first TEST_SEND_NMSGS-1 send should succeed. The last * one should fail with errno == ETIMEDOUT */ memset(msg_buffer, 0xaa, TEST_MSGLEN); nbytes = mq_timedreceive(mqfd, msg_buffer, TEST_MSGLEN, 0, &time); if (nbytes < 0) { if (i == TEST_SEND_NMSGS-1 && errno == ETIMEDOUT) { printf("receiver_thread: Receive %d timed out as expected\n", i); } else { printf("receiver_thread: ERROR mq_timedreceive failure=%d on msg %d\n", errno, i); nerrors++; } } else if (nbytes != TEST_MSGLEN) { printf("receiver_thread: mq_timedreceive return bad size %d on msg %d\n", nbytes, i); nerrors++; } else if (memcmp(TEST_MESSAGE, msg_buffer, nbytes) != 0) { int j; printf("receiver_thread: mq_timedreceive returned corrupt message on msg %d\n", i); printf("receiver_thread: i Expected Received\n"); for (j = 0; j < TEST_MSGLEN-1; j++) { if (isprint(msg_buffer[j])) { printf("receiver_thread: %2d %02x (%c) %02x (%c)\n", j, TEST_MESSAGE[j], TEST_MESSAGE[j], msg_buffer[j], msg_buffer[j]); } else { printf("receiver_thread: %2d %02x (%c) %02x\n", j, TEST_MESSAGE[j], TEST_MESSAGE[j], msg_buffer[j]); } } printf("receiver_thread: %2d 00 %02x\n", j, msg_buffer[j]); } else if (i == TEST_SEND_NMSGS-1) { printf("receiver_thread: ERROR mq_timedreceive of msg %d succeeded\n", i); nerrors++; } else { printf("receiver_thread: mq_timedreceive succeeded on msg %d\n", i); } } /* Close the queue and return success */ if (mq_close(mqfd) < 0) { printf("receiver_thread: ERROR mq_close failed\n"); nerrors++; } /* Destroy the queue */ if (mq_unlink("testmq") < 0) { printf("receiver_thread: ERROR mq_close failed\n"); nerrors++; } printf("receiver_thread: returning nerrors=%d\n", nerrors); FFLUSH(); pthread_exit((pthread_addr_t)nerrors); return (pthread_addr_t)nerrors; } void timedmqueue_test(void) { pthread_t sender; pthread_t receiver; void *result; pthread_attr_t attr; int status; /* Start the sending thread at the default priority */ printf("timedmqueue_test: Starting sender\n"); status = pthread_attr_init(&attr); if (status != 0) { printf("timedmqueue_test: pthread_attr_init failed, status=%d\n", status); } status = pthread_attr_setstacksize(&attr, STACKSIZE); if (status != 0) { printf("timedmqueue_test: pthread_attr_setstacksize failed, status=%d\n", status); } status = pthread_create(&sender, &attr, sender_thread, NULL); if (status != 0) { printf("timedmqueue_test: pthread_create failed, status=%d\n", status); } /* Wait for the sending thread to complete */ printf("timedmqueue_test: Waiting for sender to complete\n"); pthread_join(sender, &result); if (result != (void*)0) { printf("timedmqueue_test: ERROR sender thread exited with %d errors\n", (int)result); } /* Start the receiving thread at the default priority */ printf("timedmqueue_test: Starting receiver\n"); status = pthread_attr_init(&attr); if (status != 0) { printf("timedmqueue_test: pthread_attr_init failed, status=%d\n", status); } status = pthread_attr_setstacksize(&attr, STACKSIZE); if (status != 0) { printf("timedmqueue_test: pthread_attr_setstacksize failed, status=%d\n", status); } status = pthread_create(&receiver, &attr, receiver_thread, NULL); if (status != 0) { printf("timedmqueue_test: pthread_create failed, status=%d\n", status); } /* Wait for the receiving thread to complete */ printf("timedmqueue_test: Waiting for receiver to complete\n"); pthread_join(receiver, &result); if (result != (void*)0) { printf("timedmqueue_test: ERROR receiver thread exited with %d errors\n", (int)result); } printf("timedmqueue_test: Test complete\n"); }