1  /* Test all open message queues descriptors are closed during exec*.
       2     Copyright (C) 2004-2023 Free Software Foundation, Inc.
       3     This file is part of the GNU C Library.
       4  
       5     The GNU C Library is free software; you can redistribute it and/or
       6     modify it under the terms of the GNU Lesser General Public
       7     License as published by the Free Software Foundation; either
       8     version 2.1 of the License, or (at your option) any later version.
       9  
      10     The GNU C Library is distributed in the hope that it will be useful,
      11     but WITHOUT ANY WARRANTY; without even the implied warranty of
      12     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
      13     Lesser General Public License for more details.
      14  
      15     You should have received a copy of the GNU Lesser General Public
      16     License along with the GNU C Library; if not, see
      17     <https://www.gnu.org/licenses/>.  */
      18  
      19  #include <errno.h>
      20  #include <fcntl.h>
      21  #include <mqueue.h>
      22  #include <stdio.h>
      23  #include <stdlib.h>
      24  #include <string.h>
      25  #include <sys/wait.h>
      26  #include <time.h>
      27  #include <unistd.h>
      28  #include <support/check.h>
      29  
      30  #define OPT_AFTEREXEC 20000
      31  
      32  static mqd_t after_exec = (mqd_t) -1;
      33  
      34  #define CMDLINE_OPTIONS \
      35    { "after-exec", required_argument, NULL, OPT_AFTEREXEC },
      36  
      37  #define CMDLINE_PROCESS \
      38    case OPT_AFTEREXEC:					\
      39      after_exec = (mqd_t) strtoul (optarg, NULL, 0);	\
      40      break;
      41  
      42  static int
      43  do_after_exec (void)
      44  {
      45    int result = 0;
      46  
      47    struct mq_attr attr;
      48    if (mq_getattr (after_exec, &attr) == 0)
      49      {
      50        puts ("mq_getattr after exec unexpectedly succeeded");
      51        result = 1;
      52      }
      53    else if (errno != EBADF)
      54      {
      55        printf ("mq_getattr after exec did not fail with EBADF: %m\n");
      56        result = 1;
      57      }
      58  
      59    return result;
      60  }
      61  
      62  static int
      63  do_test (int argc, char **argv)
      64  {
      65    if (after_exec != (mqd_t) -1)
      66      return do_after_exec ();
      67  
      68    char name[sizeof "/tst-mqueue7-" + sizeof (pid_t) * 3];
      69    snprintf (name, sizeof (name), "/tst-mqueue7-%u", getpid ());
      70  
      71    struct mq_attr attr = { .mq_maxmsg = 10, .mq_msgsize = 1 };
      72    mqd_t q = mq_open (name, O_CREAT | O_EXCL | O_WRONLY, 0600, &attr);
      73  
      74    if (q == (mqd_t) -1)
      75      {
      76        printf ("mq_open failed with: %m\n");
      77        return 0;
      78      }
      79    else if (mq_unlink (name) != 0)
      80      {
      81        printf ("mq_unlink failed with: %m\n");
      82        return 1;
      83      }
      84  
      85    if (mq_getattr (q, &attr) != 0)
      86      {
      87        printf ("mq_getattr failed: %m\n");
      88        return 1;
      89      }
      90  
      91    char after_exec_arg[sizeof "--after-exec=0x" + sizeof (long) * 3];
      92    snprintf (after_exec_arg, sizeof (after_exec_arg),
      93  	    "--after-exec=0x%lx", (long) q);
      94  
      95    const char *newargv[argc + 2];
      96    for (int i = 1; i < argc; ++i)
      97      newargv[i - 1] = argv[i];
      98    newargv[argc - 1] = "--direct";
      99    newargv[argc] = after_exec_arg;
     100    newargv[argc + 1] = NULL;
     101  
     102    /* Verify that exec* has the effect of mq_close (q).  */
     103    execv (newargv[0], (char * const *) newargv);
     104    printf ("execv failed: %m\n");
     105    return 1;
     106  }
     107  
     108  #include "../test-skeleton.c"