82a877cffea804149e05a7fe92de563f194b6846
[kopensolaris-gnu/glibc.git] / linuxthreads / tst-context.c
1 #include <errno.h>
2 #include <error.h>
3 #include <pthread.h>
4 #include <stdio.h>
5 #include <stdlib.h>
6 #include <ucontext.h>
7
8 #include "pt-machine.h"
9
10
11 #define N       4
12
13 #ifdef FLOATING_STACKS
14 static char stacks[N][8192];
15 static ucontext_t ctx[N][2];
16 static volatile int failures;
17
18 static void
19 fct (long int n)
20 {
21   /* Just to use the thread local descriptor.  */
22   printf ("%ld: in %s now\n", n, __FUNCTION__);
23   errno = 0;
24 }
25
26 static void *
27 threadfct (void *arg)
28 {
29   int n = (int) (long int) arg;
30
31   printf ("%d: %s: before makecontext\n", n, __FUNCTION__);
32
33   ctx[n][1].uc_stack.ss_sp = stacks[n];
34   ctx[n][1].uc_stack.ss_size = 8192;
35   ctx[n][1].uc_link = &ctx[n][0];
36   makecontext (&ctx[n][1], (void (*) (void)) fct, 1, (long int) n);
37
38   printf ("%d: %s: before swapcontext\n", n, __FUNCTION__);
39
40   if (swapcontext (&ctx[n][0], &ctx[n][1]) != 0)
41     {
42       ++failures;
43       printf ("%d: %s: swapcontext failed\n", n, __FUNCTION__);
44     }
45   else
46     printf ("%d: back in %s\n", n, __FUNCTION__);
47
48   return NULL;
49 }
50 #endif
51
52
53 static volatile int global;
54
55 int
56 main (void)
57 {
58 #ifndef FLOATING_STACKS
59   puts ("not supported");
60   return 0;
61 #else
62   int n;
63   pthread_t th[N];
64   ucontext_t mctx;
65
66   puts ("making contexts");
67   if (getcontext (&mctx) != 0)
68     {
69       if (errno == ENOSYS)
70         {
71           puts ("context handling not supported");
72           exit (0);
73         }
74
75       printf ("%s: getcontext: %m\n", __FUNCTION__);
76       exit (1);
77     }
78
79   /* Play some tricks with this context.  */
80   if (++global == 1)
81     if (setcontext (&mctx) != 0)
82       {
83         printf ("%s: setcontext: %m\n", __FUNCTION__);
84         exit (1);
85       }
86   if (global != 2)
87     {
88       printf ("%s: 'global' not incremented twice\n", __FUNCTION__);
89       exit (1);
90     }
91
92   for (n = 0; n < N; ++n)
93     if (pthread_create (&th[n], NULL, threadfct, (void *) n) != 0)
94       error (EXIT_FAILURE, errno, "cannot create all threads");
95
96   for (n = 0; n < N; ++n)
97     pthread_join (th[n], NULL);
98
99   return failures;
100 #endif
101 }