tests/workq_sigprof.c

   1 #include <pthread.h>
   2 #include <stdbool.h>
   3 #include <signal.h>
   4 #include <stdio.h>
   5 #include <string.h>
   6 #include <unistd.h>
   7 #include <sys/time.h>
   8 #include <mach/mach_time.h>
   9 #include <dispatch/dispatch.h>
  10
  11 #include <darwintest.h>
  12
  13 #if !TARGET_OS_IPHONE
  14
  15 T_GLOBAL_META(T_META_RUN_CONCURRENTLY(true));
  16
  17 static pthread_t workq_thread;
  18 static bool signal_received;
  19
  20 static void
  21 signal_handler(int sig __unused, siginfo_t *b __unused, void* unused __unused)
  22 {
  23         if (pthread_self() == workq_thread) {
  24                 signal_received = true;
  25         }
  26 }
  27
  28 static void
  29 workq_block(void *unused __unused)
  30 {
  31         workq_thread = pthread_self();
  32
  33         /*
  34          *  sigset_t set;
  35          *  sigemptyset(&set);
  36          *  sigaddset(&set, SIGPROF);
  37          *  pthread_sigmask(SIG_UNBLOCK, &set, NULL);
  38          */
  39
  40         uint64_t spin_start = mach_absolute_time();
  41         while (mach_absolute_time() - spin_start < 30 * NSEC_PER_SEC) {
  42                 if (signal_received) {
  43                         T_PASS("Got SIGPROF!");
  44                         T_END;
  45                 }
  46         }
  47 }
  48
  49 T_DECL(workq_sigprof, "test that workqueue threads can receive sigprof")
  50 {
  51         struct sigaction sa = {
  52                 .sa_sigaction = signal_handler
  53         };
  54         sigfillset(&sa.sa_mask);
  55         T_ASSERT_POSIX_ZERO(sigaction(SIGPROF, &sa, NULL), NULL);
  56
  57         dispatch_queue_t q = dispatch_get_global_queue(0, 0);
  58         dispatch_async_f(q, NULL, workq_block);
  59
  60         struct itimerval timerval = {
  61                 .it_interval = {.tv_usec = 10000},
  62                 .it_value = {.tv_usec = 10000}
  63         };
  64         T_ASSERT_POSIX_ZERO(setitimer(ITIMER_PROF, &timerval, NULL), NULL);
  65
  66         dispatch_main();
  67 }
  68
  69 #else //!TARGET_OS_IPHONE
  70
  71 T_DECL(workq_sigprof, "test that workqueue threads can receive sigprof")
  72 {
  73         T_EXPECTFAIL;
  74         T_FAIL("<rdar://problem/25864196> setitimer/sigprof doesn't seem to be delivered on embeded platforms");
  75 }
  76
  77 #endif //!TARGET_OS_IPHONE