sdcc/support/regression/tcc/114_bound_signal.c

   1 #include <stdio.h>
   2 #include <pthread.h>
   3 #include <semaphore.h>
   4 #include <signal.h>
   5 #include <string.h>
   6 #include <sys/types.h>
   7 #include <sys/wait.h>
   8 #include <unistd.h>
   9 #include <stdlib.h>
  10 #include <errno.h>
  11 #include <setjmp.h>
  12
  13 static volatile int run = 1;
  14 static sem_t sem;
  15 static sem_t sem_child;
  16
  17 static void
  18 add (int n)
  19 {
  20     int i;
  21     int arr[n];
  22
  23     for (i = 0; i < n; i++) {
  24         arr[i]++;
  25     }
  26     memset (&arr[0], 0, n * sizeof(int));
  27 }
  28
  29 static void *
  30 high_load (void *unused)
  31 {
  32     while (run) {
  33         add(10);
  34         add(20);
  35     }
  36     return NULL;
  37 }
  38
  39 static void *
  40 do_signal (void *unused)
  41 {
  42     while (run) {
  43         kill (getpid(), SIGUSR1);
  44         while (sem_wait(&sem) < 0 && errno == EINTR);
  45     }
  46     return NULL;
  47 }
  48
  49 static void *
  50 do_fork (void *unused)
  51 {
  52     pid_t pid;
  53
  54     while (run) {
  55         switch ((pid = fork())) {
  56         case 0:
  57             add(1000);
  58             add(2000);
  59             exit(0);
  60             break;
  61         case -1:
  62             return NULL;
  63         default:
  64             while (sem_wait(&sem_child) < 0 && errno == EINTR);
  65             wait(NULL);
  66             break;
  67         }
  68     }
  69     return NULL;
  70 }
  71
  72 static void signal_handler(int sig)
  73 {
  74     add(10);
  75     add(20);
  76     sem_post (&sem);
  77 }
  78
  79 static void child_handler(int sig)
  80 {
  81     add(10);
  82     add(20);
  83     sem_post (&sem_child);
  84 }
  85
  86 int
  87 main (void)
  88 {
  89     int i;
  90     pthread_t id1, id2, id3;
  91     struct sigaction act;
  92     sigjmp_buf sj;
  93     sigset_t m;
  94     time_t end;
  95
  96     memset (&act, 0, sizeof (act));
  97     act.sa_handler = signal_handler;
  98     act.sa_flags = 0;
  99     sigemptyset (&act.sa_mask);
 100     sigaction (SIGUSR1, &act, NULL);
 101     act.sa_handler = child_handler;
 102     sigaction (SIGCHLD, &act, NULL);
 103
 104     printf ("start\n"); fflush(stdout);
 105
 106     sem_init (&sem, 0, 0);
 107     sem_init (&sem_child, 0, 0);
 108     pthread_create(&id1, NULL, high_load, NULL);
 109     pthread_create(&id2, NULL, do_signal, NULL);
 110 #if !defined(__APPLE__)
 111     pthread_create(&id3, NULL, do_fork, NULL);
 112 #endif
 113
 114     /* sleep does not work !!! */
 115     end = time(NULL) + 2;
 116     while (time(NULL) < end) ;
 117     run = 0;
 118
 119     pthread_join(id1, NULL);
 120     pthread_join(id2, NULL);
 121 #if !defined(__APPLE__)
 122     pthread_join(id3, NULL);
 123 #endif
 124     sem_destroy (&sem);
 125     sem_destroy (&sem_child);
 126
 127     printf ("end\n"); fflush(stdout);
 128
 129     sigemptyset (&m);
 130     sigprocmask (SIG_SETMASK, &m, NULL);
 131     if (sigsetjmp (sj, 0) == 0)
 132     {
 133         sigaddset (&m, SIGUSR1);
 134         sigprocmask (SIG_SETMASK, &m, NULL);
 135         siglongjmp (sj, 1);
 136         printf ("failed");
 137         return 1;
 138     }
 139     sigprocmask (SIG_SETMASK, NULL, &m);
 140     if (!sigismember (&m, SIGUSR1))
 141         printf ("failed");
 142     return 0;
 143 }