rt/tst-mqueue1.c

   1 /* Test message queue passing.
   2    Copyright (C) 2004-2015 Free Software Foundation, Inc.
   3    This file is part of the GNU C Library.
   4    Contributed by Jakub Jelinek <jakub@redhat.com>, 2004.
   5
   6    The GNU C Library is free software; you can redistribute it and/or
   7    modify it under the terms of the GNU Lesser General Public
   8    License as published by the Free Software Foundation; either
   9    version 2.1 of the License, or (at your option) any later version.
  10
  11    The GNU C Library is distributed in the hope that it will be useful,
  12    but WITHOUT ANY WARRANTY; without even the implied warranty of
  13    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14    Lesser General Public License for more details.
  15
  16    You should have received a copy of the GNU Lesser General Public
  17    License along with the GNU C Library; if not, see
  18    <http://www.gnu.org/licenses/>.  */
  19
  20 #include <errno.h>
  21 #include <fcntl.h>
  22 #include <mqueue.h>
  23 #include <stdio.h>
  24 #include <stdlib.h>
  25 #include <string.h>
  26 #include <sys/wait.h>
  27 #include <time.h>
  28 #include <unistd.h>
  29 #include <stdint.h>
  30 #include "tst-mqueue.h"
  31
  32 static int
  33 intcmp (const void *a, const void *b)
  34 {
  35   if (*(unsigned char *)a < *(unsigned char *)b)
  36     return 1;
  37   if (*(unsigned char *)a > *(unsigned char *)b)
  38     return -1;
  39   return 0;
  40 }
  41
  42 static int
  43 check_attrs (struct mq_attr *attr, int nonblock, long cnt)
  44 {
  45   int result = 0;
  46
  47   if (attr->mq_maxmsg != 10 || attr->mq_msgsize != 1)
  48     {
  49       printf ("attributes don't match those passed to mq_open\n"
  50               "mq_maxmsg %jd, mq_msgsize %jd\n",
  51               (intmax_t) attr->mq_maxmsg, (intmax_t) attr->mq_msgsize);
  52       result = 1;
  53     }
  54
  55   if ((attr->mq_flags & O_NONBLOCK) != nonblock)
  56     {
  57       printf ("mq_flags %jx != %x\n",
  58               (intmax_t) (attr->mq_flags & O_NONBLOCK), nonblock);
  59       result = 1;
  60     }
  61
  62   if (attr->mq_curmsgs != cnt)
  63     {
  64       printf ("mq_curmsgs %jd != %ld\n", (intmax_t) attr->mq_curmsgs, cnt);
  65       result = 1;
  66     }
  67
  68   return result;
  69 }
  70
  71 static int
  72 do_one_test (mqd_t q, const char *name, int nonblock)
  73 {
  74   int result = 0;
  75
  76   unsigned char v []
  77     = { 0x32, 0x62, 0x22, 0x31, 0x11, 0x73, 0x61, 0x21, 0x72, 0x71, 0x81 };
  78
  79   struct mq_attr attr;
  80   memset (&attr, 0xaa, sizeof (attr));
  81   if (mq_getattr (q, &attr) != 0)
  82     {
  83       printf ("mq_getattr failed: %m\n");
  84       result = 1;
  85     }
  86   else
  87     result |= check_attrs (&attr, nonblock, 0);
  88
  89   if (mq_receive (q, (char *) &v[0], 1, NULL) != -1)
  90     {
  91       puts ("mq_receive on O_WRONLY mqd_t unexpectedly succeeded");
  92       result = 1;
  93     }
  94   else if (errno != EBADF)
  95     {
  96       printf ("mq_receive on O_WRONLY mqd_t did not fail with EBADF: %m\n");
  97       result = 1;
  98     }
  99
 100   struct timespec ts;
 101   if (clock_gettime (CLOCK_REALTIME, &ts) == 0)
 102     --ts.tv_sec;
 103   else
 104     {
 105       ts.tv_sec = time (NULL) - 1;
 106       ts.tv_nsec = 0;
 107     }
 108
 109   int ret;
 110   for (int i = 0; i < 10; ++i)
 111     {
 112       if (i & 1)
 113         ret = mq_send (q, (char *) &v[i], 1, v[i] >> 4);
 114       else
 115         ret = mq_timedsend (q, (char *) &v[i], 1, v[i] >> 4, &ts);
 116
 117       if (ret)
 118         {
 119           printf ("mq_%ssend failed: %m\n", (i & 1) ? "" : "timed");
 120           result = 1;
 121         }
 122     }
 123
 124   ret = mq_timedsend (q, (char *) &v[10], 1, 8, &ts);
 125   if (ret != -1)
 126     {
 127       puts ("mq_timedsend on full queue did not fail");
 128       result = 1;
 129     }
 130   else if (errno != (nonblock ? EAGAIN : ETIMEDOUT))
 131     {
 132       printf ("mq_timedsend on full queue did not fail with %s: %m\n",
 133               nonblock ? "EAGAIN" : "ETIMEDOUT");
 134       result = 1;
 135     }
 136
 137   if (nonblock)
 138     {
 139       ret = mq_send (q, (char *) &v[10], 1, 8);
 140       if (ret != -1)
 141         {
 142           puts ("mq_send on full non-blocking queue did not fail");
 143           result = 1;
 144         }
 145       else if (errno != EAGAIN)
 146         {
 147           printf ("mq_send on full non-blocking queue did not fail"
 148                   "with EAGAIN: %m\n");
 149           result = 1;
 150         }
 151     }
 152
 153   memset (&attr, 0xaa, sizeof (attr));
 154   if (mq_getattr (q, &attr) != 0)
 155     {
 156       printf ("mq_getattr failed: %m\n");
 157       result = 1;
 158     }
 159   else
 160     result |= check_attrs (&attr, nonblock, 10);
 161
 162   pid_t pid = fork ();
 163   if (pid == -1)
 164     {
 165       printf ("fork failed: %m\n");
 166       result = 1;
 167     }
 168   else if (pid == 0)
 169     {
 170       result = 0;
 171
 172       if (mq_close (q) != 0)
 173         {
 174           printf ("mq_close in child failed: %m\n");
 175           result = 1;
 176         }
 177
 178       q = mq_open (name, O_RDONLY | nonblock);
 179       if (q == (mqd_t) -1)
 180         {
 181           printf ("mq_open in child failed: %m\n");
 182           exit (1);
 183         }
 184
 185       memset (&attr, 0xaa, sizeof (attr));
 186       if (mq_getattr (q, &attr) != 0)
 187         {
 188           printf ("mq_getattr failed: %m\n");
 189           result = 1;
 190         }
 191       else
 192         result |= check_attrs (&attr, nonblock, 10);
 193
 194       unsigned char vr[11] = { };
 195       unsigned int prio;
 196       ssize_t rets;
 197
 198       if (mq_send (q, (char *) &v[0], 1, 1) != -1)
 199         {
 200           puts ("mq_send on O_RDONLY mqd_t unexpectedly succeeded");
 201           result = 1;
 202         }
 203       else if (errno != EBADF)
 204         {
 205           printf ("mq_send on O_WRONLY mqd_t did not fail with EBADF: %m\n");
 206           result = 1;
 207         }
 208
 209       for (int i = 0; i < 10; ++i)
 210         {
 211           if (i & 1)
 212             rets = mq_receive (q, (char *) &vr[i], 1, &prio);
 213           else
 214             rets = mq_timedreceive (q, (char *) &vr[i], 1, &prio, &ts);
 215
 216           if (rets != 1)
 217             {
 218               if (rets == -1)
 219                 printf ("mq_%sreceive failed: %m\n", (i & 1) ? "" : "timed");
 220               else
 221                 printf ("mq_%sreceive returned %zd != 1\n",
 222                         (i & 1) ? "" : "timed", rets);
 223               result = 1;
 224             }
 225           else if (prio != (unsigned int) vr[i] >> 4)
 226             {
 227               printf ("unexpected priority %x for value %02x\n", prio,
 228                       vr[i]);
 229               result = 1;
 230             }
 231         }
 232
 233       qsort (v, 10, 1, intcmp);
 234       if (memcmp (v, vr, 10) != 0)
 235         {
 236           puts ("messages not received in expected order");
 237           result = 1;
 238         }
 239
 240       rets = mq_timedreceive (q, (char *) &vr[10], 1, &prio, &ts);
 241       if (rets != -1)
 242         {
 243           puts ("mq_timedreceive on empty queue did not fail");
 244           result = 1;
 245         }
 246       else if (errno != (nonblock ? EAGAIN : ETIMEDOUT))
 247         {
 248           printf ("mq_timedreceive on empty queue did not fail with %s: %m\n",
 249                   nonblock ? "EAGAIN" : "ETIMEDOUT");
 250           result = 1;
 251         }
 252
 253       if (nonblock)
 254         {
 255           ret = mq_receive (q, (char *) &vr[10], 1, &prio);
 256           if (ret != -1)
 257             {
 258               puts ("mq_receive on empty non-blocking queue did not fail");
 259               result = 1;
 260             }
 261           else if (errno != EAGAIN)
 262             {
 263               printf ("mq_receive on empty non-blocking queue did not fail"
 264                       "with EAGAIN: %m\n");
 265               result = 1;
 266             }
 267         }
 268
 269       memset (&attr, 0xaa, sizeof (attr));
 270       if (mq_getattr (q, &attr) != 0)
 271         {
 272           printf ("mq_getattr failed: %m\n");
 273           result = 1;
 274         }
 275       else
 276         result |= check_attrs (&attr, nonblock, 0);
 277
 278       if (mq_close (q) != 0)
 279         {
 280           printf ("mq_close in child failed: %m\n");
 281           result = 1;
 282         }
 283
 284       exit (result);
 285     }
 286
 287   int status;
 288   if (TEMP_FAILURE_RETRY (waitpid (pid, &status, 0)) != pid)
 289     {
 290       printf ("waitpid failed: %m\n");
 291       kill (pid, SIGKILL);
 292       result = 1;
 293     }
 294   else if (!WIFEXITED (status) || WEXITSTATUS (status))
 295     {
 296       printf ("child failed: %d\n", status);
 297       result = 1;
 298     }
 299
 300   memset (&attr, 0xaa, sizeof (attr));
 301   if (mq_getattr (q, &attr) != 0)
 302     {
 303       printf ("mq_getattr failed: %m\n");
 304       result = 1;
 305     }
 306   else
 307     result |= check_attrs (&attr, nonblock, 0);
 308
 309   return result;
 310 }
 311
 312 #define TEST_FUNCTION do_test ()
 313 static int
 314 do_test (void)
 315 {
 316   int result = 0;
 317
 318   char name[sizeof "/tst-mqueue1-" + sizeof (pid_t) * 3];
 319   snprintf (name, sizeof (name), "/tst-mqueue1-%u", getpid ());
 320
 321   struct mq_attr attr = { .mq_maxmsg = 10, .mq_msgsize = 1 };
 322   mqd_t q = mq_open (name, O_CREAT | O_EXCL | O_WRONLY, 0600, &attr);
 323
 324   if (q == (mqd_t) -1)
 325     {
 326       printf ("mq_open failed with: %m\n");
 327       return result;
 328     }
 329   else
 330     add_temp_mq (name);
 331
 332   result |= do_one_test (q, name, 0);
 333
 334   mqd_t q2 = mq_open (name, O_WRONLY | O_NONBLOCK);
 335   if (q2 == (mqd_t) -1)
 336     {
 337       printf ("mq_open failed with: %m\n");
 338       q2 = q;
 339       result = 1;
 340     }
 341   else
 342     {
 343       if (mq_close (q) != 0)
 344         {
 345           printf ("mq_close in parent failed: %m\n");
 346           result = 1;
 347         }
 348
 349       q = q2;
 350       result |= do_one_test (q, name, O_NONBLOCK);
 351
 352       if (mq_getattr (q, &attr) != 0)
 353         {
 354           printf ("mq_getattr failed: %m\n");
 355           result = 1;
 356         }
 357       else
 358         {
 359           attr.mq_flags ^= O_NONBLOCK;
 360
 361           struct mq_attr attr2;
 362           memset (&attr2, 0x55, sizeof (attr2));
 363           if (mq_setattr (q, &attr, &attr2) != 0)
 364             {
 365               printf ("mq_setattr failed: %m\n");
 366               result = 1;
 367             }
 368           else if (attr.mq_flags != (attr2.mq_flags ^ O_NONBLOCK)
 369                    || attr.mq_maxmsg != attr2.mq_maxmsg
 370                    || attr.mq_msgsize != attr2.mq_msgsize
 371                    || attr.mq_curmsgs != 0
 372                    || attr2.mq_curmsgs != 0)
 373             {
 374               puts ("mq_setattr returned unexpected values in *omqstat");
 375               result = 1;
 376             }
 377           else
 378             {
 379               result |= do_one_test (q, name, 0);
 380
 381               if (mq_setattr (q, &attr2, NULL) != 0)
 382                 {
 383                   printf ("mq_setattr failed: %m\n");
 384                   result = 1;
 385                 }
 386               else
 387                 result |= do_one_test (q, name, O_NONBLOCK);
 388             }
 389         }
 390     }
 391
 392   if (mq_unlink (name) != 0)
 393     {
 394       printf ("mq_unlink failed: %m\n");
 395       result = 1;
 396     }
 397
 398   if (mq_close (q) != 0)
 399     {
 400       printf ("mq_close in parent failed: %m\n");
 401       result = 1;
 402     }
 403
 404   if (mq_close (q) != -1)
 405     {
 406       puts ("second mq_close did not fail");
 407       result = 1;
 408     }
 409   else if (errno != EBADF)
 410     {
 411       printf ("second mq_close did not fail with EBADF: %m\n");
 412       result = 1;
 413     }
 414
 415   return result;
 416 }
 417
 418 #include "../test-skeleton.c"