nptl/tst-cancel10.c

   1 /* Copyright (C) 2003, 2004 Free Software Foundation, Inc.
   2    This file is part of the GNU C Library.
   3    Contributed by Ulrich Drepper <drepper@redhat.com>, 2003.
   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, write to the Free
  17    Software Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA
  18    02111-1307 USA.  */
  19
  20 #include <pthread.h>
  21 #include <stdio.h>
  22 #include <stdlib.h>
  23 #include <string.h>
  24
  25
  26 static void
  27 cleanup (void *arg)
  28 {
  29   /* Just for fun.  */
  30   if (pthread_cancel (pthread_self ()) != 0)
  31     {
  32       puts ("cleanup: cancel failed");
  33       exit (1);
  34     }
  35
  36   printf ("cleanup for %ld\n", (long int) arg);
  37 }
  38
  39
  40 static void *
  41 tf (void *arg)
  42 {
  43   long int n = (long int) arg;
  44
  45   pthread_cleanup_push (cleanup, arg);
  46
  47   if (pthread_setcanceltype ((n & 1) == 0
  48                              ? PTHREAD_CANCEL_DEFERRED
  49                              : PTHREAD_CANCEL_ASYNCHRONOUS, NULL) != 0)
  50     {
  51       puts ("setcanceltype failed");
  52       exit (1);
  53     }
  54
  55   if (pthread_cancel (pthread_self ()) != 0)
  56     {
  57       puts ("cancel failed");
  58       exit (1);
  59     }
  60
  61   pthread_testcancel ();
  62
  63   /* We should never come here.  */
  64
  65   pthread_cleanup_pop (0);
  66
  67   return NULL;
  68 }
  69
  70
  71 static int
  72 do_test (void)
  73 {
  74   pthread_attr_t at;
  75
  76   if (pthread_attr_init (&at) != 0)
  77     {
  78       puts ("attr_init failed");
  79       return 1;
  80     }
  81
  82   if (pthread_attr_setstacksize (&at, 1 * 1024 * 1024) != 0)
  83     {
  84       puts ("attr_setstacksize failed");
  85       return 1;
  86     }
  87
  88 #define N 20
  89   int i;
  90   pthread_t th[N];
  91
  92   for (i = 0; i < N; ++i)
  93     if (pthread_create (&th[i], &at, tf, (void *) (long int) i) != 0)
  94       {
  95         puts ("create failed");
  96         exit (1);
  97       }
  98
  99   if (pthread_attr_destroy (&at) != 0)
 100     {
 101       puts ("attr_destroy failed");
 102       return 1;
 103     }
 104
 105   for (i = 0; i < N; ++i)
 106     {
 107       void *r;
 108       if (pthread_join (th[i], &r) != 0)
 109         {
 110           puts ("join failed");
 111           exit (1);
 112         }
 113
 114       if (r != PTHREAD_CANCELED)
 115         {
 116           puts ("thread not canceled");
 117           exit (1);
 118         }
 119     }
 120
 121   return 0;
 122 }
 123
 124
 125 #define TEST_FUNCTION do_test ()
 126 #include "../test-skeleton.c"