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_lseek (fd, 0, SEEK_END);
  __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);
}


/* This does not work yet.  */
volatile int cleanupokcnt;

static void
cleanupok (void *arg)
{
  ++cleanupokcnt;
}


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


static void
innerok (int a)
{
  pthread_cleanup_push (cleanupok, (void *) (long int) a);
  pthread_exit (NULL);
  pthread_cleanup_pop (0);
}


static void *
t4 (void *arg)
{
  pthread_cleanup_push (cleanupok, (void *) (long int) 6);
  innerok ((int) (long int) arg);
  pthread_cleanup_pop (0);
  return NULL;
}


int
main (int argc, char *argv[])
{
  pthread_t td;
  int err;
  char *tmp;
  const char *prefix;
  const char template[] = "thtstXXXXXX";
  struct stat64 st;
  int result = 0;

  prefix = argc > 1 ? argv[1] : "";
  tmp = (char *) alloca (strlen (prefix) + sizeof template);
  strcpy (stpcpy (tmp, prefix), template);

  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);
    }

#ifdef NOT_YET
  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);
    }
#endif

  err = pthread_create (&td, NULL, t4, (void *) 7);
  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);
      __lseek (fd, 0, SEEK_SET);
      while (1)
	{
	  ssize_t n = read (fd, buf, sizeof buf);
	  if (n <= 0)
	    break;
	  write (STDOUT_FILENO, buf, n);
	}
      result = 1;
    }

  // if (cleanupokcnt != 3)  will be three once t3 runs
  if (cleanupokcnt != 2)
    {
      printf ("cleanupokcnt = %d\n", cleanupokcnt);
      result = 1;
    }

  return result;
}