nptl/tst-cancel7.c

   1 /* Copyright (C) 2002-2023 Free Software Foundation, Inc.
   2    This file is part of the GNU C Library.
   3
   4    The GNU C Library is free software; you can redistribute it and/or
   5    modify it under the terms of the GNU Lesser General Public
   6    License as published by the Free Software Foundation; either
   7    version 2.1 of the License, or (at your option) any later version.
   8
   9    The GNU C Library is distributed in the hope that it will be useful,
  10    but WITHOUT ANY WARRANTY; without even the implied warranty of
  11    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  12    Lesser General Public License for more details.
  13
  14    You should have received a copy of the GNU Lesser General Public
  15    License along with the GNU C Library; if not, see
  16    <https://www.gnu.org/licenses/>.  */
  17
  18 #include <errno.h>
  19 #include <fcntl.h>
  20 #include <getopt.h>
  21 #include <signal.h>
  22 #include <stdlib.h>
  23 #include <semaphore.h>
  24 #include <sys/mman.h>
  25
  26 #include <support/check.h>
  27 #include <support/support.h>
  28 #include <support/temp_file.h>
  29 #include <support/xstdio.h>
  30 #include <support/xunistd.h>
  31 #include <support/xthread.h>
  32
  33 static const char *command;
  34 static const char *pidfile;
  35 static const char *semfile;
  36 static char *pidfilename;
  37 static char *semfilename;
  38
  39 static sem_t *sem;
  40
  41 static void *
  42 tf (void *arg)
  43 {
  44   char *cmd = xasprintf ("%s --direct --sem %s --pidfile %s",
  45                          command, semfilename, pidfilename);
  46   if (system (cmd))
  47     FAIL_EXIT1("system call unexpectedly returned");
  48   /* This call should never return.  */
  49   return NULL;
  50 }
  51
  52 static void
  53 sl (void)
  54 {
  55   FILE *f = xfopen (pidfile, "w");
  56
  57   fprintf (f, "%lld\n", (long long) getpid ());
  58   fflush (f);
  59
  60   if (sem_post (sem) != 0)
  61     FAIL_EXIT1 ("sem_post: %m");
  62
  63   struct flock fl =
  64     {
  65       .l_type = F_WRLCK,
  66       .l_start = 0,
  67       .l_whence = SEEK_SET,
  68       .l_len = 1
  69     };
  70   if (fcntl (fileno (f), F_SETLK, &fl) != 0)
  71     FAIL_EXIT1 ("fcntl (F_SETFL): %m");
  72
  73   sigset_t ss;
  74   sigfillset (&ss);
  75   sigsuspend (&ss);
  76   exit (0);
  77 }
  78
  79
  80 static void
  81 do_prepare (int argc, char *argv[])
  82 {
  83   int semfd;
  84   if (semfile == NULL)
  85     semfd = create_temp_file ("tst-cancel7.", &semfilename);
  86   else
  87     semfd = open (semfile, O_RDWR);
  88   TEST_VERIFY_EXIT (semfd != -1);
  89
  90   sem = xmmap (NULL, sizeof (sem_t), PROT_READ | PROT_WRITE, MAP_SHARED,
  91                semfd);
  92   TEST_VERIFY_EXIT (sem != SEM_FAILED);
  93   if (semfile == NULL)
  94     {
  95       xftruncate (semfd, sizeof (sem_t));
  96       TEST_VERIFY_EXIT (sem_init (sem, 1, 0) != -1);
  97     }
  98
  99   if (command == NULL)
 100     command = argv[0];
 101
 102   if (pidfile)
 103     sl ();
 104
 105   int fd = create_temp_file ("tst-cancel7-pid-", &pidfilename);
 106   if (fd == -1)
 107     FAIL_EXIT1 ("create_temp_file failed: %m");
 108
 109   xwrite (fd, " ", 1);
 110   xclose (fd);
 111 }
 112
 113
 114 static int
 115 do_test (void)
 116 {
 117   pthread_t th = xpthread_create (NULL, tf, NULL);
 118
 119   /* Wait to cancel until after the pid is written.  */
 120   if (sem_wait (sem) != 0)
 121     FAIL_EXIT1 ("sem_wait: %m");
 122
 123   xpthread_cancel (th);
 124   void *r = xpthread_join (th);
 125
 126   FILE *f = xfopen (pidfilename, "r+");
 127
 128   long long ll;
 129   if (fscanf (f, "%lld\n", &ll) != 1)
 130     FAIL_EXIT1 ("fscanf: %m");
 131
 132   struct flock fl =
 133     {
 134       .l_type = F_WRLCK,
 135       .l_start = 0,
 136       .l_whence = SEEK_SET,
 137       .l_len = 1
 138     };
 139   if (fcntl (fileno (f), F_GETLK, &fl) != 0)
 140     FAIL_EXIT1 ("fcntl: %m");
 141
 142   if (fl.l_type != F_UNLCK)
 143     {
 144       printf ("child %lld still running\n", (long long) fl.l_pid);
 145       if (fl.l_pid == ll)
 146         kill (fl.l_pid, SIGKILL);
 147
 148       return 1;
 149     }
 150
 151   xfclose (f);
 152
 153   return r != PTHREAD_CANCELED;
 154 }
 155
 156 static void
 157 do_cleanup (void)
 158 {
 159   FILE *f = fopen (pidfilename, "r+");
 160   long long ll;
 161
 162   if (f != NULL && fscanf (f, "%lld\n", &ll) == 1)
 163     {
 164       struct flock fl =
 165         {
 166           .l_type = F_WRLCK,
 167           .l_start = 0,
 168           .l_whence = SEEK_SET,
 169           .l_len = 1
 170         };
 171       if (fcntl (fileno (f), F_GETLK, &fl) == 0 && fl.l_type != F_UNLCK
 172           && fl.l_pid == ll)
 173         kill (fl.l_pid, SIGKILL);
 174
 175       fclose (f);
 176     }
 177 }
 178
 179 #define OPT_COMMAND     10000
 180 #define OPT_PIDFILE     10001
 181 #define OPT_SEMFILE     10002
 182 #define CMDLINE_OPTIONS \
 183   { "command", required_argument, NULL, OPT_COMMAND },  \
 184   { "pidfile", required_argument, NULL, OPT_PIDFILE },  \
 185   { "sem",     required_argument, NULL, OPT_SEMFILE },
 186 static void
 187 cmdline_process (int c)
 188 {
 189   switch (c)
 190     {
 191     case OPT_COMMAND:
 192       command = optarg;
 193       break;
 194     case OPT_PIDFILE:
 195       pidfile = optarg;
 196       break;
 197     case OPT_SEMFILE:
 198       semfile = optarg;
 199       break;
 200     }
 201 }
 202 #define CMDLINE_PROCESS cmdline_process
 203 #define CLEANUP_HANDLER do_cleanup
 204 #define PREPARE do_prepare
 205 #include <support/test-driver.c>