glibc/linuxthreads/tst-cancel.c

/* Tests for cancelation handling.  */

#include <pthread.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <sys/stat.h>

int fd;

pthread_barrier_t bar;


static void
cleanup (void *arg)
{
  int nr = (int) (long int) arg;
  char s[30];
  char *cp = stpcpy (s, "cleanup ");
  *cp++ = '0' + nr;
  *cp++ = '\n';
  __libc_write (fd, s, cp - s);
}


static void *
t1 (void *arg)
{
  pthread_cleanup_push (cleanup, (void *) (long int) 1);
  return NULL;
  pthread_cleanup_pop (0);
}


static void
inner (int a)
{
  pthread_cleanup_push (cleanup, (void *) (long int) a);
  if (a)
    return;
  pthread_cleanup_pop (0);
}


static void *
t2 (void *arg)
{
  pthread_cleanup_push (cleanup, (void *) (long int) 2);
  inner ((int) (long int) arg);
  return NULL;
  pthread_cleanup_pop (0);
}


static void *
t3 (void *arg)
{
  pthread_cleanup_push (cleanup, (void *) (long int) 4);
  inner ((int) (long int) arg);
  pthread_exit (NULL);
  pthread_cleanup_pop (0);
}


int
main (void)
{
  pthread_t td;
  int err;
  char tmp[] = "thtstXXXXXX";
  struct stat64 st;
  int result = 0;

  fd = mkstemp (tmp);
  if (fd == -1)
    {
      printf ("cannot create temporary file: %m");
      exit (1);
    }
  unlink (tmp);

  err = pthread_barrier_init (&bar, NULL, 2);
  if (err != 0 )
    {
      printf ("cannot create barrier: %s\n", strerror (err));
      exit (1);
    }

  err = pthread_create (&td, NULL, t1, NULL);
  if (err != 0)
    {
      printf ("cannot create thread t1: %s\n", strerror (err));
      exit (1);
    }

  err = pthread_join (td, NULL);
  if (err != 0)
    {
      printf ("cannot join thread: %s\n", strerror (err));
      exit (1);
    }

  err = pthread_create (&td, NULL, t2, (void *) 3);
  if (err != 0)
    {
      printf ("cannot create thread t2: %s\n", strerror (err));
      exit (1);
    }

  err = pthread_join (td, NULL);
  if (err != 0)
    {
      printf ("cannot join thread: %s\n", strerror (err));
      exit (1);
    }

  err = pthread_create (&td, NULL, t3, (void *) 5);
  if (err != 0)
    {
      printf ("cannot create thread t3: %s\n", strerror (err));
      exit (1);
    }

  err = pthread_join (td, NULL);
  if (err != 0)
    {
      printf ("cannot join thread: %s\n", strerror (err));
      exit (1);
    }

  if (fstat64 (fd, &st) < 0)
    {
      printf ("cannot stat temporary file: %m\n");
      result = 1;
    }
  else if (st.st_size != 0)
    {
      char buf[512];
      puts ("some cleanup handlers ran:");
      fflush (stdout);
      while (1)
	{
	  ssize_t n = read (fd, buf, sizeof buf);
	  if (n < 0)
	    break;
	  write (STDOUT_FILENO, buf, n);
	}
      result = 1;
    }

  return result;
}