diff options
Diffstat (limited to 'test/nptl/tst-mqueue1.c')
| -rw-r--r-- | test/nptl/tst-mqueue1.c | 417 | 
1 files changed, 417 insertions, 0 deletions
| diff --git a/test/nptl/tst-mqueue1.c b/test/nptl/tst-mqueue1.c new file mode 100644 index 000000000..70e42377b --- /dev/null +++ b/test/nptl/tst-mqueue1.c @@ -0,0 +1,417 @@ +/* Test message queue passing. +   Copyright (C) 2004 Free Software Foundation, Inc. +   This file is part of the GNU C Library. +   Contributed by Jakub Jelinek <jakub@redhat.com>, 2004. + +   The GNU C Library is free software; you can redistribute it and/or +   modify it under the terms of the GNU Lesser General Public +   License as published by the Free Software Foundation; either +   version 2.1 of the License, or (at your option) any later version. + +   The GNU C Library is distributed in the hope that it will be useful, +   but WITHOUT ANY WARRANTY; without even the implied warranty of +   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU +   Lesser General Public License for more details. + +   You should have received a copy of the GNU Lesser General Public +   License along with the GNU C Library; if not, write to the Free +   Software Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA +   02111-1307 USA.  */ + +#include <errno.h> +#include <fcntl.h> +#include <mqueue.h> +#include <stdio.h> +#include <stdlib.h> +#include <string.h> +#include <sys/wait.h> +#include <time.h> +#include <unistd.h> +#include "tst-mqueue.h" + +static int +intcmp (const void *a, const void *b) +{ +  if (*(unsigned char *)a < *(unsigned char *)b) +    return 1; +  if (*(unsigned char *)a > *(unsigned char *)b) +    return -1; +  return 0; +} + +static int +check_attrs (struct mq_attr *attr, int nonblock, long cnt) +{ +  int result = 0; + +  if (attr->mq_maxmsg != 10 || attr->mq_msgsize != 1) +    { +      printf ("attributes don't match those passed to mq_open\n" +	      "mq_maxmsg %ld, mq_msgsize %ld\n", +	      attr->mq_maxmsg, attr->mq_msgsize); +      result = 1; +    } + +  if ((attr->mq_flags & O_NONBLOCK) != nonblock) +    { +      printf ("mq_flags %lx != %x\n", (attr->mq_flags & O_NONBLOCK), nonblock); +      result = 1; +    } + +  if (attr->mq_curmsgs != cnt) +    { +      printf ("mq_curmsgs %ld != %ld\n", attr->mq_curmsgs, cnt); +      result = 1; +    } + +  return result; +} + +static int +do_one_test (mqd_t q, const char *name, int nonblock) +{ +  int result = 0; + +  char v [] +    = { 0x32, 0x62, 0x22, 0x31, 0x11, 0x73, 0x61, 0x21, 0x72, 0x71, 0x81 }; + +  struct mq_attr attr; +  memset (&attr, 0xaa, sizeof (attr)); +  if (mq_getattr (q, &attr) != 0) +    { +      printf ("mq_getattr failed: %m\n"); +      result = 1; +    } +  else +    result |= check_attrs (&attr, nonblock, 0); + +  if (mq_receive (q, &v[0], 1, NULL) != -1) +    { +      puts ("mq_receive on O_WRONLY mqd_t unexpectedly succeeded"); +      result = 1; +    } +  else if (errno != EBADF) +    { +      printf ("mq_receive on O_WRONLY mqd_t did not fail with EBADF: %m\n"); +      result = 1; +    } + +  struct timespec ts; +  if (clock_gettime (CLOCK_REALTIME, &ts) == 0) +    --ts.tv_sec; +  else +    { +      ts.tv_sec = time (NULL) - 1; +      ts.tv_nsec = 0; +    } + +  int ret; +  for (int i = 0; i < 10; ++i) +    { +      if (i & 1) +	ret = mq_send (q, &v[i], 1, v[i] >> 4); +      else +	ret = mq_timedsend (q, &v[i], 1, v[i] >> 4, &ts); + +      if (ret) +	{ +	  printf ("mq_%ssend failed: %m\n", (i & 1) ? "" : "timed"); +	  result = 1; +	} +    } + +  ret = mq_timedsend (q, &v[10], 1, 8, &ts); +  if (ret != -1) +    { +      puts ("mq_timedsend on full queue did not fail"); +      result = 1; +    } +  else if (errno != (nonblock ? EAGAIN : ETIMEDOUT)) +    { +      printf ("mq_timedsend on full queue did not fail with %s: %m\n", +	      nonblock ? "EAGAIN" : "ETIMEDOUT"); +      result = 1; +    } + +  if (nonblock) +    { +      ret = mq_send (q, &v[10], 1, 8); +      if (ret != -1) +	{ +	  puts ("mq_send on full non-blocking queue did not fail"); +	  result = 1; +	} +      else if (errno != EAGAIN) +	{ +	  printf ("mq_send on full non-blocking queue did not fail" +		  "with EAGAIN: %m\n"); +	  result = 1; +	} +    } + +  memset (&attr, 0xaa, sizeof (attr)); +  if (mq_getattr (q, &attr) != 0) +    { +      printf ("mq_getattr failed: %m\n"); +      result = 1; +    } +  else +    result |= check_attrs (&attr, nonblock, 10); + +  pid_t pid = fork (); +  if (pid == -1) +    { +      printf ("fork failed: %m\n"); +      result = 1; +    } +  else if (pid == 0) +    { +      result = 0; + +      if (mq_close (q) != 0) +	{ +	  printf ("mq_close in child failed: %m\n"); +	  result = 1; +	} + +      q = mq_open (name, O_RDONLY | nonblock); +      if (q == (mqd_t) -1) +        { +	  printf ("mq_open in child failed: %m\n"); +	  exit (1); +        } + +      memset (&attr, 0xaa, sizeof (attr)); +      if (mq_getattr (q, &attr) != 0) +	{ +	  printf ("mq_getattr failed: %m\n"); +	  result = 1; +	} +      else +	result |= check_attrs (&attr, nonblock, 10); + +      char vr[11] = { }; +      unsigned int prio; +      ssize_t rets; + +      if (mq_send (q, &v[0], 1, 1) != -1) +	{ +	  puts ("mq_send on O_RDONLY mqd_t unexpectedly succeeded"); +	  result = 1; +	} +      else if (errno != EBADF) +	{ +	  printf ("mq_send on O_WRONLY mqd_t did not fail with EBADF: %m\n"); +	  result = 1; +	} + +      for (int i = 0; i < 10; ++i) +	{ +	  if (i & 1) +	    rets = mq_receive (q, &vr[i], 1, &prio); +	  else +	    rets = mq_timedreceive (q, &vr[i], 1, &prio, &ts); + +	  if (rets != 1) +	    { +	      if (rets == -1) +		printf ("mq_%sreceive failed: %m\n", (i & 1) ? "" : "timed"); +	      else +		printf ("mq_%sreceive returned %zd != 1\n", +			(i & 1) ? "" : "timed", rets); +	      result = 1; +	    } +	  else if (prio != (unsigned int) vr[i] >> 4) +	    { +	      printf ("unexpected priority %x for value %02x\n", prio, +		      vr[i]); +	      result = 1; +	    } +	} + +      qsort (v, 10, 1, intcmp); +      if (memcmp (v, vr, 10) != 0) +	{ +	  puts ("messages not received in expected order"); +	  result = 1; +	} + +      rets = mq_timedreceive (q, &vr[10], 1, &prio, &ts); +      if (rets != -1) +	{ +	  puts ("mq_timedreceive on empty queue did not fail"); +	  result = 1; +	} +      else if (errno != (nonblock ? EAGAIN : ETIMEDOUT)) +	{ +	  printf ("mq_timedreceive on empty queue did not fail with %s: %m\n", +		  nonblock ? "EAGAIN" : "ETIMEDOUT"); +	  result = 1; +	} + +      if (nonblock) +	{ +	  ret = mq_receive (q, &vr[10], 1, &prio); +	  if (ret != -1) +	    { +	      puts ("mq_receive on empty non-blocking queue did not fail"); +	      result = 1; +	    } +	  else if (errno != EAGAIN) +	    { +	      printf ("mq_receive on empty non-blocking queue did not fail" +		      "with EAGAIN: %m\n"); +	      result = 1; +	    } +	} + +      memset (&attr, 0xaa, sizeof (attr)); +      if (mq_getattr (q, &attr) != 0) +	{ +	  printf ("mq_getattr failed: %m\n"); +	  result = 1; +	} +      else +	result |= check_attrs (&attr, nonblock, 0); + +      if (mq_close (q) != 0) +	{ +	  printf ("mq_close in child failed: %m\n"); +	  result = 1; +	} + +      exit (result); +    } + +  int status; +  if (TEMP_FAILURE_RETRY (waitpid (pid, &status, 0)) != pid) +    { +      printf ("waitpid failed: %m\n"); +      kill (pid, SIGKILL); +      result = 1; +    } +  else if (!WIFEXITED (status) || WEXITSTATUS (status)) +    { +      printf ("child failed: %d\n", status); +      result = 1; +    } + +  memset (&attr, 0xaa, sizeof (attr)); +  if (mq_getattr (q, &attr) != 0) +    { +      printf ("mq_getattr failed: %m\n"); +      result = 1; +    } +  else +    result |= check_attrs (&attr, nonblock, 0); + +  return result; +} + +#define TEST_FUNCTION do_test () +static int +do_test (void) +{ +  int result = 0; + +  char name[sizeof "/tst-mqueue1-" + sizeof (pid_t) * 3]; +  snprintf (name, sizeof (name), "/tst-mqueue1-%u", getpid ()); + +  struct mq_attr attr = { .mq_maxmsg = 10, .mq_msgsize = 1 }; +  mqd_t q = mq_open (name, O_CREAT | O_EXCL | O_WRONLY, 0600, &attr); + +  if (q == (mqd_t) -1) +    { +      printf ("mq_open failed with: %m\n"); +      return result; +    } +  else +    add_temp_mq (name); + +  result |= do_one_test (q, name, 0); + +  mqd_t q2 = mq_open (name, O_WRONLY | O_NONBLOCK); +  if (q2 == (mqd_t) -1) +    { +      printf ("mq_open failed with: %m\n"); +      q2 = q; +      result = 1; +    } +  else +    { +      if (mq_close (q) != 0) +	{ +	  printf ("mq_close in parent failed: %m\n"); +	  result = 1; +	} + +      q = q2; +      result |= do_one_test (q, name, O_NONBLOCK); + +      if (mq_getattr (q, &attr) != 0) +	{ +	  printf ("mq_getattr failed: %m\n"); +	  result = 1; +	} +      else +	{ +	  attr.mq_flags ^= O_NONBLOCK; + +	  struct mq_attr attr2; +	  memset (&attr2, 0x55, sizeof (attr2)); +	  if (mq_setattr (q, &attr, &attr2) != 0) +	    { +	      printf ("mq_setattr failed: %m\n"); +	      result = 1; +	    } +	  else if (attr.mq_flags != (attr2.mq_flags ^ O_NONBLOCK) +		   || attr.mq_maxmsg != attr2.mq_maxmsg +		   || attr.mq_msgsize != attr2.mq_msgsize +		   || attr.mq_curmsgs != 0 +		   || attr2.mq_curmsgs != 0) +	    { +	      puts ("mq_setattr returned unexpected values in *omqstat"); +	      result = 1; +	    } +	  else +	    { +	      result |= do_one_test (q, name, 0); + +	      if (mq_setattr (q, &attr2, NULL) != 0) +		{ +		  printf ("mq_setattr failed: %m\n"); +		  result = 1; +		} +	      else +		result |= do_one_test (q, name, O_NONBLOCK); +	    } +	} +    } + +  if (mq_unlink (name) != 0) +    { +      printf ("mq_unlink failed: %m\n"); +      result = 1; +    } + +  if (mq_close (q) != 0) +    { +      printf ("mq_close in parent failed: %m\n"); +      result = 1; +    } + +  if (mq_close (q) != -1) +    { +      puts ("second mq_close did not fail"); +      result = 1; +    } +  else if (errno != EBADF) +    { +      printf ("second mq_close did not fail with EBADF: %m\n"); +      result = 1; +    } + +  return result; +} + +#include "../test-skeleton.c" | 
