glibc/linuxthreads/tst-cancel.c

214 lines
3.8 KiB
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;
}