sysdeps/pthread/tst-cancel13.c

   1 /* Test sem_wait cancellation for contended case.
   2    Copyright (C) 2003-2021 Free Software Foundation, Inc.
   3    This file is part of the GNU C Library.
   4    Contributed by Ulrich Drepper <drepper@redhat.com>, 2003.
   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    <https://www.gnu.org/licenses/>.  */
  19
  20 #include <errno.h>
  21 #include <pthread.h>
  22 #include <semaphore.h>
  23 #include <stdio.h>
  24 #include <stdlib.h>
  25 #include <string.h>
  26 #include <unistd.h>
  27
  28
  29 static pthread_barrier_t bar;
  30 static sem_t sem;
  31
  32
  33 static void
  34 cleanup (void *arg)
  35 {
  36   static int ncall;
  37
  38   if (++ncall != 1)
  39     {
  40       puts ("second call to cleanup");
  41       exit (1);
  42     }
  43 }
  44
  45
  46 static void *
  47 tf (void *arg)
  48 {
  49   pthread_cleanup_push (cleanup, NULL);
  50
  51   int e = pthread_barrier_wait (&bar);
  52   if (e != 0 && e != PTHREAD_BARRIER_SERIAL_THREAD)
  53     {
  54       puts ("error: tf: 1st barrier_wait failed");
  55       exit (1);
  56     }
  57
  58   /* This call should block and be cancelable.  */
  59   sem_wait (&sem);
  60
  61   pthread_cleanup_pop (0);
  62
  63   return NULL;
  64 }
  65
  66
  67 static int
  68 do_test (void)
  69 {
  70   pthread_t th;
  71
  72   if (pthread_barrier_init (&bar, NULL, 2) != 0)
  73     {
  74       puts ("error: barrier_init failed");
  75       exit (1);
  76     }
  77
  78   /* A value equal to 0 will check for contended pthread cancellation,
  79      where the sem_wait operation will block.  */
  80   if (sem_init (&sem, 0, 0) != 0)
  81     {
  82       puts ("error: sem_init failed");
  83       exit (1);
  84     }
  85
  86   if (pthread_create (&th, NULL, tf, NULL) != 0)
  87     {
  88       puts ("error: create failed");
  89       exit (1);
  90     }
  91
  92   int e = pthread_barrier_wait (&bar);
  93   if (e != 0 && e != PTHREAD_BARRIER_SERIAL_THREAD)
  94     {
  95       puts ("error: 1st barrier_wait failed");
  96       exit (1);
  97     }
  98
  99   /* Give the child a chance to go to sleep in sem_wait.  */
 100   sleep (1);
 101
 102   /* Check whether cancellation is honored when waiting in sem_wait.  */
 103   if (pthread_cancel (th) != 0)
 104     {
 105       puts ("error: 1st cancel failed");
 106       exit (1);
 107     }
 108
 109   void *r;
 110   if (pthread_join (th, &r) != 0)
 111     {
 112       puts ("error: join failed");
 113       exit (1);
 114     }
 115
 116   if (r != PTHREAD_CANCELED)
 117     {
 118       puts ("thread not canceled");
 119       exit (1);
 120     }
 121
 122   return 0;
 123 }
 124
 125
 126 #define TEST_FUNCTION do_test ()
 127 #include "../test-skeleton.c"