libgc/pthread_stop_world.c

   1 #include "private/pthread_support.h"
   2
   3 #if defined(GC_PTHREADS) && !defined(GC_SOLARIS_THREADS) \
   4      && !defined(GC_IRIX_THREADS) && !defined(GC_WIN32_THREADS) \
   5      && !defined(GC_DARWIN_THREADS) && !defined(GC_AIX_THREADS)
   6
   7 #include <signal.h>
   8 #include <semaphore.h>
   9 #include <errno.h>
  10 #include <unistd.h>
  11
  12 #if DEBUG_THREADS
  13
  14 #ifndef NSIG
  15 # if defined(MAXSIG)
  16 #  define NSIG (MAXSIG+1)
  17 # elif defined(_NSIG)
  18 #  define NSIG _NSIG
  19 # elif defined(__SIGRTMAX)
  20 #  define NSIG (__SIGRTMAX+1)
  21 # else
  22   --> please fix it
  23 # endif
  24 #endif
  25
  26 void GC_print_sig_mask()
  27 {
  28     sigset_t blocked;
  29     int i;
  30
  31     if (pthread_sigmask(SIG_BLOCK, NULL, &blocked) != 0)
  32         ABORT("pthread_sigmask");
  33     GC_printf0("Blocked: ");
  34     for (i = 1; i < NSIG; i++) {
  35         if (sigismember(&blocked, i)) { GC_printf1("%ld ",(long) i); }
  36     }
  37     GC_printf0("\n");
  38 }
  39
  40 #endif
  41
  42 /* Remove the signals that we want to allow in thread stopping  */
  43 /* handler from a set.                                          */
  44 void GC_remove_allowed_signals(sigset_t *set)
  45 {
  46 #   ifdef NO_SIGNALS
  47       if (sigdelset(set, SIGINT) != 0
  48           || sigdelset(set, SIGQUIT) != 0
  49           || sigdelset(set, SIGABRT) != 0
  50           || sigdelset(set, SIGTERM) != 0) {
  51         ABORT("sigdelset() failed");
  52       }
  53 #   endif
  54
  55 #   ifdef MPROTECT_VDB
  56       /* Handlers write to the thread structure, which is in the heap,  */
  57       /* and hence can trigger a protection fault.                      */
  58       if (sigdelset(set, SIGSEGV) != 0
  59 #         ifdef SIGBUS
  60             || sigdelset(set, SIGBUS) != 0
  61 #         endif
  62           ) {
  63         ABORT("sigdelset() failed");
  64       }
  65 #   endif
  66 }
  67
  68 static sigset_t suspend_handler_mask;
  69
  70 word GC_stop_count;     /* Incremented at the beginning of GC_stop_world. */
  71
  72 #ifdef GC_OSF1_THREADS
  73   GC_bool GC_retry_signals = TRUE;
  74 #else
  75   GC_bool GC_retry_signals = FALSE;
  76 #endif
  77
  78 /*
  79  * We use signals to stop threads during GC.
  80  *
  81  * Suspended threads wait in signal handler for SIG_THR_RESTART.
  82  * That's more portable than semaphores or condition variables.
  83  * (We do use sem_post from a signal handler, but that should be portable.)
  84  *
  85  * The thread suspension signal SIG_SUSPEND is now defined in gc_priv.h.
  86  * Note that we can't just stop a thread; we need it to save its stack
  87  * pointer(s) and acknowledge.
  88  */
  89
  90 #ifndef SIG_THR_RESTART
  91 #  if defined(GC_HPUX_THREADS) || defined(GC_OSF1_THREADS)
  92 #    ifdef _SIGRTMIN
  93 #      define SIG_THR_RESTART _SIGRTMIN + 5
  94 #    else
  95 #      define SIG_THR_RESTART SIGRTMIN + 5
  96 #    endif
  97 #  else
  98 #   define SIG_THR_RESTART SIGXCPU
  99 #  endif
 100 #endif
 101
 102 sem_t GC_suspend_ack_sem;
 103
 104 void GC_suspend_handler(int sig)
 105 {
 106     int dummy;
 107     pthread_t my_thread = pthread_self();
 108     GC_thread me;
 109 #   ifdef PARALLEL_MARK
 110         word my_mark_no = GC_mark_no;
 111         /* Marker can't proceed until we acknowledge.  Thus this is     */
 112         /* guaranteed to be the mark_no correspending to our            */
 113         /* suspension, i.e. the marker can't have incremented it yet.   */
 114 #   endif
 115     word my_stop_count = GC_stop_count;
 116
 117     if (sig != SIG_SUSPEND) ABORT("Bad signal in suspend_handler");
 118
 119 #if DEBUG_THREADS
 120     GC_printf1("Suspending 0x%lx\n", my_thread);
 121 #endif
 122
 123     me = GC_lookup_thread(my_thread);
 124     /* The lookup here is safe, since I'm doing this on behalf  */
 125     /* of a thread which holds the allocation lock in order     */
 126     /* to stop the world.  Thus concurrent modification of the  */
 127     /* data structure is impossible.                            */
 128     if (me -> stop_info.last_stop_count == my_stop_count) {
 129         /* Duplicate signal.  OK if we are retrying.    */
 130         if (!GC_retry_signals) {
 131             WARN("Duplicate suspend signal in thread %lx\n",
 132                  pthread_self());
 133         }
 134         return;
 135     }
 136 #   ifdef SPARC
 137         me -> stop_info.stack_ptr = (ptr_t)GC_save_regs_in_stack();
 138 #   else
 139         me -> stop_info.stack_ptr = (ptr_t)(&dummy);
 140 #   endif
 141 #   ifdef IA64
 142         me -> backing_store_ptr = (ptr_t)GC_save_regs_in_stack();
 143 #   endif
 144
 145     /* Tell the thread that wants to stop the world that this   */
 146     /* thread has been stopped.  Note that sem_post() is        */
 147     /* the only async-signal-safe primitive in LinuxThreads.    */
 148     sem_post(&GC_suspend_ack_sem);
 149     me -> stop_info.last_stop_count = my_stop_count;
 150
 151     /* Wait until that thread tells us to restart by sending    */
 152     /* this thread a SIG_THR_RESTART signal.                    */
 153     /* SIG_THR_RESTART should be masked at this point.  Thus there      */
 154     /* is no race.                                              */
 155     do {
 156             me->stop_info.signal = 0;
 157             sigsuspend(&suspend_handler_mask);        /* Wait for signal */
 158     } while (me->stop_info.signal != SIG_THR_RESTART);
 159     /* If the RESTART signal gets lost, we can still lose.  That should be  */
 160     /* less likely than losing the SUSPEND signal, since we don't do much   */
 161     /* between the sem_post and sigsuspend.                                 */
 162     /* We'd need more handshaking to work around that, since we don't want  */
 163     /* to accidentally leave a RESTART signal pending, thus causing us to   */
 164     /* continue prematurely in a future round.                              */
 165
 166 #if DEBUG_THREADS
 167     GC_printf1("Continuing 0x%lx\n", my_thread);
 168 #endif
 169 }
 170
 171 void GC_restart_handler(int sig)
 172 {
 173     pthread_t my_thread = pthread_self();
 174     GC_thread me;
 175
 176     if (sig != SIG_THR_RESTART) ABORT("Bad signal in suspend_handler");
 177
 178     /* Let the GC_suspend_handler() know that we got a SIG_THR_RESTART. */
 179     /* The lookup here is safe, since I'm doing this on behalf  */
 180     /* of a thread which holds the allocation lock in order     */
 181     /* to stop the world.  Thus concurrent modification of the  */
 182     /* data structure is impossible.                            */
 183     me = GC_lookup_thread(my_thread);
 184     me->stop_info.signal = SIG_THR_RESTART;
 185
 186     /*
 187     ** Note: even if we didn't do anything useful here,
 188     ** it would still be necessary to have a signal handler,
 189     ** rather than ignoring the signals, otherwise
 190     ** the signals will not be delivered at all, and
 191     ** will thus not interrupt the sigsuspend() above.
 192     */
 193
 194 #if DEBUG_THREADS
 195     GC_printf1("In GC_restart_handler for 0x%lx\n", pthread_self());
 196 #endif
 197 }
 198
 199 # ifdef IA64
 200 #   define IF_IA64(x) x
 201 # else
 202 #   define IF_IA64(x)
 203 # endif
 204 /* We hold allocation lock.  Should do exactly the right thing if the   */
 205 /* world is stopped.  Should not fail if it isn't.                      */
 206 static void pthread_push_all_stacks()
 207 {
 208     GC_bool found_me = FALSE;
 209     int i;
 210     GC_thread p;
 211     ptr_t lo, hi;
 212     /* On IA64, we also need to scan the register backing store. */
 213     IF_IA64(ptr_t bs_lo; ptr_t bs_hi;)
 214     pthread_t me = pthread_self();
 215
 216     if (!GC_thr_initialized) GC_thr_init();
 217     #if DEBUG_THREADS
 218         GC_printf1("Pushing stacks from thread 0x%lx\n", (unsigned long) me);
 219     #endif
 220     for (i = 0; i < THREAD_TABLE_SZ; i++) {
 221       for (p = GC_threads[i]; p != 0; p = p -> next) {
 222         if (p -> flags & FINISHED) continue;
 223         if (pthread_equal(p -> id, me)) {
 224 #           ifdef SPARC
 225                 lo = (ptr_t)GC_save_regs_in_stack();
 226 #           else
 227                 lo = GC_approx_sp();
 228 #           endif
 229             found_me = TRUE;
 230             IF_IA64(bs_hi = (ptr_t)GC_save_regs_in_stack();)
 231         } else {
 232             lo = p -> stop_info.stack_ptr;
 233             IF_IA64(bs_hi = p -> backing_store_ptr;)
 234         }
 235         if ((p -> flags & MAIN_THREAD) == 0) {
 236             hi = p -> stack_end;
 237             IF_IA64(bs_lo = p -> backing_store_end);
 238         } else {
 239             /* The original stack. */
 240             hi = GC_stackbottom;
 241             IF_IA64(bs_lo = BACKING_STORE_BASE;)
 242         }
 243         #if DEBUG_THREADS
 244             GC_printf3("Stack for thread 0x%lx = [%lx,%lx)\n",
 245                 (unsigned long) p -> id,
 246                 (unsigned long) lo, (unsigned long) hi);
 247         #endif
 248         if (0 == lo) ABORT("GC_push_all_stacks: sp not set!\n");
 249 #       ifdef STACK_GROWS_UP
 250           /* We got them backwards! */
 251           GC_push_all_stack(hi, lo);
 252 #       else
 253           GC_push_all_stack(lo, hi);
 254 #       endif
 255 #       ifdef IA64
 256 #         if DEBUG_THREADS
 257             GC_printf3("Reg stack for thread 0x%lx = [%lx,%lx)\n",
 258                 (unsigned long) p -> id,
 259                 (unsigned long) bs_lo, (unsigned long) bs_hi);
 260 #         endif
 261           if (pthread_equal(p -> id, me)) {
 262             GC_push_all_eager(bs_lo, bs_hi);
 263           } else {
 264             GC_push_all_stack(bs_lo, bs_hi);
 265           }
 266 #       endif
 267       }
 268     }
 269     if (!found_me && !GC_in_thread_creation)
 270       ABORT("Collecting from unknown thread.");
 271 }
 272
 273 /* We hold allocation lock.  Should do exactly the right thing if the   */
 274 /* world is stopped.  Should not fail if it isn't.                      */
 275 void GC_push_all_stacks()
 276 {
 277     gc_thread_vtable->push_all_stacks();
 278 }
 279
 280 /* There seems to be a very rare thread stopping problem.  To help us  */
 281 /* debug that, we save the ids of the stopping thread. */
 282 pthread_t GC_stopping_thread;
 283 int GC_stopping_pid;
 284
 285 /* We hold the allocation lock.  Suspend all threads that might */
 286 /* still be running.  Return the number of suspend signals that */
 287 /* were sent. */
 288 int GC_suspend_all()
 289 {
 290     int n_live_threads = 0;
 291     int i;
 292     GC_thread p;
 293     int result;
 294     pthread_t my_thread = pthread_self();
 295
 296     GC_stopping_thread = my_thread;    /* debugging only.      */
 297     GC_stopping_pid = getpid();                /* debugging only.      */
 298     for (i = 0; i < THREAD_TABLE_SZ; i++) {
 299       for (p = GC_threads[i]; p != 0; p = p -> next) {
 300         if (p -> id != my_thread) {
 301             if (p -> flags & FINISHED) continue;
 302             if (p -> stop_info.last_stop_count == GC_stop_count) continue;
 303             if (p -> thread_blocked) /* Will wait */ continue;
 304             n_live_threads++;
 305             #if DEBUG_THREADS
 306               GC_printf1("Sending suspend signal to 0x%lx\n", p -> id);
 307             #endif
 308
 309         result = pthread_kill(p -> id, SIG_SUSPEND);
 310             switch(result) {
 311                 case ESRCH:
 312                     /* Not really there anymore.  Possible? */
 313                     n_live_threads--;
 314                     break;
 315                 case 0:
 316                     break;
 317                 default:
 318                     ABORT("pthread_kill failed");
 319             }
 320         }
 321       }
 322     }
 323     return n_live_threads;
 324 }
 325
 326 /* Caller holds allocation lock.        */
 327 static void pthread_stop_world()
 328 {
 329     int i;
 330     int n_live_threads;
 331     int code;
 332
 333     #if DEBUG_THREADS
 334     GC_printf1("Stopping the world from 0x%lx\n", pthread_self());
 335     #endif
 336
 337     n_live_threads = GC_suspend_all();
 338
 339       if (GC_retry_signals) {
 340           unsigned long wait_usecs = 0;  /* Total wait since retry.     */
 341 #         define WAIT_UNIT 3000
 342 #         define RETRY_INTERVAL 100000
 343           for (;;) {
 344               int ack_count;
 345
 346               sem_getvalue(&GC_suspend_ack_sem, &ack_count);
 347               if (ack_count == n_live_threads) break;
 348               if (wait_usecs > RETRY_INTERVAL) {
 349                   int newly_sent = GC_suspend_all();
 350
 351 #                 ifdef CONDPRINT
 352                     if (GC_print_stats) {
 353                       GC_printf1("Resent %ld signals after timeout\n",
 354                                  newly_sent);
 355                     }
 356 #                 endif
 357                   sem_getvalue(&GC_suspend_ack_sem, &ack_count);
 358                   if (newly_sent < n_live_threads - ack_count) {
 359                       WARN("Lost some threads during GC_stop_world?!\n",0);
 360                       n_live_threads = ack_count + newly_sent;
 361                   }
 362                   wait_usecs = 0;
 363               }
 364               usleep(WAIT_UNIT);
 365               wait_usecs += WAIT_UNIT;
 366           }
 367       }
 368     for (i = 0; i < n_live_threads; i++) {
 369           if (0 != (code = sem_wait(&GC_suspend_ack_sem))) {
 370               GC_err_printf1("Sem_wait returned %ld\n", (unsigned long)code);
 371               ABORT("sem_wait for handler failed");
 372           }
 373     }
 374     #if DEBUG_THREADS
 375       GC_printf1("World stopped from 0x%lx\n", pthread_self());
 376     #endif
 377     GC_stopping_thread = 0;  /* debugging only */
 378 }
 379
 380 /* Caller holds allocation lock.        */
 381 void GC_stop_world()
 382 {
 383     /* Make sure all free list construction has stopped before we start. */
 384     /* No new construction can start, since free list construction is   */
 385     /* required to acquire and release the GC lock before it starts,    */
 386     /* and we have the lock.                                            */
 387 #   ifdef PARALLEL_MARK
 388       GC_acquire_mark_lock();
 389       GC_ASSERT(GC_fl_builder_count == 0);
 390       /* We should have previously waited for it to become zero. */
 391 #   endif /* PARALLEL_MARK */
 392     ++GC_stop_count;
 393     gc_thread_vtable->stop_world ();
 394 #   ifdef PARALLEL_MARK
 395       GC_release_mark_lock();
 396 #   endif
 397 }
 398
 399 /* Caller holds allocation lock, and has held it continuously since     */
 400 /* the world stopped.                                                   */
 401 static void pthread_start_world()
 402 {
 403     pthread_t my_thread = pthread_self();
 404     register int i;
 405     register GC_thread p;
 406     register int n_live_threads = 0;
 407     register int result;
 408
 409 #   if DEBUG_THREADS
 410       GC_printf0("World starting\n");
 411 #   endif
 412
 413     for (i = 0; i < THREAD_TABLE_SZ; i++) {
 414       for (p = GC_threads[i]; p != 0; p = p -> next) {
 415         if (p -> id != my_thread) {
 416             if (p -> flags & FINISHED) continue;
 417             if (p -> thread_blocked) continue;
 418             n_live_threads++;
 419             #if DEBUG_THREADS
 420               GC_printf1("Sending restart signal to 0x%lx\n", p -> id);
 421             #endif
 422
 423         result = pthread_kill(p -> id, SIG_THR_RESTART);
 424             switch(result) {
 425                 case ESRCH:
 426                     /* Not really there anymore.  Possible? */
 427                     n_live_threads--;
 428                     break;
 429                 case 0:
 430                     break;
 431                 default:
 432                     ABORT("pthread_kill failed");
 433             }
 434         }
 435       }
 436     }
 437     #if DEBUG_THREADS
 438       GC_printf0("World started\n");
 439     #endif
 440 }
 441
 442 void GC_start_world()
 443 {
 444     gc_thread_vtable->start_world();
 445 }
 446
 447 static void pthread_stop_init() {
 448     struct sigaction act;
 449
 450     if (sem_init(&GC_suspend_ack_sem, 0, 0) != 0)
 451         ABORT("sem_init failed");
 452
 453     act.sa_flags = SA_RESTART;
 454     if (sigfillset(&act.sa_mask) != 0) {
 455         ABORT("sigfillset() failed");
 456     }
 457     GC_remove_allowed_signals(&act.sa_mask);
 458     /* SIG_THR_RESTART is set in the resulting mask.            */
 459     /* It is unmasked by the handler when necessary.            */
 460     act.sa_handler = GC_suspend_handler;
 461     if (sigaction(SIG_SUSPEND, &act, NULL) != 0) {
 462         ABORT("Cannot set SIG_SUSPEND handler");
 463     }
 464
 465     act.sa_handler = GC_restart_handler;
 466     if (sigaction(SIG_THR_RESTART, &act, NULL) != 0) {
 467         ABORT("Cannot set SIG_THR_RESTART handler");
 468     }
 469
 470     /* Inititialize suspend_handler_mask. It excludes SIG_THR_RESTART. */
 471       if (sigfillset(&suspend_handler_mask) != 0) ABORT("sigfillset() failed");
 472       GC_remove_allowed_signals(&suspend_handler_mask);
 473       if (sigdelset(&suspend_handler_mask, SIG_THR_RESTART) != 0)
 474           ABORT("sigdelset() failed");
 475
 476     /* Check for GC_RETRY_SIGNALS.      */
 477       if (0 != GETENV("GC_RETRY_SIGNALS")) {
 478           GC_retry_signals = TRUE;
 479       }
 480       if (0 != GETENV("GC_NO_RETRY_SIGNALS")) {
 481           GC_retry_signals = FALSE;
 482       }
 483 #     ifdef CONDPRINT
 484           if (GC_print_stats && GC_retry_signals) {
 485               GC_printf0("Will retry suspend signal if necessary.\n");
 486           }
 487 #     endif
 488 }
 489
 490 /* We hold the allocation lock. */
 491 void GC_stop_init()
 492 {
 493     gc_thread_vtable->initialize ();
 494 }
 495
 496 /*
 497  * This is used by the Mono Debugger to stop/start the world.
 498  */
 499 GCThreadFunctions pthread_thread_vtable = {
 500     pthread_stop_init,
 501
 502     pthread_stop_world,
 503     pthread_push_all_stacks,
 504     pthread_start_world
 505 };
 506
 507 GCThreadFunctions *gc_thread_vtable = &pthread_thread_vtable;
 508
 509 #endif