X-Git-Url: http://git.megacz.com/?a=blobdiff_plain;f=ghc%2Frts%2FGC.c;h=a57fa2cac047991d7cdff9decc7d36b1c03361e4;hb=e7c3f957fd36fd9f6369183b7a31e2a4a4c21b43;hp=448b43bd280514c1ce0e5013a929d8d4650fc837;hpb=ad61a4f5e334408ada25c71b043ac95f9c6b301f;p=ghc-hetmet.git diff --git a/ghc/rts/GC.c b/ghc/rts/GC.c index 448b43b..a57fa2c 100644 --- a/ghc/rts/GC.c +++ b/ghc/rts/GC.c @@ -1,48 +1,32 @@ /* ----------------------------------------------------------------------------- - * $Id: GC.c,v 1.81 2000/04/27 16:31:46 sewardj Exp $ * - * (c) The GHC Team 1998-1999 + * (c) The GHC Team 1998-2003 * * Generational garbage collector * * ---------------------------------------------------------------------------*/ -//@menu -//* Includes:: -//* STATIC OBJECT LIST:: -//* Static function declarations:: -//* Garbage Collect:: -//* Weak Pointers:: -//* Evacuation:: -//* Scavenging:: -//* Reverting CAFs:: -//* Sanity code for CAF garbage collection:: -//* Lazy black holing:: -//* Stack squeezing:: -//* Pausing a thread:: -//* Index:: -//@end menu - -//@node Includes, STATIC OBJECT LIST -//@subsection Includes - +#include "PosixSource.h" #include "Rts.h" #include "RtsFlags.h" #include "RtsUtils.h" +#include "Apply.h" #include "Storage.h" -#include "StoragePriv.h" +#include "LdvProfile.h" +#include "Updates.h" #include "Stats.h" #include "Schedule.h" -#include "SchedAPI.h" /* for ReverCAFs prototype */ #include "Sanity.h" -#include "GC.h" #include "BlockAlloc.h" -#include "Main.h" +#include "MBlock.h" #include "ProfHeap.h" #include "SchedAPI.h" #include "Weak.h" -#include "StablePriv.h" #include "Prelude.h" +#include "ParTicky.h" // ToDo: move into Rts.h +#include "GCCompact.h" +#include "Signals.h" +#include "STM.h" #if defined(GRAN) || defined(PAR) # include "GranSimRts.h" # include "ParallelRts.h" @@ -52,9 +36,15 @@ # include "ParallelDebug.h" # endif #endif +#include "HsFFI.h" +#include "Linker.h" +#if defined(RTS_GTK_FRONTPANEL) +#include "FrontPanel.h" +#endif -//@node STATIC OBJECT LIST, Static function declarations, Includes -//@subsection STATIC OBJECT LIST +#include "RetainerProfile.h" + +#include /* STATIC OBJECT LIST. * @@ -90,8 +80,8 @@ * We build up a static object list while collecting generations 0..N, * which is then appended to the static object list of generation N+1. */ -StgClosure* static_objects; /* live static objects */ -StgClosure* scavenged_static_objects; /* static objects scavenged so far */ +static StgClosure* static_objects; // live static objects +StgClosure* scavenged_static_objects; // static objects scavenged so far /* N is the oldest generation being collected, where the generations * are numbered starting at 0. A major GC (indicated by the major_gc @@ -109,13 +99,18 @@ static nat evac_gen; /* Weak pointers */ -static StgWeak *old_weak_ptr_list; /* also pending finaliser list */ -static rtsBool weak_done; /* all done for this pass */ +StgWeak *old_weak_ptr_list; // also pending finaliser list + +/* Which stage of processing various kinds of weak pointer are we at? + * (see traverse_weak_ptr_list() below for discussion). + */ +typedef enum { WeakPtrs, WeakThreads, WeakDone } WeakStage; +static WeakStage weak_stage; /* List of all threads during GC */ static StgTSO *old_all_threads; -static StgTSO *resurrected_threads; +StgTSO *resurrected_threads; /* Flag indicating failure to evacuate an object to the desired * generation. @@ -124,71 +119,184 @@ static rtsBool failed_to_evac; /* Old to-space (used for two-space collector only) */ -bdescr *old_to_space; - +static bdescr *old_to_blocks; /* Data used for allocation area sizing. */ -lnat new_blocks; /* blocks allocated during this GC */ -lnat g0s0_pcnt_kept = 30; /* percentage of g0s0 live at last minor GC */ +static lnat new_blocks; // blocks allocated during this GC +static lnat g0s0_pcnt_kept = 30; // percentage of g0s0 live at last minor GC -//@node Static function declarations, Garbage Collect, STATIC OBJECT LIST -//@subsection Static function declarations +/* Used to avoid long recursion due to selector thunks + */ +static lnat thunk_selector_depth = 0; +#define MAX_THUNK_SELECTOR_DEPTH 8 /* ----------------------------------------------------------------------------- Static function declarations -------------------------------------------------------------------------- */ -static StgClosure * evacuate ( StgClosure *q ); +static bdescr * gc_alloc_block ( step *stp ); +static void mark_root ( StgClosure **root ); + +// Use a register argument for evacuate, if available. +#if __GNUC__ >= 2 +#define REGPARM1 __attribute__((regparm(1))) +#else +#define REGPARM1 +#endif + +REGPARM1 static StgClosure * evacuate (StgClosure *q); + static void zero_static_object_list ( StgClosure* first_static ); -static void zero_mutable_list ( StgMutClosure *first ); static rtsBool traverse_weak_ptr_list ( void ); -static void cleanup_weak_ptr_list ( StgWeak **list ); +static void mark_weak_ptr_list ( StgWeak **list ); -static void scavenge_stack ( StgPtr p, StgPtr stack_end ); -static void scavenge_large ( step *step ); -static void scavenge ( step *step ); -static void scavenge_static ( void ); -static void scavenge_mutable_list ( generation *g ); -static void scavenge_mut_once_list ( generation *g ); +static StgClosure * eval_thunk_selector ( nat field, StgSelector * p ); -#ifdef DEBUG + +static void scavenge ( step * ); +static void scavenge_mark_stack ( void ); +static void scavenge_stack ( StgPtr p, StgPtr stack_end ); +static rtsBool scavenge_one ( StgPtr p ); +static void scavenge_large ( step * ); +static void scavenge_static ( void ); +static void scavenge_mutable_list ( generation *g ); + +static void scavenge_large_bitmap ( StgPtr p, + StgLargeBitmap *large_bitmap, + nat size ); + +#if 0 && defined(DEBUG) static void gcCAFs ( void ); #endif -//@node Garbage Collect, Weak Pointers, Static function declarations -//@subsection Garbage Collect +/* ----------------------------------------------------------------------------- + inline functions etc. for dealing with the mark bitmap & stack. + -------------------------------------------------------------------------- */ + +#define MARK_STACK_BLOCKS 4 + +static bdescr *mark_stack_bdescr; +static StgPtr *mark_stack; +static StgPtr *mark_sp; +static StgPtr *mark_splim; + +// Flag and pointers used for falling back to a linear scan when the +// mark stack overflows. +static rtsBool mark_stack_overflowed; +static bdescr *oldgen_scan_bd; +static StgPtr oldgen_scan; + +STATIC_INLINE rtsBool +mark_stack_empty(void) +{ + return mark_sp == mark_stack; +} + +STATIC_INLINE rtsBool +mark_stack_full(void) +{ + return mark_sp >= mark_splim; +} + +STATIC_INLINE void +reset_mark_stack(void) +{ + mark_sp = mark_stack; +} + +STATIC_INLINE void +push_mark_stack(StgPtr p) +{ + *mark_sp++ = p; +} + +STATIC_INLINE StgPtr +pop_mark_stack(void) +{ + return *--mark_sp; +} + +/* ----------------------------------------------------------------------------- + Allocate a new to-space block in the given step. + -------------------------------------------------------------------------- */ + +static bdescr * +gc_alloc_block(step *stp) +{ + bdescr *bd = allocBlock(); + bd->gen_no = stp->gen_no; + bd->step = stp; + bd->link = NULL; + + // blocks in to-space in generations up to and including N + // get the BF_EVACUATED flag. + if (stp->gen_no <= N) { + bd->flags = BF_EVACUATED; + } else { + bd->flags = 0; + } + + // Start a new to-space block, chain it on after the previous one. + if (stp->hp_bd == NULL) { + stp->hp_bd = bd; + } else { + stp->hp_bd->free = stp->hp; + stp->hp_bd->link = bd; + stp->hp_bd = bd; + } + + stp->hp = bd->start; + stp->hpLim = stp->hp + BLOCK_SIZE_W; + + stp->n_to_blocks++; + new_blocks++; + + return bd; +} /* ----------------------------------------------------------------------------- GarbageCollect - For garbage collecting generation N (and all younger generations): + Rough outline of the algorithm: for garbage collecting generation N + (and all younger generations): - follow all pointers in the root set. the root set includes all - mutable objects in all steps in all generations. + mutable objects in all generations (mutable_list). - for each pointer, evacuate the object it points to into either - + to-space in the next higher step in that generation, if one exists, - + if the object's generation == N, then evacuate it to the next - generation if one exists, or else to-space in the current - generation. - + if the object's generation < N, then evacuate it to to-space - in the next generation. + + + to-space of the step given by step->to, which is the next + highest step in this generation or the first step in the next + generation if this is the last step. + + + to-space of generations[evac_gen]->steps[0], if evac_gen != 0. + When we evacuate an object we attempt to evacuate + everything it points to into the same generation - this is + achieved by setting evac_gen to the desired generation. If + we can't do this, then an entry in the mut list has to + be made for the cross-generation pointer. + + + if the object is already in a generation > N, then leave + it alone. - repeatedly scavenge to-space from each step in each generation being collected until no more objects can be evacuated. - free from-space in each step, and set from-space = to-space. + Locks held: sched_mutex + -------------------------------------------------------------------------- */ -//@cindex GarbageCollect -void GarbageCollect ( void (*get_roots)(void), rtsBool force_major_gc ) +void +GarbageCollect ( void (*get_roots)(evac_fn), rtsBool force_major_gc ) { bdescr *bd; - step *step; + step *stp; lnat live, allocated, collected = 0, copied = 0; + lnat oldgen_saved_blocks = 0; nat g, s; #ifdef PROFILING @@ -196,20 +304,33 @@ void GarbageCollect ( void (*get_roots)(void), rtsBool force_major_gc ) #endif #if defined(DEBUG) && defined(GRAN) - IF_DEBUG(gc, belch("@@ Starting garbage collection at %ld (%lx)\n", + IF_DEBUG(gc, debugBelch("@@ Starting garbage collection at %ld (%lx)\n", Now, Now)); #endif - /* tell the stats department that we've started a GC */ +#if defined(RTS_USER_SIGNALS) + // block signals + blockUserSignals(); +#endif + + // tell the STM to discard any cached closures its hoping to re-use + stmPreGCHook(); + + // tell the stats department that we've started a GC stat_startGC(); - /* attribute any costs to CCS_GC */ + // Init stats and print par specific (timing) info + PAR_TICKY_PAR_START(); + + // attribute any costs to CCS_GC #ifdef PROFILING prev_CCS = CCCS; CCCS = CCS_GC; #endif - /* Approximate how much we allocated */ + /* Approximate how much we allocated. + * Todo: only when generating stats? + */ allocated = calcAllocated(); /* Figure out which generation to collect @@ -220,14 +341,22 @@ void GarbageCollect ( void (*get_roots)(void), rtsBool force_major_gc ) } else { N = 0; for (g = 0; g < RtsFlags.GcFlags.generations; g++) { - if (generations[g].steps[0].n_blocks >= generations[g].max_blocks) { + if (generations[g].steps[0].n_blocks + + generations[g].steps[0].n_large_blocks + >= generations[g].max_blocks) { N = g; } } major_gc = (N == RtsFlags.GcFlags.generations-1); } - /* check stack sanity *before* GC (ToDo: check all threads) */ +#ifdef RTS_GTK_FRONTPANEL + if (RtsFlags.GcFlags.frontpanel) { + updateFrontPanelBeforeGC(N); + } +#endif + + // check stack sanity *before* GC (ToDo: check all threads) #if defined(GRAN) // ToDo!: check sanity IF_DEBUG(sanity, checkTSOsSanity()); #endif @@ -238,18 +367,12 @@ void GarbageCollect ( void (*get_roots)(void), rtsBool force_major_gc ) static_objects = END_OF_STATIC_LIST; scavenged_static_objects = END_OF_STATIC_LIST; - /* zero the mutable list for the oldest generation (see comment by - * zero_mutable_list below). - */ - if (major_gc) { - zero_mutable_list(generations[RtsFlags.GcFlags.generations-1].mut_once_list); - } - /* Save the old to-space if we're doing a two-space collection */ if (RtsFlags.GcFlags.generations == 1) { - old_to_space = g0s0->to_space; - g0s0->to_space = NULL; + old_to_blocks = g0s0->to_blocks; + g0s0->to_blocks = NULL; + g0s0->n_to_blocks = 0; } /* Keep a count of how many new blocks we allocated during this GC @@ -257,78 +380,124 @@ void GarbageCollect ( void (*get_roots)(void), rtsBool force_major_gc ) */ new_blocks = 0; - /* Initialise to-space in all the generations/steps that we're - * collecting. - */ + // Initialise to-space in all the generations/steps that we're + // collecting. + // for (g = 0; g <= N; g++) { - generations[g].mut_once_list = END_MUT_LIST; - generations[g].mut_list = END_MUT_LIST; + + // throw away the mutable list. Invariant: the mutable list + // always has at least one block; this means we can avoid a check for + // NULL in recordMutable(). + if (g != 0) { + freeChain(generations[g].mut_list); + generations[g].mut_list = allocBlock(); + } for (s = 0; s < generations[g].n_steps; s++) { - /* generation 0, step 0 doesn't need to-space */ + // generation 0, step 0 doesn't need to-space if (g == 0 && s == 0 && RtsFlags.GcFlags.generations > 1) { continue; } - /* Get a free block for to-space. Extra blocks will be chained on - * as necessary. - */ - bd = allocBlock(); - step = &generations[g].steps[s]; - ASSERT(step->gen->no == g); - ASSERT(step->hp ? Bdescr(step->hp)->step == step : rtsTrue); - bd->gen = &generations[g]; - bd->step = step; - bd->link = NULL; - bd->evacuated = 1; /* it's a to-space block */ - step->hp = bd->start; - step->hpLim = step->hp + BLOCK_SIZE_W; - step->hp_bd = bd; - step->to_space = bd; - step->to_blocks = 1; - step->scan = bd->start; - step->scan_bd = bd; - step->new_large_objects = NULL; - step->scavenged_large_objects = NULL; - new_blocks++; - /* mark the large objects as not evacuated yet */ - for (bd = step->large_objects; bd; bd = bd->link) { - bd->evacuated = 0; + stp = &generations[g].steps[s]; + ASSERT(stp->gen_no == g); + + // start a new to-space for this step. + stp->hp = NULL; + stp->hp_bd = NULL; + stp->to_blocks = NULL; + + // allocate the first to-space block; extra blocks will be + // chained on as necessary. + bd = gc_alloc_block(stp); + stp->to_blocks = bd; + stp->scan = bd->start; + stp->scan_bd = bd; + + // initialise the large object queues. + stp->new_large_objects = NULL; + stp->scavenged_large_objects = NULL; + stp->n_scavenged_large_blocks = 0; + + // mark the large objects as not evacuated yet + for (bd = stp->large_objects; bd; bd = bd->link) { + bd->flags &= ~BF_EVACUATED; + } + + // for a compacted step, we need to allocate the bitmap + if (stp->is_compacted) { + nat bitmap_size; // in bytes + bdescr *bitmap_bdescr; + StgWord *bitmap; + + bitmap_size = stp->n_blocks * BLOCK_SIZE / (sizeof(W_)*BITS_PER_BYTE); + + if (bitmap_size > 0) { + bitmap_bdescr = allocGroup((nat)BLOCK_ROUND_UP(bitmap_size) + / BLOCK_SIZE); + stp->bitmap = bitmap_bdescr; + bitmap = bitmap_bdescr->start; + + IF_DEBUG(gc, debugBelch("bitmap_size: %d, bitmap: %p", + bitmap_size, bitmap);); + + // don't forget to fill it with zeros! + memset(bitmap, 0, bitmap_size); + + // For each block in this step, point to its bitmap from the + // block descriptor. + for (bd=stp->blocks; bd != NULL; bd = bd->link) { + bd->u.bitmap = bitmap; + bitmap += BLOCK_SIZE_W / (sizeof(W_)*BITS_PER_BYTE); + + // Also at this point we set the BF_COMPACTED flag + // for this block. The invariant is that + // BF_COMPACTED is always unset, except during GC + // when it is set on those blocks which will be + // compacted. + bd->flags |= BF_COMPACTED; + } + } } } } /* make sure the older generations have at least one block to - * allocate into (this makes things easier for copy(), see below. + * allocate into (this makes things easier for copy(), see below). */ for (g = N+1; g < RtsFlags.GcFlags.generations; g++) { for (s = 0; s < generations[g].n_steps; s++) { - step = &generations[g].steps[s]; - if (step->hp_bd == NULL) { - bd = allocBlock(); - bd->gen = &generations[g]; - bd->step = step; - bd->link = NULL; - bd->evacuated = 0; /* *not* a to-space block */ - step->hp = bd->start; - step->hpLim = step->hp + BLOCK_SIZE_W; - step->hp_bd = bd; - step->blocks = bd; - step->n_blocks = 1; - new_blocks++; + stp = &generations[g].steps[s]; + if (stp->hp_bd == NULL) { + ASSERT(stp->blocks == NULL); + bd = gc_alloc_block(stp); + stp->blocks = bd; + stp->n_blocks = 1; } /* Set the scan pointer for older generations: remember we * still have to scavenge objects that have been promoted. */ - step->scan = step->hp; - step->scan_bd = step->hp_bd; - step->to_space = NULL; - step->to_blocks = 0; - step->new_large_objects = NULL; - step->scavenged_large_objects = NULL; + stp->scan = stp->hp; + stp->scan_bd = stp->hp_bd; + stp->to_blocks = NULL; + stp->n_to_blocks = 0; + stp->new_large_objects = NULL; + stp->scavenged_large_objects = NULL; + stp->n_scavenged_large_blocks = 0; } } + /* Allocate a mark stack if we're doing a major collection. + */ + if (major_gc) { + mark_stack_bdescr = allocGroup(MARK_STACK_BLOCKS); + mark_stack = (StgPtr *)mark_stack_bdescr->start; + mark_sp = mark_stack; + mark_splim = mark_stack + (MARK_STACK_BLOCKS * BLOCK_SIZE_W); + } else { + mark_stack_bdescr = NULL; + } + /* ----------------------------------------------------------------------- * follow all the roots that we know about: * - mutable lists from each generation > N @@ -345,23 +514,12 @@ void GarbageCollect ( void (*get_roots)(void), rtsBool force_major_gc ) int st; for (g = RtsFlags.GcFlags.generations-1; g > N; g--) { generations[g].saved_mut_list = generations[g].mut_list; - generations[g].mut_list = END_MUT_LIST; - } - - /* Do the mut-once lists first */ - for (g = RtsFlags.GcFlags.generations-1; g > N; g--) { - IF_PAR_DEBUG(verbose, - printMutOnceList(&generations[g])); - scavenge_mut_once_list(&generations[g]); - evac_gen = g; - for (st = generations[g].n_steps-1; st >= 0; st--) { - scavenge(&generations[g].steps[st]); - } + generations[g].mut_list = allocBlock(); + // mut_list always has at least one block. } for (g = RtsFlags.GcFlags.generations-1; g > N; g--) { - IF_PAR_DEBUG(verbose, - printMutableList(&generations[g])); + IF_PAR_DEBUG(verbose, printMutableList(&generations[g])); scavenge_mutable_list(&generations[g]); evac_gen = g; for (st = generations[g].n_steps-1; st >= 0; st--) { @@ -370,10 +528,15 @@ void GarbageCollect ( void (*get_roots)(void), rtsBool force_major_gc ) } } + /* follow roots from the CAF list (used by GHCi) + */ + evac_gen = 0; + markCAFs(mark_root); + /* follow all the roots that the application knows about. */ evac_gen = 0; - get_roots(); + get_roots(mark_root); #if defined(PAR) /* And don't forget to mark the TSO if we got here direct from @@ -384,16 +547,19 @@ void GarbageCollect ( void (*get_roots)(void), rtsBool force_major_gc ) } */ - /* Mark the entries in the GALA table of the parallel system */ + // Mark the entries in the GALA table of the parallel system markLocalGAs(major_gc); + // Mark all entries on the list of pending fetches + markPendingFetches(major_gc); #endif /* Mark the weak pointer list, and prepare to detect dead weak * pointers. */ + mark_weak_ptr_list(&weak_ptr_list); old_weak_ptr_list = weak_ptr_list; weak_ptr_list = NULL; - weak_done = rtsFalse; + weak_stage = WeakPtrs; /* The all_threads list is like the weak_ptr_list. * See traverse_weak_ptr_list() for the details. @@ -404,18 +570,7 @@ void GarbageCollect ( void (*get_roots)(void), rtsBool force_major_gc ) /* Mark the stable pointer table. */ - markStablePtrTable(major_gc); - -#ifdef INTERPRETER - { - /* ToDo: To fix the caf leak, we need to make the commented out - * parts of this code do something sensible - as described in - * the CAF document. - */ - extern void markHugsObjects(void); - markHugsObjects(); - } -#endif + markStablePtrTable(mark_root); /* ------------------------------------------------------------------------- * Repeatedly scavenge all the areas we know about until there's no @@ -426,11 +581,10 @@ void GarbageCollect ( void (*get_roots)(void), rtsBool force_major_gc ) loop: flag = rtsFalse; - /* scavenge static objects */ + // scavenge static objects if (major_gc && static_objects != END_OF_STATIC_LIST) { - IF_DEBUG(sanity, - checkStaticObjects()); - scavenge_static(); + IF_DEBUG(sanity, checkStaticObjects(static_objects)); + scavenge_static(); } /* When scavenging the older generations: Objects may have been @@ -442,121 +596,184 @@ void GarbageCollect ( void (*get_roots)(void), rtsBool force_major_gc ) * generation. */ - /* scavenge each step in generations 0..maxgen */ + // scavenge each step in generations 0..maxgen { - int gen, st; + long gen; + int st; + loop2: - for (gen = RtsFlags.GcFlags.generations-1; gen >= 0; gen--) { - for (st = generations[gen].n_steps-1; st >= 0 ; st--) { + // scavenge objects in compacted generation + if (mark_stack_overflowed || oldgen_scan_bd != NULL || + (mark_stack_bdescr != NULL && !mark_stack_empty())) { + scavenge_mark_stack(); + flag = rtsTrue; + } + + for (gen = RtsFlags.GcFlags.generations; --gen >= 0; ) { + for (st = generations[gen].n_steps; --st >= 0; ) { if (gen == 0 && st == 0 && RtsFlags.GcFlags.generations > 1) { continue; } - step = &generations[gen].steps[st]; + stp = &generations[gen].steps[st]; evac_gen = gen; - if (step->hp_bd != step->scan_bd || step->scan < step->hp) { - scavenge(step); + if (stp->hp_bd != stp->scan_bd || stp->scan < stp->hp) { + scavenge(stp); flag = rtsTrue; goto loop2; } - if (step->new_large_objects != NULL) { - scavenge_large(step); + if (stp->new_large_objects != NULL) { + scavenge_large(stp); flag = rtsTrue; goto loop2; } } } } + if (flag) { goto loop; } - /* must be last... */ - if (traverse_weak_ptr_list()) { /* returns rtsTrue if evaced something */ + // must be last... invariant is that everything is fully + // scavenged at this point. + if (traverse_weak_ptr_list()) { // returns rtsTrue if evaced something goto loop; } } - /* Final traversal of the weak pointer list (see comment by - * cleanUpWeakPtrList below). - */ - cleanup_weak_ptr_list(&weak_ptr_list); - - /* Now see which stable names are still alive. + /* Update the pointers from the "main thread" list - these are + * treated as weak pointers because we want to allow a main thread + * to get a BlockedOnDeadMVar exception in the same way as any other + * thread. Note that the threads should all have been retained by + * GC by virtue of being on the all_threads list, we're just + * updating pointers here. */ - gcStablePtrTable(major_gc); + { + StgMainThread *m; + StgTSO *tso; + for (m = main_threads; m != NULL; m = m->link) { + tso = (StgTSO *) isAlive((StgClosure *)m->tso); + if (tso == NULL) { + barf("main thread has been GC'd"); + } + m->tso = tso; + } + } #if defined(PAR) - /* Reconstruct the Global Address tables used in GUM */ + // Reconstruct the Global Address tables used in GUM rebuildGAtables(major_gc); - IF_DEBUG(sanity, checkGlobalTSOList(rtsTrue/*check TSOs, too*/)); IF_DEBUG(sanity, checkLAGAtable(rtsTrue/*check closures, too*/)); #endif - /* Set the maximum blocks for the oldest generation, based on twice - * the amount of live data now, adjusted to fit the maximum heap - * size if necessary. - * - * This is an approximation, since in the worst case we'll need - * twice the amount of live data plus whatever space the other - * generations need. - */ - if (RtsFlags.GcFlags.generations > 1) { - if (major_gc) { - oldest_gen->max_blocks = - stg_max(oldest_gen->steps[0].to_blocks * RtsFlags.GcFlags.oldGenFactor, - RtsFlags.GcFlags.minOldGenSize); - if (oldest_gen->max_blocks > RtsFlags.GcFlags.maxHeapSize / 2) { - oldest_gen->max_blocks = RtsFlags.GcFlags.maxHeapSize / 2; - if (((int)oldest_gen->max_blocks - - (int)oldest_gen->steps[0].to_blocks) < - (RtsFlags.GcFlags.pcFreeHeap * - RtsFlags.GcFlags.maxHeapSize / 200)) { - heapOverflow(); - } + // Now see which stable names are still alive. + gcStablePtrTable(); + + // Tidy the end of the to-space chains + for (g = 0; g < RtsFlags.GcFlags.generations; g++) { + for (s = 0; s < generations[g].n_steps; s++) { + stp = &generations[g].steps[s]; + if (!(g == 0 && s == 0 && RtsFlags.GcFlags.generations > 1)) { + ASSERT(Bdescr(stp->hp) == stp->hp_bd); + stp->hp_bd->free = stp->hp; + } } - } } +#ifdef PROFILING + // We call processHeapClosureForDead() on every closure destroyed during + // the current garbage collection, so we invoke LdvCensusForDead(). + if (RtsFlags.ProfFlags.doHeapProfile == HEAP_BY_LDV + || RtsFlags.ProfFlags.bioSelector != NULL) + LdvCensusForDead(N); +#endif + + // NO MORE EVACUATION AFTER THIS POINT! + // Finally: compaction of the oldest generation. + if (major_gc && oldest_gen->steps[0].is_compacted) { + // save number of blocks for stats + oldgen_saved_blocks = oldest_gen->steps[0].n_blocks; + compact(get_roots); + } + + IF_DEBUG(sanity, checkGlobalTSOList(rtsFalse)); + /* run through all the generations/steps and tidy up */ copied = new_blocks * BLOCK_SIZE_W; for (g = 0; g < RtsFlags.GcFlags.generations; g++) { if (g <= N) { - generations[g].collections++; /* for stats */ + generations[g].collections++; // for stats + } + + // Count the mutable list as bytes "copied" for the purposes of + // stats. Every mutable list is copied during every GC. + if (g > 0) { + for (bd = generations[g].mut_list; bd != NULL; bd = bd->link) { + copied += (bd->free - bd->start) * sizeof(StgWord); + } } for (s = 0; s < generations[g].n_steps; s++) { bdescr *next; - step = &generations[g].steps[s]; + stp = &generations[g].steps[s]; if (!(g == 0 && s == 0 && RtsFlags.GcFlags.generations > 1)) { - /* Tidy the end of the to-space chains */ - step->hp_bd->free = step->hp; - step->hp_bd->link = NULL; - /* stats information: how much we copied */ + // stats information: how much we copied if (g <= N) { - copied -= step->hp_bd->start + BLOCK_SIZE_W - - step->hp_bd->free; + copied -= stp->hp_bd->start + BLOCK_SIZE_W - + stp->hp_bd->free; } } - /* for generations we collected... */ + // for generations we collected... if (g <= N) { - collected += step->n_blocks * BLOCK_SIZE_W; /* for stats */ + // rough calculation of garbage collected, for stats output + if (stp->is_compacted) { + collected += (oldgen_saved_blocks - stp->n_blocks) * BLOCK_SIZE_W; + } else { + collected += stp->n_blocks * BLOCK_SIZE_W; + } /* free old memory and shift to-space into from-space for all * the collected steps (except the allocation area). These * freed blocks will probaby be quickly recycled. */ if (!(g == 0 && s == 0)) { - freeChain(step->blocks); - step->blocks = step->to_space; - step->n_blocks = step->to_blocks; - step->to_space = NULL; - step->to_blocks = 0; - for (bd = step->blocks; bd != NULL; bd = bd->link) { - bd->evacuated = 0; /* now from-space */ - } + if (stp->is_compacted) { + // for a compacted step, just shift the new to-space + // onto the front of the now-compacted existing blocks. + for (bd = stp->to_blocks; bd != NULL; bd = bd->link) { + bd->flags &= ~BF_EVACUATED; // now from-space + } + // tack the new blocks on the end of the existing blocks + if (stp->blocks == NULL) { + stp->blocks = stp->to_blocks; + } else { + for (bd = stp->blocks; bd != NULL; bd = next) { + next = bd->link; + if (next == NULL) { + bd->link = stp->to_blocks; + } + // NB. this step might not be compacted next + // time, so reset the BF_COMPACTED flags. + // They are set before GC if we're going to + // compact. (search for BF_COMPACTED above). + bd->flags &= ~BF_COMPACTED; + } + } + // add the new blocks to the block tally + stp->n_blocks += stp->n_to_blocks; + } else { + freeChain(stp->blocks); + stp->blocks = stp->to_blocks; + stp->n_blocks = stp->n_to_blocks; + for (bd = stp->blocks; bd != NULL; bd = bd->link) { + bd->flags &= ~BF_EVACUATED; // now from-space + } + } + stp->to_blocks = NULL; + stp->n_to_blocks = 0; } /* LARGE OBJECTS. The current live large objects are chained on @@ -564,52 +781,117 @@ void GarbageCollect ( void (*get_roots)(void), rtsBool force_major_gc ) * collection from large_objects. Any objects left on * large_objects list are therefore dead, so we free them here. */ - for (bd = step->large_objects; bd != NULL; bd = next) { + for (bd = stp->large_objects; bd != NULL; bd = next) { next = bd->link; freeGroup(bd); bd = next; } - for (bd = step->scavenged_large_objects; bd != NULL; bd = bd->link) { - bd->evacuated = 0; - } - step->large_objects = step->scavenged_large_objects; - - /* Set the maximum blocks for this generation, interpolating - * between the maximum size of the oldest and youngest - * generations. - * - * max_blocks = oldgen_max_blocks * G - * ---------------------- - * oldest_gen - */ - if (g != 0) { -#if 0 - generations[g].max_blocks = (oldest_gen->max_blocks * g) - / (RtsFlags.GcFlags.generations-1); -#endif - generations[g].max_blocks = oldest_gen->max_blocks; + + // update the count of blocks used by large objects + for (bd = stp->scavenged_large_objects; bd != NULL; bd = bd->link) { + bd->flags &= ~BF_EVACUATED; } + stp->large_objects = stp->scavenged_large_objects; + stp->n_large_blocks = stp->n_scavenged_large_blocks; - /* for older generations... */ } else { + // for older generations... /* For older generations, we need to append the * scavenged_large_object list (i.e. large objects that have been * promoted during this GC) to the large_object list for that step. */ - for (bd = step->scavenged_large_objects; bd; bd = next) { + for (bd = stp->scavenged_large_objects; bd; bd = next) { next = bd->link; - bd->evacuated = 0; - dbl_link_onto(bd, &step->large_objects); + bd->flags &= ~BF_EVACUATED; + dbl_link_onto(bd, &stp->large_objects); } - /* add the new blocks we promoted during this GC */ - step->n_blocks += step->to_blocks; + // add the new blocks we promoted during this GC + stp->n_blocks += stp->n_to_blocks; + stp->n_to_blocks = 0; + stp->n_large_blocks += stp->n_scavenged_large_blocks; } } } - - /* Guess the amount of live data for stats. */ + + /* Reset the sizes of the older generations when we do a major + * collection. + * + * CURRENT STRATEGY: make all generations except zero the same size. + * We have to stay within the maximum heap size, and leave a certain + * percentage of the maximum heap size available to allocate into. + */ + if (major_gc && RtsFlags.GcFlags.generations > 1) { + nat live, size, min_alloc; + nat max = RtsFlags.GcFlags.maxHeapSize; + nat gens = RtsFlags.GcFlags.generations; + + // live in the oldest generations + live = oldest_gen->steps[0].n_blocks + + oldest_gen->steps[0].n_large_blocks; + + // default max size for all generations except zero + size = stg_max(live * RtsFlags.GcFlags.oldGenFactor, + RtsFlags.GcFlags.minOldGenSize); + + // minimum size for generation zero + min_alloc = stg_max((RtsFlags.GcFlags.pcFreeHeap * max) / 200, + RtsFlags.GcFlags.minAllocAreaSize); + + // Auto-enable compaction when the residency reaches a + // certain percentage of the maximum heap size (default: 30%). + if (RtsFlags.GcFlags.generations > 1 && + (RtsFlags.GcFlags.compact || + (max > 0 && + oldest_gen->steps[0].n_blocks > + (RtsFlags.GcFlags.compactThreshold * max) / 100))) { + oldest_gen->steps[0].is_compacted = 1; +// debugBelch("compaction: on\n", live); + } else { + oldest_gen->steps[0].is_compacted = 0; +// debugBelch("compaction: off\n", live); + } + + // if we're going to go over the maximum heap size, reduce the + // size of the generations accordingly. The calculation is + // different if compaction is turned on, because we don't need + // to double the space required to collect the old generation. + if (max != 0) { + + // this test is necessary to ensure that the calculations + // below don't have any negative results - we're working + // with unsigned values here. + if (max < min_alloc) { + heapOverflow(); + } + + if (oldest_gen->steps[0].is_compacted) { + if ( (size + (size - 1) * (gens - 2) * 2) + min_alloc > max ) { + size = (max - min_alloc) / ((gens - 1) * 2 - 1); + } + } else { + if ( (size * (gens - 1) * 2) + min_alloc > max ) { + size = (max - min_alloc) / ((gens - 1) * 2); + } + } + + if (size < live) { + heapOverflow(); + } + } + +#if 0 + debugBelch("live: %d, min_alloc: %d, size : %d, max = %d\n", live, + min_alloc, size, max); +#endif + + for (g = 0; g < gens; g++) { + generations[g].max_blocks = size; + } + } + + // Guess the amount of live data for stats. live = calcLive(); /* Free the small objects allocated via allocate(), since this will @@ -624,25 +906,45 @@ void GarbageCollect ( void (*get_roots)(void), rtsBool force_major_gc ) alloc_HpLim = NULL; alloc_blocks_lim = RtsFlags.GcFlags.minAllocAreaSize; + // Start a new pinned_object_block + pinned_object_block = NULL; + + /* Free the mark stack. + */ + if (mark_stack_bdescr != NULL) { + freeGroup(mark_stack_bdescr); + } + + /* Free any bitmaps. + */ + for (g = 0; g <= N; g++) { + for (s = 0; s < generations[g].n_steps; s++) { + stp = &generations[g].steps[s]; + if (stp->is_compacted && stp->bitmap != NULL) { + freeGroup(stp->bitmap); + } + } + } + /* Two-space collector: * Free the old to-space, and estimate the amount of live data. */ if (RtsFlags.GcFlags.generations == 1) { nat blocks; - if (old_to_space != NULL) { - freeChain(old_to_space); + if (old_to_blocks != NULL) { + freeChain(old_to_blocks); } - for (bd = g0s0->to_space; bd != NULL; bd = bd->link) { - bd->evacuated = 0; /* now from-space */ + for (bd = g0s0->to_blocks; bd != NULL; bd = bd->link) { + bd->flags = 0; // now from-space } /* For a two-space collector, we need to resize the nursery. */ /* set up a new nursery. Allocate a nursery size based on a - * function of the amount of live data (currently a factor of 2, - * should be configurable (ToDo)). Use the blocks from the old - * nursery if possible, freeing up any left over blocks. + * function of the amount of live data (by default a factor of 2) + * Use the blocks from the old nursery if possible, freeing up any + * left over blocks. * * If we get near the maximum heap size, then adjust our nursery * size accordingly. If the nursery is the same size as the live @@ -651,17 +953,18 @@ void GarbageCollect ( void (*get_roots)(void), rtsBool force_major_gc ) * * A normal 2-space collector would need 4L bytes to give the same * performance we get from 3L bytes, reducing to the same - * performance at 2L bytes. + * performance at 2L bytes. */ - blocks = g0s0->to_blocks; + blocks = g0s0->n_to_blocks; - if ( blocks * RtsFlags.GcFlags.oldGenFactor * 2 > - RtsFlags.GcFlags.maxHeapSize ) { - int adjusted_blocks; /* signed on purpose */ + if ( RtsFlags.GcFlags.maxHeapSize != 0 && + blocks * RtsFlags.GcFlags.oldGenFactor * 2 > + RtsFlags.GcFlags.maxHeapSize ) { + long adjusted_blocks; // signed on purpose int pc_free; adjusted_blocks = (RtsFlags.GcFlags.maxHeapSize - 2 * blocks); - IF_DEBUG(gc, fprintf(stderr, "@@ Near maximum heap size of 0x%x blocks, blocks = %d, adjusted to %d\n", RtsFlags.GcFlags.maxHeapSize, blocks, adjusted_blocks)); + IF_DEBUG(gc, debugBelch("@@ Near maximum heap size of 0x%x blocks, blocks = %d, adjusted to %ld", RtsFlags.GcFlags.maxHeapSize, blocks, adjusted_blocks)); pc_free = adjusted_blocks * 100 / RtsFlags.GcFlags.maxHeapSize; if (pc_free < RtsFlags.GcFlags.pcFreeHeap) /* might even be < 0 */ { heapOverflow(); @@ -683,11 +986,11 @@ void GarbageCollect ( void (*get_roots)(void), rtsBool force_major_gc ) */ if (RtsFlags.GcFlags.heapSizeSuggestion) { - int blocks; - nat needed = calcNeeded(); /* approx blocks needed at next GC */ + long blocks; + nat needed = calcNeeded(); // approx blocks needed at next GC /* Guess how much will be live in generation 0 step 0 next time. - * A good approximation is the obtained by finding the + * A good approximation is obtained by finding the * percentage of g0s0 that was live at the last minor GC. */ if (N == 0) { @@ -707,63 +1010,90 @@ void GarbageCollect ( void (*get_roots)(void), rtsBool force_major_gc ) * collection for collecting all steps except g0s0. */ blocks = - (((int)RtsFlags.GcFlags.heapSizeSuggestion - (int)needed) * 100) / - (100 + (int)g0s0_pcnt_kept); + (((long)RtsFlags.GcFlags.heapSizeSuggestion - (long)needed) * 100) / + (100 + (long)g0s0_pcnt_kept); - if (blocks < (int)RtsFlags.GcFlags.minAllocAreaSize) { + if (blocks < (long)RtsFlags.GcFlags.minAllocAreaSize) { blocks = RtsFlags.GcFlags.minAllocAreaSize; } resizeNursery((nat)blocks); + + } else { + // we might have added extra large blocks to the nursery, so + // resize back to minAllocAreaSize again. + resizeNursery(RtsFlags.GcFlags.minAllocAreaSize); } } - /* mark the garbage collected CAFs as dead */ -#ifdef DEBUG + // mark the garbage collected CAFs as dead +#if 0 && defined(DEBUG) // doesn't work at the moment if (major_gc) { gcCAFs(); } #endif - /* zero the scavenged static object list */ +#ifdef PROFILING + // resetStaticObjectForRetainerProfiling() must be called before + // zeroing below. + resetStaticObjectForRetainerProfiling(); +#endif + + // zero the scavenged static object list if (major_gc) { zero_static_object_list(scavenged_static_objects); } - /* Reset the nursery - */ + // Reset the nursery resetNurseries(); - /* start any pending finalizers */ + RELEASE_LOCK(&sched_mutex); + + // start any pending finalizers scheduleFinalizers(old_weak_ptr_list); - /* send exceptions to any threads which were about to die */ + // send exceptions to any threads which were about to die resurrectThreads(resurrected_threads); + + ACQUIRE_LOCK(&sched_mutex); + + // Update the stable pointer hash table. + updateStablePtrTable(major_gc); - /* check sanity after GC */ - IF_DEBUG(sanity, checkSanity(N)); + // check sanity after GC + IF_DEBUG(sanity, checkSanity()); - /* extra GC trace info */ - IF_DEBUG(gc, stat_describe_gens()); + // extra GC trace info + IF_DEBUG(gc, statDescribeGens()); #ifdef DEBUG - /* symbol-table based profiling */ - /* heapCensus(to_space); */ /* ToDo */ + // symbol-table based profiling + /* heapCensus(to_blocks); */ /* ToDo */ #endif - /* restore enclosing cost centre */ + // restore enclosing cost centre #ifdef PROFILING - heapCensus(); CCCS = prev_CCS; #endif - /* check for memory leaks if sanity checking is on */ + // check for memory leaks if sanity checking is on IF_DEBUG(sanity, memInventory()); - /* ok, GC over: tell the stats department what happened. */ +#ifdef RTS_GTK_FRONTPANEL + if (RtsFlags.GcFlags.frontpanel) { + updateFrontPanelAfterGC( N, live ); + } +#endif + + // ok, GC over: tell the stats department what happened. stat_endGC(allocated, collected, live, copied, N); + +#if defined(RTS_USER_SIGNALS) + // unblock signals again + unblockUserSignals(); +#endif + + //PAR_TICKY_TP(); } -//@node Weak Pointers, Evacuation, Garbage Collect -//@subsection Weak Pointers /* ----------------------------------------------------------------------------- Weak Pointers @@ -783,8 +1113,31 @@ void GarbageCollect ( void (*get_roots)(void), rtsBool force_major_gc ) older generations than the one we're collecting. This could probably be optimised by keeping per-generation lists of weak pointers, but for a few weak pointers this scheme will work. + + There are three distinct stages to processing weak pointers: + + - weak_stage == WeakPtrs + + We process all the weak pointers whos keys are alive (evacuate + their values and finalizers), and repeat until we can find no new + live keys. If no live keys are found in this pass, then we + evacuate the finalizers of all the dead weak pointers in order to + run them. + + - weak_stage == WeakThreads + + Now, we discover which *threads* are still alive. Pointers to + threads from the all_threads and main thread lists are the + weakest of all: a pointers from the finalizer of a dead weak + pointer can keep a thread alive. Any threads found to be unreachable + are evacuated and placed on the resurrected_threads list so we + can send them a signal later. + + - weak_stage == WeakDone + + No more evacuation is done. + -------------------------------------------------------------------------- */ -//@cindex traverse_weak_ptr_list static rtsBool traverse_weak_ptr_list(void) @@ -793,123 +1146,155 @@ traverse_weak_ptr_list(void) StgClosure *new; rtsBool flag = rtsFalse; - if (weak_done) { return rtsFalse; } - - /* doesn't matter where we evacuate values/finalizers to, since - * these pointers are treated as roots (iff the keys are alive). - */ - evac_gen = 0; - - last_w = &old_weak_ptr_list; - for (w = old_weak_ptr_list; w; w = next_w) { - - /* First, this weak pointer might have been evacuated. If so, - * remove the forwarding pointer from the weak_ptr_list. - */ - if (get_itbl(w)->type == EVACUATED) { - w = (StgWeak *)((StgEvacuated *)w)->evacuee; - *last_w = w; - } + switch (weak_stage) { - /* There might be a DEAD_WEAK on the list if finalizeWeak# was - * called on a live weak pointer object. Just remove it. - */ - if (w->header.info == &DEAD_WEAK_info) { - next_w = ((StgDeadWeak *)w)->link; - *last_w = next_w; - continue; - } + case WeakDone: + return rtsFalse; - ASSERT(get_itbl(w)->type == WEAK); + case WeakPtrs: + /* doesn't matter where we evacuate values/finalizers to, since + * these pointers are treated as roots (iff the keys are alive). + */ + evac_gen = 0; + + last_w = &old_weak_ptr_list; + for (w = old_weak_ptr_list; w != NULL; w = next_w) { + + /* There might be a DEAD_WEAK on the list if finalizeWeak# was + * called on a live weak pointer object. Just remove it. + */ + if (w->header.info == &stg_DEAD_WEAK_info) { + next_w = ((StgDeadWeak *)w)->link; + *last_w = next_w; + continue; + } + + switch (get_itbl(w)->type) { + + case EVACUATED: + next_w = (StgWeak *)((StgEvacuated *)w)->evacuee; + *last_w = next_w; + continue; + + case WEAK: + /* Now, check whether the key is reachable. + */ + new = isAlive(w->key); + if (new != NULL) { + w->key = new; + // evacuate the value and finalizer + w->value = evacuate(w->value); + w->finalizer = evacuate(w->finalizer); + // remove this weak ptr from the old_weak_ptr list + *last_w = w->link; + // and put it on the new weak ptr list + next_w = w->link; + w->link = weak_ptr_list; + weak_ptr_list = w; + flag = rtsTrue; + IF_DEBUG(weak, debugBelch("Weak pointer still alive at %p -> %p", + w, w->key)); + continue; + } + else { + last_w = &(w->link); + next_w = w->link; + continue; + } - /* Now, check whether the key is reachable. - */ - if ((new = isAlive(w->key))) { - w->key = new; - /* evacuate the value and finalizer */ - w->value = evacuate(w->value); - w->finalizer = evacuate(w->finalizer); - /* remove this weak ptr from the old_weak_ptr list */ - *last_w = w->link; - /* and put it on the new weak ptr list */ - next_w = w->link; - w->link = weak_ptr_list; - weak_ptr_list = w; - flag = rtsTrue; - IF_DEBUG(weak, fprintf(stderr,"Weak pointer still alive at %p -> %p\n", w, w->key)); - continue; - } - else { - last_w = &(w->link); - next_w = w->link; - continue; - } - } + default: + barf("traverse_weak_ptr_list: not WEAK"); + } + } + + /* If we didn't make any changes, then we can go round and kill all + * the dead weak pointers. The old_weak_ptr list is used as a list + * of pending finalizers later on. + */ + if (flag == rtsFalse) { + for (w = old_weak_ptr_list; w; w = w->link) { + w->finalizer = evacuate(w->finalizer); + } - /* Now deal with the all_threads list, which behaves somewhat like - * the weak ptr list. If we discover any threads that are about to - * become garbage, we wake them up and administer an exception. - */ - { - StgTSO *t, *tmp, *next, **prev; + // Next, move to the WeakThreads stage after fully + // scavenging the finalizers we've just evacuated. + weak_stage = WeakThreads; + } - prev = &old_all_threads; - for (t = old_all_threads; t != END_TSO_QUEUE; t = next) { + return rtsTrue; - /* Threads which have finished or died get dropped from - * the list. + case WeakThreads: + /* Now deal with the all_threads list, which behaves somewhat like + * the weak ptr list. If we discover any threads that are about to + * become garbage, we wake them up and administer an exception. */ - switch (t->what_next) { - case ThreadKilled: - case ThreadComplete: - next = t->global_link; - *prev = next; - continue; - default: + { + StgTSO *t, *tmp, *next, **prev; + + prev = &old_all_threads; + for (t = old_all_threads; t != END_TSO_QUEUE; t = next) { + + tmp = (StgTSO *)isAlive((StgClosure *)t); + + if (tmp != NULL) { + t = tmp; + } + + ASSERT(get_itbl(t)->type == TSO); + switch (t->what_next) { + case ThreadRelocated: + next = t->link; + *prev = next; + continue; + case ThreadKilled: + case ThreadComplete: + // finshed or died. The thread might still be alive, but we + // don't keep it on the all_threads list. Don't forget to + // stub out its global_link field. + next = t->global_link; + t->global_link = END_TSO_QUEUE; + *prev = next; + continue; + default: + ; + } + + if (tmp == NULL) { + // not alive (yet): leave this thread on the + // old_all_threads list. + prev = &(t->global_link); + next = t->global_link; + } + else { + // alive: move this thread onto the all_threads list. + next = t->global_link; + t->global_link = all_threads; + all_threads = t; + *prev = next; + } + } } - - /* Threads which have already been determined to be alive are - * moved onto the all_threads list. + + /* And resurrect any threads which were about to become garbage. */ - (StgClosure *)tmp = isAlive((StgClosure *)t); - if (tmp != NULL) { - next = tmp->global_link; - tmp->global_link = all_threads; - all_threads = tmp; - *prev = next; - } else { - prev = &(t->global_link); - next = t->global_link; - } - } - } - - /* If we didn't make any changes, then we can go round and kill all - * the dead weak pointers. The old_weak_ptr list is used as a list - * of pending finalizers later on. - */ - if (flag == rtsFalse) { - cleanup_weak_ptr_list(&old_weak_ptr_list); - for (w = old_weak_ptr_list; w; w = w->link) { - w->finalizer = evacuate(w->finalizer); - } - - /* And resurrect any threads which were about to become garbage. - */ - { - StgTSO *t, *tmp, *next; - for (t = old_all_threads; t != END_TSO_QUEUE; t = next) { - next = t->global_link; - (StgClosure *)tmp = evacuate((StgClosure *)t); - tmp->global_link = resurrected_threads; - resurrected_threads = tmp; + { + StgTSO *t, *tmp, *next; + for (t = old_all_threads; t != END_TSO_QUEUE; t = next) { + next = t->global_link; + tmp = (StgTSO *)evacuate((StgClosure *)t); + tmp->global_link = resurrected_threads; + resurrected_threads = tmp; + } } - } + + weak_stage = WeakDone; // *now* we're done, + return rtsTrue; // but one more round of scavenging, please - weak_done = rtsTrue; + default: + barf("traverse_weak_ptr_list"); + return rtsTrue; } - return rtsTrue; } /* ----------------------------------------------------------------------------- @@ -924,26 +1309,20 @@ traverse_weak_ptr_list(void) evacuated need to be evacuated now. -------------------------------------------------------------------------- */ -//@cindex cleanup_weak_ptr_list static void -cleanup_weak_ptr_list ( StgWeak **list ) +mark_weak_ptr_list ( StgWeak **list ) { StgWeak *w, **last_w; last_w = list; for (w = *list; w; w = w->link) { - - if (get_itbl(w)->type == EVACUATED) { - w = (StgWeak *)((StgEvacuated *)w)->evacuee; - *last_w = w; - } - - if (Bdescr((P_)w)->evacuated == 0) { - (StgClosure *)w = evacuate((StgClosure *)w); + // w might be WEAK, EVACUATED, or DEAD_WEAK (actually CON_STATIC) here + ASSERT(w->header.info == &stg_DEAD_WEAK_info + || get_itbl(w)->type == WEAK || get_itbl(w)->type == EVACUATED); + w = (StgWeak *)evacuate((StgClosure *)w); *last_w = w; - } - last_w = &(w->link); + last_w = &(w->link); } } @@ -951,130 +1330,108 @@ cleanup_weak_ptr_list ( StgWeak **list ) isAlive determines whether the given closure is still alive (after a garbage collection) or not. It returns the new address of the closure if it is alive, or NULL otherwise. + + NOTE: Use it before compaction only! -------------------------------------------------------------------------- */ -//@cindex isAlive StgClosure * isAlive(StgClosure *p) { const StgInfoTable *info; - nat size; + bdescr *bd; while (1) { + ASSERT(LOOKS_LIKE_CLOSURE_PTR(p)); info = get_itbl(p); - /* ToDo: for static closures, check the static link field. - * Problem here is that we sometimes don't set the link field, eg. - * for static closures with an empty SRT or CONSTR_STATIC_NOCAFs. - */ + // ignore static closures + // + // ToDo: for static closures, check the static link field. + // Problem here is that we sometimes don't set the link field, eg. + // for static closures with an empty SRT or CONSTR_STATIC_NOCAFs. + // + if (!HEAP_ALLOCED(p)) { + return p; + } - /* ignore closures in generations that we're not collecting. */ - if (LOOKS_LIKE_STATIC(p) || Bdescr((P_)p)->gen->no > N) { - return p; + // ignore closures in generations that we're not collecting. + bd = Bdescr((P_)p); + if (bd->gen_no > N) { + return p; } - + + // if it's a pointer into to-space, then we're done + if (bd->flags & BF_EVACUATED) { + return p; + } + + // large objects use the evacuated flag + if (bd->flags & BF_LARGE) { + return NULL; + } + + // check the mark bit for compacted steps + if ((bd->flags & BF_COMPACTED) && is_marked((P_)p,bd)) { + return p; + } + switch (info->type) { - + case IND: case IND_STATIC: case IND_PERM: - case IND_OLDGEN: /* rely on compatible layout with StgInd */ + case IND_OLDGEN: // rely on compatible layout with StgInd case IND_OLDGEN_PERM: - /* follow indirections */ + // follow indirections p = ((StgInd *)p)->indirectee; continue; - + case EVACUATED: - /* alive! */ + // alive! return ((StgEvacuated *)p)->evacuee; - case BCO: - size = bco_sizeW((StgBCO*)p); - goto large; - - case ARR_WORDS: - size = arr_words_sizeW((StgArrWords *)p); - goto large; - - case MUT_ARR_PTRS: - case MUT_ARR_PTRS_FROZEN: - size = mut_arr_ptrs_sizeW((StgMutArrPtrs *)p); - goto large; - case TSO: if (((StgTSO *)p)->what_next == ThreadRelocated) { p = (StgClosure *)((StgTSO *)p)->link; continue; - } - - size = tso_sizeW((StgTSO *)p); - large: - if (size >= LARGE_OBJECT_THRESHOLD/sizeof(W_) - && Bdescr((P_)p)->evacuated) - return p; - else - return NULL; + } + return NULL; default: - /* dead. */ + // dead. return NULL; } } } -//@cindex MarkRoot -StgClosure * -MarkRoot(StgClosure *root) -{ -# if 0 && defined(PAR) && defined(DEBUG) - StgClosure *foo = evacuate(root); - // ASSERT(closure_STATIC(foo) || maybeLarge(foo) || Bdescr(foo)->evacuated); - ASSERT(isAlive(foo)); // must be in to-space - return foo; -# else - return evacuate(root); -# endif -} - -//@cindex addBlock -static void addBlock(step *step) +static void +mark_root(StgClosure **root) { - bdescr *bd = allocBlock(); - bd->gen = step->gen; - bd->step = step; - - if (step->gen->no <= N) { - bd->evacuated = 1; - } else { - bd->evacuated = 0; - } - - step->hp_bd->free = step->hp; - step->hp_bd->link = bd; - step->hp = bd->start; - step->hpLim = step->hp + BLOCK_SIZE_W; - step->hp_bd = bd; - step->to_blocks++; - new_blocks++; + *root = evacuate(*root); } -//@cindex upd_evacuee - -static __inline__ void +STATIC_INLINE void upd_evacuee(StgClosure *p, StgClosure *dest) { - p->header.info = &EVACUATED_info; - ((StgEvacuated *)p)->evacuee = dest; + // Source object must be in from-space: + ASSERT((Bdescr((P_)p)->flags & BF_EVACUATED) == 0); + // not true: (ToDo: perhaps it should be) + // ASSERT(Bdescr((P_)dest)->flags & BF_EVACUATED); + SET_INFO(p, &stg_EVACUATED_info); + ((StgEvacuated *)p)->evacuee = dest; } -//@cindex copy -static __inline__ StgClosure * -copy(StgClosure *src, nat size, step *step) +STATIC_INLINE StgClosure * +copy(StgClosure *src, nat size, step *stp) { P_ to, from, dest; +#ifdef PROFILING + // @LDV profiling + nat size_org = size; +#endif TICK_GC_WORDS_COPIED(size); /* Find out where we're going, using the handy "to" pointer in @@ -1082,28 +1439,33 @@ copy(StgClosure *src, nat size, step *step) * evacuate to an older generation, adjust it here (see comment * by evacuate()). */ - if (step->gen->no < evac_gen) { + if (stp->gen_no < evac_gen) { #ifdef NO_EAGER_PROMOTION failed_to_evac = rtsTrue; #else - step = &generations[evac_gen].steps[0]; + stp = &generations[evac_gen].steps[0]; #endif } /* chain a new block onto the to-space for the destination step if * necessary. */ - if (step->hp + size >= step->hpLim) { - addBlock(step); + if (stp->hp + size >= stp->hpLim) { + gc_alloc_block(stp); } - for(to = step->hp, from = (P_)src; size>0; --size) { + for(to = stp->hp, from = (P_)src; size>0; --size) { *to++ = *from++; } - dest = step->hp; - step->hp = to; + dest = stp->hp; + stp->hp = to; upd_evacuee(src,(StgClosure *)dest); +#ifdef PROFILING + // We store the size of the just evacuated object in the LDV word so that + // the profiler can guess the position of the next object later. + SET_EVACUAEE_FOR_LDV(src, size_org); +#endif return (StgClosure *)dest; } @@ -1112,139 +1474,111 @@ copy(StgClosure *src, nat size, step *step) * used to optimise evacuation of BLACKHOLEs. */ -//@cindex copyPart -static __inline__ StgClosure * -copyPart(StgClosure *src, nat size_to_reserve, nat size_to_copy, step *step) +static StgClosure * +copyPart(StgClosure *src, nat size_to_reserve, nat size_to_copy, step *stp) { P_ dest, to, from; +#ifdef PROFILING + // @LDV profiling + nat size_to_copy_org = size_to_copy; +#endif TICK_GC_WORDS_COPIED(size_to_copy); - if (step->gen->no < evac_gen) { + if (stp->gen_no < evac_gen) { #ifdef NO_EAGER_PROMOTION failed_to_evac = rtsTrue; #else - step = &generations[evac_gen].steps[0]; + stp = &generations[evac_gen].steps[0]; #endif } - if (step->hp + size_to_reserve >= step->hpLim) { - addBlock(step); + if (stp->hp + size_to_reserve >= stp->hpLim) { + gc_alloc_block(stp); } - for(to = step->hp, from = (P_)src; size_to_copy>0; --size_to_copy) { + for(to = stp->hp, from = (P_)src; size_to_copy>0; --size_to_copy) { *to++ = *from++; } - dest = step->hp; - step->hp += size_to_reserve; + dest = stp->hp; + stp->hp += size_to_reserve; upd_evacuee(src,(StgClosure *)dest); +#ifdef PROFILING + // We store the size of the just evacuated object in the LDV word so that + // the profiler can guess the position of the next object later. + // size_to_copy_org is wrong because the closure already occupies size_to_reserve + // words. + SET_EVACUAEE_FOR_LDV(src, size_to_reserve); + // fill the slop + if (size_to_reserve - size_to_copy_org > 0) + FILL_SLOP(stp->hp - 1, (int)(size_to_reserve - size_to_copy_org)); +#endif return (StgClosure *)dest; } -//@node Evacuation, Scavenging, Weak Pointers -//@subsection Evacuation /* ----------------------------------------------------------------------------- Evacuate a large object This just consists of removing the object from the (doubly-linked) - large_alloc_list, and linking it on to the (singly-linked) - new_large_objects list, from where it will be scavenged later. + step->large_objects list, and linking it on to the (singly-linked) + step->new_large_objects list, from where it will be scavenged later. - Convention: bd->evacuated is /= 0 for a large object that has been - evacuated, or 0 otherwise. + Convention: bd->flags has BF_EVACUATED set for a large object + that has been evacuated, or unset otherwise. -------------------------------------------------------------------------- */ -//@cindex evacuate_large -static inline void -evacuate_large(StgPtr p, rtsBool mutable) +STATIC_INLINE void +evacuate_large(StgPtr p) { bdescr *bd = Bdescr(p); - step *step; + step *stp; - /* should point to the beginning of the block */ - ASSERT(((W_)p & BLOCK_MASK) == 0); - - /* already evacuated? */ - if (bd->evacuated) { + // object must be at the beginning of the block (or be a ByteArray) + ASSERT(get_itbl((StgClosure *)p)->type == ARR_WORDS || + (((W_)p & BLOCK_MASK) == 0)); + + // already evacuated? + if (bd->flags & BF_EVACUATED) { /* Don't forget to set the failed_to_evac flag if we didn't get * the desired destination (see comments in evacuate()). */ - if (bd->gen->no < evac_gen) { + if (bd->gen_no < evac_gen) { failed_to_evac = rtsTrue; TICK_GC_FAILED_PROMOTION(); } return; } - step = bd->step; - /* remove from large_object list */ - if (bd->back) { - bd->back->link = bd->link; - } else { /* first object in the list */ - step->large_objects = bd->link; + stp = bd->step; + // remove from large_object list + if (bd->u.back) { + bd->u.back->link = bd->link; + } else { // first object in the list + stp->large_objects = bd->link; } if (bd->link) { - bd->link->back = bd->back; + bd->link->u.back = bd->u.back; } /* link it on to the evacuated large object list of the destination step */ - step = bd->step->to; - if (step->gen->no < evac_gen) { + stp = bd->step->to; + if (stp->gen_no < evac_gen) { #ifdef NO_EAGER_PROMOTION failed_to_evac = rtsTrue; #else - step = &generations[evac_gen].steps[0]; + stp = &generations[evac_gen].steps[0]; #endif } - bd->step = step; - bd->gen = step->gen; - bd->link = step->new_large_objects; - step->new_large_objects = bd; - bd->evacuated = 1; - - if (mutable) { - recordMutable((StgMutClosure *)p); - } -} - -/* ----------------------------------------------------------------------------- - Adding a MUT_CONS to an older generation. - - This is necessary from time to time when we end up with an - old-to-new generation pointer in a non-mutable object. We defer - the promotion until the next GC. - -------------------------------------------------------------------------- */ - -//@cindex mkMutCons - -static StgClosure * -mkMutCons(StgClosure *ptr, generation *gen) -{ - StgMutVar *q; - step *step; - - step = &gen->steps[0]; - - /* chain a new block onto the to-space for the destination step if - * necessary. - */ - if (step->hp + sizeofW(StgIndOldGen) >= step->hpLim) { - addBlock(step); - } - - q = (StgMutVar *)step->hp; - step->hp += sizeofW(StgMutVar); - - SET_HDR(q,&MUT_CONS_info,CCS_GC); - q->var = ptr; - recordOldToNewPtrs((StgMutClosure *)q); - - return (StgClosure *)q; + bd->step = stp; + bd->gen_no = stp->gen_no; + bd->link = stp->new_large_objects; + stp->new_large_objects = bd; + bd->flags |= BF_EVACUATED; } /* ----------------------------------------------------------------------------- @@ -1270,93 +1604,128 @@ mkMutCons(StgClosure *ptr, generation *gen) if M < evac_gen set failed_to_evac flag to indicate that we didn't manage to evacuate this object into evac_gen. + + OPTIMISATION NOTES: + + evacuate() is the single most important function performance-wise + in the GC. Various things have been tried to speed it up, but as + far as I can tell the code generated by gcc 3.2 with -O2 is about + as good as it's going to get. We pass the argument to evacuate() + in a register using the 'regparm' attribute (see the prototype for + evacuate() near the top of this file). + + Changing evacuate() to take an (StgClosure **) rather than + returning the new pointer seems attractive, because we can avoid + writing back the pointer when it hasn't changed (eg. for a static + object, or an object in a generation > N). However, I tried it and + it doesn't help. One reason is that the (StgClosure **) pointer + gets spilled to the stack inside evacuate(), resulting in far more + extra reads/writes than we save. -------------------------------------------------------------------------- */ -//@cindex evacuate -static StgClosure * +REGPARM1 static StgClosure * evacuate(StgClosure *q) { StgClosure *to; bdescr *bd = NULL; - step *step; + step *stp; const StgInfoTable *info; loop: if (HEAP_ALLOCED(q)) { bd = Bdescr((P_)q); - if (bd->gen->no > N) { - /* Can't evacuate this object, because it's in a generation - * older than the ones we're collecting. Let's hope that it's - * in evac_gen or older, or we will have to make an IND_OLDGEN object. - */ - if (bd->gen->no < evac_gen) { - /* nope */ - failed_to_evac = rtsTrue; - TICK_GC_FAILED_PROMOTION(); - } - return q; + + if (bd->gen_no > N) { + /* Can't evacuate this object, because it's in a generation + * older than the ones we're collecting. Let's hope that it's + * in evac_gen or older, or we will have to arrange to track + * this pointer using the mutable list. + */ + if (bd->gen_no < evac_gen) { + // nope + failed_to_evac = rtsTrue; + TICK_GC_FAILED_PROMOTION(); + } + return q; + } + + /* evacuate large objects by re-linking them onto a different list. + */ + if (bd->flags & BF_LARGE) { + info = get_itbl(q); + if (info->type == TSO && + ((StgTSO *)q)->what_next == ThreadRelocated) { + q = (StgClosure *)((StgTSO *)q)->link; + goto loop; + } + evacuate_large((P_)q); + return q; + } + + /* If the object is in a step that we're compacting, then we + * need to use an alternative evacuate procedure. + */ + if (bd->flags & BF_COMPACTED) { + if (!is_marked((P_)q,bd)) { + mark((P_)q,bd); + if (mark_stack_full()) { + mark_stack_overflowed = rtsTrue; + reset_mark_stack(); + } + push_mark_stack((P_)q); + } + return q; } - step = bd->step->to; + + stp = bd->step->to; } #ifdef DEBUG - else step = NULL; /* make sure copy() will crash if HEAP_ALLOCED is wrong */ + else stp = NULL; // make sure copy() will crash if HEAP_ALLOCED is wrong #endif - /* make sure the info pointer is into text space */ - ASSERT(q && (LOOKS_LIKE_GHC_INFO(GET_INFO(q)) - || IS_HUGS_CONSTR_INFO(GET_INFO(q)))); + // make sure the info pointer is into text space + ASSERT(LOOKS_LIKE_CLOSURE_PTR(q)); info = get_itbl(q); - /* - if (info->type==RBH) { - info = REVERT_INFOPTR(info); - IF_DEBUG(gc, - belch("@_ Trying to evacuate an RBH %p (%s); reverting to IP %p (%s)", - q, info_type(q), info, info_type_by_ip(info))); - } - */ switch (info -> type) { - case BCO: - { - nat size = bco_sizeW((StgBCO*)q); - - if (size >= LARGE_OBJECT_THRESHOLD/sizeof(W_)) { - evacuate_large((P_)q, rtsFalse); - to = q; - } else { - /* just copy the block */ - to = copy(q,size,step); - } - return to; - } - case MUT_VAR: - ASSERT(q->header.info != &MUT_CONS_info); case MVAR: - to = copy(q,sizeW_fromITBL(info),step); - recordMutable((StgMutClosure *)to); - return to; + return copy(q,sizeW_fromITBL(info),stp); + + case CONSTR_0_1: + { + StgWord w = (StgWord)q->payload[0]; + if (q->header.info == Czh_con_info && + // unsigned, so always true: (StgChar)w >= MIN_CHARLIKE && + (StgChar)w <= MAX_CHARLIKE) { + return (StgClosure *)CHARLIKE_CLOSURE((StgChar)w); + } + if (q->header.info == Izh_con_info && + (StgInt)w >= MIN_INTLIKE && (StgInt)w <= MAX_INTLIKE) { + return (StgClosure *)INTLIKE_CLOSURE((StgInt)w); + } + // else, fall through ... + } case FUN_1_0: case FUN_0_1: case CONSTR_1_0: - case CONSTR_0_1: - return copy(q,sizeofW(StgHeader)+1,step); - - case THUNK_1_0: /* here because of MIN_UPD_SIZE */ + case THUNK_1_0: case THUNK_0_1: + return copy(q,sizeofW(StgHeader)+1,stp); + case THUNK_1_1: case THUNK_0_2: case THUNK_2_0: #ifdef NO_PROMOTE_THUNKS - if (bd->gen->no == 0 && + if (bd->gen_no == 0 && bd->step->no != 0 && - bd->step->no == bd->gen->n_steps-1) { - step = bd->step; + bd->step->no == generations[bd->gen_no].n_steps-1) { + stp = bd->step; } #endif - return copy(q,sizeofW(StgHeader)+2,step); + return copy(q,sizeofW(StgHeader)+2,stp); case FUN_1_1: case FUN_0_2: @@ -1364,127 +1733,68 @@ loop: case CONSTR_1_1: case CONSTR_0_2: case CONSTR_2_0: - return copy(q,sizeofW(StgHeader)+2,step); + return copy(q,sizeofW(StgHeader)+2,stp); case FUN: case THUNK: case CONSTR: case IND_PERM: case IND_OLDGEN_PERM: - case CAF_UNENTERED: - case CAF_ENTERED: case WEAK: case FOREIGN: case STABLE_NAME: - return copy(q,sizeW_fromITBL(info),step); + return copy(q,sizeW_fromITBL(info),stp); + + case BCO: + return copy(q,bco_sizeW((StgBCO *)q),stp); case CAF_BLACKHOLE: case SE_CAF_BLACKHOLE: case SE_BLACKHOLE: case BLACKHOLE: - return copyPart(q,BLACKHOLE_sizeW(),sizeofW(StgHeader),step); + return copyPart(q,BLACKHOLE_sizeW(),sizeofW(StgHeader),stp); case BLACKHOLE_BQ: - to = copy(q,BLACKHOLE_sizeW(),step); - recordMutable((StgMutClosure *)to); + to = copy(q,BLACKHOLE_sizeW(),stp); return to; case THUNK_SELECTOR: { - const StgInfoTable* selectee_info; - StgClosure* selectee = ((StgSelector*)q)->selectee; - - selector_loop: - selectee_info = get_itbl(selectee); - switch (selectee_info->type) { - case CONSTR: - case CONSTR_1_0: - case CONSTR_0_1: - case CONSTR_2_0: - case CONSTR_1_1: - case CONSTR_0_2: - case CONSTR_STATIC: - { - StgWord32 offset = info->layout.selector_offset; - - /* check that the size is in range */ - ASSERT(offset < - (StgWord32)(selectee_info->layout.payload.ptrs + - selectee_info->layout.payload.nptrs)); - - /* perform the selection! */ - q = selectee->payload[offset]; - - /* if we're already in to-space, there's no need to continue - * with the evacuation, just update the source address with - * a pointer to the (evacuated) constructor field. - */ - if (HEAP_ALLOCED(q)) { - bdescr *bd = Bdescr((P_)q); - if (bd->evacuated) { - if (bd->gen->no < evac_gen) { - failed_to_evac = rtsTrue; - TICK_GC_FAILED_PROMOTION(); - } - return q; - } - } + StgClosure *p; - /* otherwise, carry on and evacuate this constructor field, - * (but not the constructor itself) - */ - goto loop; + if (thunk_selector_depth > MAX_THUNK_SELECTOR_DEPTH) { + return copy(q,THUNK_SELECTOR_sizeW(),stp); } - case IND: - case IND_STATIC: - case IND_PERM: - case IND_OLDGEN: - case IND_OLDGEN_PERM: - selectee = ((StgInd *)selectee)->indirectee; - goto selector_loop; - - case CAF_ENTERED: - selectee = ((StgCAF *)selectee)->value; - goto selector_loop; - - case EVACUATED: - selectee = ((StgEvacuated *)selectee)->evacuee; - goto selector_loop; - - case THUNK: - case THUNK_1_0: - case THUNK_0_1: - case THUNK_2_0: - case THUNK_1_1: - case THUNK_0_2: - case THUNK_STATIC: - case THUNK_SELECTOR: - /* aargh - do recursively???? */ - case CAF_UNENTERED: - case CAF_BLACKHOLE: - case SE_CAF_BLACKHOLE: - case SE_BLACKHOLE: - case BLACKHOLE: - case BLACKHOLE_BQ: - /* not evaluated yet */ - break; + p = eval_thunk_selector(info->layout.selector_offset, + (StgSelector *)q); - default: - barf("evacuate: THUNK_SELECTOR: strange selectee %d", - (int)(selectee_info->type)); - } + if (p == NULL) { + return copy(q,THUNK_SELECTOR_sizeW(),stp); + } else { + // q is still BLACKHOLE'd. + thunk_selector_depth++; + p = evacuate(p); + thunk_selector_depth--; + upd_evacuee(q,p); +#ifdef PROFILING + // We store the size of the just evacuated object in the + // LDV word so that the profiler can guess the position of + // the next object later. + SET_EVACUAEE_FOR_LDV(q, THUNK_SELECTOR_sizeW()); +#endif + return p; + } } - return copy(q,THUNK_SELECTOR_sizeW(),step); case IND: case IND_OLDGEN: - /* follow chains of indirections, don't evacuate them */ + // follow chains of indirections, don't evacuate them q = ((StgInd*)q)->indirectee; goto loop; case THUNK_STATIC: - if (info->srt_len > 0 && major_gc && + if (info->srt_bitmap != 0 && major_gc && THUNK_STATIC_LINK((StgClosure *)q) == NULL) { THUNK_STATIC_LINK((StgClosure *)q) = static_objects; static_objects = (StgClosure *)q; @@ -1492,7 +1802,7 @@ loop: return q; case FUN_STATIC: - if (info->srt_len > 0 && major_gc && + if (info->srt_bitmap != 0 && major_gc && FUN_STATIC_LINK((StgClosure *)q) == NULL) { FUN_STATIC_LINK((StgClosure *)q) = static_objects; static_objects = (StgClosure *)q; @@ -1500,9 +1810,15 @@ loop: return q; case IND_STATIC: - if (major_gc && IND_STATIC_LINK((StgClosure *)q) == NULL) { - IND_STATIC_LINK((StgClosure *)q) = static_objects; - static_objects = (StgClosure *)q; + /* If q->saved_info != NULL, then it's a revertible CAF - it'll be + * on the CAF list, so don't do anything with it here (we'll + * scavenge it later). + */ + if (major_gc + && ((StgIndStatic *)q)->saved_info == NULL + && IND_STATIC_LINK((StgClosure *)q) == NULL) { + IND_STATIC_LINK((StgClosure *)q) = static_objects; + static_objects = (StgClosure *)q; } return q; @@ -1530,28 +1846,18 @@ loop: case UPDATE_FRAME: case STOP_FRAME: case CATCH_FRAME: - case SEQ_FRAME: - /* shouldn't see these */ + case CATCH_STM_FRAME: + case CATCH_RETRY_FRAME: + case ATOMICALLY_FRAME: + // shouldn't see these barf("evacuate: stack frame at %p\n", q); - case AP_UPD: case PAP: - /* PAPs and AP_UPDs are special - the payload is a copy of a chunk - * of stack, tagging and all. - * - * They can be larger than a block in size. Both are only - * allocated via allocate(), so they should be chained on to the - * large_object list. - */ - { - nat size = pap_sizeW((StgPAP*)q); - if (size >= LARGE_OBJECT_THRESHOLD/sizeof(W_)) { - evacuate_large((P_)q, rtsFalse); - return q; - } else { - return copy(q,size,step); - } - } + case AP: + return copy(q,pap_sizeW((StgPAP*)q),stp); + + case AP_STACK: + return copy(q,ap_stack_sizeW((StgAP_STACK*)q),stp); case EVACUATED: /* Already evacuated, just return the forwarding address. @@ -1561,10 +1867,9 @@ loop: * set the failed_to_evac flag to indicate that we couldn't * manage to promote the object to the desired generation. */ - if (evac_gen > 0) { /* optimisation */ + if (evac_gen > 0) { // optimisation StgClosure *p = ((StgEvacuated*)q)->evacuee; - if (Bdescr((P_)p)->gen->no < evac_gen) { - IF_DEBUG(gc, belch("@@ evacuate: evac of EVACUATED node %p failed!", p)); + if (HEAP_ALLOCED(p) && Bdescr((P_)p)->gen_no < evac_gen) { failed_to_evac = rtsTrue; TICK_GC_FAILED_PROMOTION(); } @@ -1572,41 +1877,18 @@ loop: return ((StgEvacuated*)q)->evacuee; case ARR_WORDS: - { - nat size = arr_words_sizeW((StgArrWords *)q); - - if (size >= LARGE_OBJECT_THRESHOLD/sizeof(W_)) { - evacuate_large((P_)q, rtsFalse); - return q; - } else { - /* just copy the block */ - return copy(q,size,step); - } - } + // just copy the block + return copy(q,arr_words_sizeW((StgArrWords *)q),stp); case MUT_ARR_PTRS: case MUT_ARR_PTRS_FROZEN: - { - nat size = mut_arr_ptrs_sizeW((StgMutArrPtrs *)q); - - if (size >= LARGE_OBJECT_THRESHOLD/sizeof(W_)) { - evacuate_large((P_)q, info->type == MUT_ARR_PTRS); - to = q; - } else { - /* just copy the block */ - to = copy(q,size,step); - if (info->type == MUT_ARR_PTRS) { - recordMutable((StgMutClosure *)to); - } - } - return to; - } + case MUT_ARR_PTRS_FROZEN0: + // just copy the block + return copy(q,mut_arr_ptrs_sizeW((StgMutArrPtrs *)q),stp); case TSO: { StgTSO *tso = (StgTSO *)q; - nat size = tso_sizeW(tso); - int diff; /* Deal with redirected TSOs (a TSO that's had its stack enlarged). */ @@ -1615,29 +1897,23 @@ loop: goto loop; } - /* Large TSOs don't get moved, so no relocation is required. - */ - if (size >= LARGE_OBJECT_THRESHOLD/sizeof(W_)) { - evacuate_large((P_)q, rtsTrue); - return q; - /* To evacuate a small TSO, we need to relocate the update frame * list it contains. */ - } else { - StgTSO *new_tso = (StgTSO *)copy((StgClosure *)tso,tso_sizeW(tso),step); - - diff = (StgPtr)new_tso - (StgPtr)tso; /* In *words* */ - - /* relocate the stack pointers... */ - new_tso->su = (StgUpdateFrame *) ((StgPtr)new_tso->su + diff); - new_tso->sp = (StgPtr)new_tso->sp + diff; - new_tso->splim = (StgPtr)new_tso->splim + diff; - - relocate_TSO(tso, new_tso); - - recordMutable((StgMutClosure *)new_tso); - return (StgClosure *)new_tso; + { + StgTSO *new_tso; + StgPtr p, q; + + new_tso = (StgTSO *)copyPart((StgClosure *)tso, + tso_sizeW(tso), + sizeofW(StgTSO), stp); + move_TSO(tso, new_tso); + for (p = tso->sp, q = new_tso->sp; + p < tso->stack+tso->stack_size;) { + *q++ = *p++; + } + + return (StgClosure *)new_tso; } } @@ -1645,41 +1921,55 @@ loop: case RBH: // cf. BLACKHOLE_BQ { //StgInfoTable *rip = get_closure_info(q, &size, &ptrs, &nonptrs, &vhs, str); - to = copy(q,BLACKHOLE_sizeW(),step); + to = copy(q,BLACKHOLE_sizeW(),stp); //ToDo: derive size etc from reverted IP - //to = copy(q,size,step); - recordMutable((StgMutClosure *)to); + //to = copy(q,size,stp); IF_DEBUG(gc, - belch("@@ evacuate: RBH %p (%s) to %p (%s)", + debugBelch("@@ evacuate: RBH %p (%s) to %p (%s)", q, info_type(q), to, info_type(to))); return to; } case BLOCKED_FETCH: ASSERT(sizeofW(StgBlockedFetch) >= MIN_NONUPD_SIZE); - to = copy(q,sizeofW(StgBlockedFetch),step); + to = copy(q,sizeofW(StgBlockedFetch),stp); IF_DEBUG(gc, - belch("@@ evacuate: %p (%s) to %p (%s)", + debugBelch("@@ evacuate: %p (%s) to %p (%s)", q, info_type(q), to, info_type(to))); return to; +# ifdef DIST + case REMOTE_REF: +# endif case FETCH_ME: ASSERT(sizeofW(StgBlockedFetch) >= MIN_UPD_SIZE); - to = copy(q,sizeofW(StgFetchMe),step); + to = copy(q,sizeofW(StgFetchMe),stp); IF_DEBUG(gc, - belch("@@ evacuate: %p (%s) to %p (%s)", + debugBelch("@@ evacuate: %p (%s) to %p (%s)", q, info_type(q), to, info_type(to))); return to; case FETCH_ME_BQ: ASSERT(sizeofW(StgBlockedFetch) >= MIN_UPD_SIZE); - to = copy(q,sizeofW(StgFetchMeBlockingQueue),step); + to = copy(q,sizeofW(StgFetchMeBlockingQueue),stp); IF_DEBUG(gc, - belch("@@ evacuate: %p (%s) to %p (%s)", + debugBelch("@@ evacuate: %p (%s) to %p (%s)", q, info_type(q), to, info_type(to))); return to; #endif + case TREC_HEADER: + return copy(q,sizeofW(StgTRecHeader),stp); + + case TVAR_WAIT_QUEUE: + return copy(q,sizeofW(StgTVarWaitQueue),stp); + + case TVAR: + return copy(q,sizeofW(StgTVar),stp); + + case TREC_CHUNK: + return copy(q,sizeofW(StgTRecChunk),stp); + default: barf("evacuate: strange closure type %d", (int)(info->type)); } @@ -1688,94 +1978,331 @@ loop: } /* ----------------------------------------------------------------------------- - relocate_TSO is called just after a TSO has been copied from src to - dest. It adjusts the update frame list for the new location. + Evaluate a THUNK_SELECTOR if possible. + + returns: NULL if we couldn't evaluate this THUNK_SELECTOR, or + a closure pointer if we evaluated it and this is the result. Note + that "evaluating" the THUNK_SELECTOR doesn't necessarily mean + reducing it to HNF, just that we have eliminated the selection. + The result might be another thunk, or even another THUNK_SELECTOR. + + If the return value is non-NULL, the original selector thunk has + been BLACKHOLE'd, and should be updated with an indirection or a + forwarding pointer. If the return value is NULL, then the selector + thunk is unchanged. -------------------------------------------------------------------------- */ -//@cindex relocate_TSO -StgTSO * -relocate_TSO(StgTSO *src, StgTSO *dest) +static inline rtsBool +is_to_space ( StgClosure *p ) { - StgUpdateFrame *su; - StgCatchFrame *cf; - StgSeqFrame *sf; - int diff; - - diff = (StgPtr)dest->sp - (StgPtr)src->sp; /* In *words* */ + bdescr *bd; + + bd = Bdescr((StgPtr)p); + if (HEAP_ALLOCED(p) && + ((bd->flags & BF_EVACUATED) + || ((bd->flags & BF_COMPACTED) && + is_marked((P_)p,bd)))) { + return rtsTrue; + } else { + return rtsFalse; + } +} - su = dest->su; +static StgClosure * +eval_thunk_selector( nat field, StgSelector * p ) +{ + StgInfoTable *info; + const StgInfoTable *info_ptr; + StgClosure *selectee; + + selectee = p->selectee; - while ((P_)su < dest->stack + dest->stack_size) { - switch (get_itbl(su)->type) { - - /* GCC actually manages to common up these three cases! */ + // Save the real info pointer (NOTE: not the same as get_itbl()). + info_ptr = p->header.info; - case UPDATE_FRAME: - su->link = (StgUpdateFrame *) ((StgPtr)su->link + diff); - su = su->link; - continue; + // If the THUNK_SELECTOR is in a generation that we are not + // collecting, then bail out early. We won't be able to save any + // space in any case, and updating with an indirection is trickier + // in an old gen. + if (Bdescr((StgPtr)p)->gen_no > N) { + return NULL; + } - case CATCH_FRAME: - cf = (StgCatchFrame *)su; - cf->link = (StgUpdateFrame *) ((StgPtr)cf->link + diff); - su = cf->link; - continue; + // BLACKHOLE the selector thunk, since it is now under evaluation. + // This is important to stop us going into an infinite loop if + // this selector thunk eventually refers to itself. + SET_INFO(p,&stg_BLACKHOLE_info); + +selector_loop: + + // We don't want to end up in to-space, because this causes + // problems when the GC later tries to evacuate the result of + // eval_thunk_selector(). There are various ways this could + // happen: + // + // 1. following an IND_STATIC + // + // 2. when the old generation is compacted, the mark phase updates + // from-space pointers to be to-space pointers, and we can't + // reliably tell which we're following (eg. from an IND_STATIC). + // + // 3. compacting GC again: if we're looking at a constructor in + // the compacted generation, it might point directly to objects + // in to-space. We must bale out here, otherwise doing the selection + // will result in a to-space pointer being returned. + // + // (1) is dealt with using a BF_EVACUATED test on the + // selectee. (2) and (3): we can tell if we're looking at an + // object in the compacted generation that might point to + // to-space objects by testing that (a) it is BF_COMPACTED, (b) + // the compacted generation is being collected, and (c) the + // object is marked. Only a marked object may have pointers that + // point to to-space objects, because that happens when + // scavenging. + // + // The to-space test is now embodied in the in_to_space() inline + // function, as it is re-used below. + // + if (is_to_space(selectee)) { + goto bale_out; + } - case SEQ_FRAME: - sf = (StgSeqFrame *)su; - sf->link = (StgUpdateFrame *) ((StgPtr)sf->link + diff); - su = sf->link; - continue; + info = get_itbl(selectee); + switch (info->type) { + case CONSTR: + case CONSTR_1_0: + case CONSTR_0_1: + case CONSTR_2_0: + case CONSTR_1_1: + case CONSTR_0_2: + case CONSTR_STATIC: + case CONSTR_NOCAF_STATIC: + // check that the size is in range + ASSERT(field < (StgWord32)(info->layout.payload.ptrs + + info->layout.payload.nptrs)); + + // Select the right field from the constructor, and check + // that the result isn't in to-space. It might be in + // to-space if, for example, this constructor contains + // pointers to younger-gen objects (and is on the mut-once + // list). + // + { + StgClosure *q; + q = selectee->payload[field]; + if (is_to_space(q)) { + goto bale_out; + } else { + return q; + } + } - case STOP_FRAME: - /* all done! */ - break; + case IND: + case IND_PERM: + case IND_OLDGEN: + case IND_OLDGEN_PERM: + case IND_STATIC: + selectee = ((StgInd *)selectee)->indirectee; + goto selector_loop; - default: - barf("relocate_TSO %d", (int)(get_itbl(su)->type)); - } - break; - } + case EVACUATED: + // We don't follow pointers into to-space; the constructor + // has already been evacuated, so we won't save any space + // leaks by evaluating this selector thunk anyhow. + break; - return dest; -} + case THUNK_SELECTOR: + { + StgClosure *val; -//@node Scavenging, Reverting CAFs, Evacuation -//@subsection Scavenging + // check that we don't recurse too much, re-using the + // depth bound also used in evacuate(). + thunk_selector_depth++; + if (thunk_selector_depth > MAX_THUNK_SELECTOR_DEPTH) { + break; + } -//@cindex scavenge_srt + val = eval_thunk_selector(info->layout.selector_offset, + (StgSelector *)selectee); -static inline void -scavenge_srt(const StgInfoTable *info) -{ - StgClosure **srt, **srt_end; + thunk_selector_depth--; - /* evacuate the SRT. If srt_len is zero, then there isn't an - * srt field in the info table. That's ok, because we'll - * never dereference it. - */ - srt = (StgClosure **)(info->srt); - srt_end = srt + info->srt_len; - for (; srt < srt_end; srt++) { - /* Special-case to handle references to closures hiding out in DLLs, since - double indirections required to get at those. The code generator knows - which is which when generating the SRT, so it stores the (indirect) - reference to the DLL closure in the table by first adding one to it. - We check for this here, and undo the addition before evacuating it. - - If the SRT entry hasn't got bit 0 set, the SRT entry points to a - closure that's fixed at link-time, and no extra magic is required. - */ -#ifdef ENABLE_WIN32_DLL_SUPPORT - if ( (unsigned long)(*srt) & 0x1 ) { - evacuate(*stgCast(StgClosure**,(stgCast(unsigned long, *srt) & ~0x1))); - } else { - evacuate(*srt); - } -#else - evacuate(*srt); + if (val == NULL) { + break; + } else { + // We evaluated this selector thunk, so update it with + // an indirection. NOTE: we don't use UPD_IND here, + // because we are guaranteed that p is in a generation + // that we are collecting, and we never want to put the + // indirection on a mutable list. +#ifdef PROFILING + // For the purposes of LDV profiling, we have destroyed + // the original selector thunk. + SET_INFO(p, info_ptr); + LDV_RECORD_DEAD_FILL_SLOP_DYNAMIC(selectee); #endif - } + ((StgInd *)selectee)->indirectee = val; + SET_INFO(selectee,&stg_IND_info); + + // For the purposes of LDV profiling, we have created an + // indirection. + LDV_RECORD_CREATE(selectee); + + selectee = val; + goto selector_loop; + } + } + + case AP: + case AP_STACK: + case THUNK: + case THUNK_1_0: + case THUNK_0_1: + case THUNK_2_0: + case THUNK_1_1: + case THUNK_0_2: + case THUNK_STATIC: + case CAF_BLACKHOLE: + case SE_CAF_BLACKHOLE: + case SE_BLACKHOLE: + case BLACKHOLE: + case BLACKHOLE_BQ: +#if defined(PAR) + case RBH: + case BLOCKED_FETCH: +# ifdef DIST + case REMOTE_REF: +# endif + case FETCH_ME: + case FETCH_ME_BQ: +#endif + // not evaluated yet + break; + + default: + barf("eval_thunk_selector: strange selectee %d", + (int)(info->type)); + } + +bale_out: + // We didn't manage to evaluate this thunk; restore the old info pointer + SET_INFO(p, info_ptr); + return NULL; +} + +/* ----------------------------------------------------------------------------- + move_TSO is called to update the TSO structure after it has been + moved from one place to another. + -------------------------------------------------------------------------- */ + +void +move_TSO (StgTSO *src, StgTSO *dest) +{ + ptrdiff_t diff; + + // relocate the stack pointer... + diff = (StgPtr)dest - (StgPtr)src; // In *words* + dest->sp = (StgPtr)dest->sp + diff; +} + +/* Similar to scavenge_large_bitmap(), but we don't write back the + * pointers we get back from evacuate(). + */ +static void +scavenge_large_srt_bitmap( StgLargeSRT *large_srt ) +{ + nat i, b, size; + StgWord bitmap; + StgClosure **p; + + b = 0; + bitmap = large_srt->l.bitmap[b]; + size = (nat)large_srt->l.size; + p = (StgClosure **)large_srt->srt; + for (i = 0; i < size; ) { + if ((bitmap & 1) != 0) { + evacuate(*p); + } + i++; + p++; + if (i % BITS_IN(W_) == 0) { + b++; + bitmap = large_srt->l.bitmap[b]; + } else { + bitmap = bitmap >> 1; + } + } +} + +/* evacuate the SRT. If srt_bitmap is zero, then there isn't an + * srt field in the info table. That's ok, because we'll + * never dereference it. + */ +STATIC_INLINE void +scavenge_srt (StgClosure **srt, nat srt_bitmap) +{ + nat bitmap; + StgClosure **p; + + bitmap = srt_bitmap; + p = srt; + + if (bitmap == (StgHalfWord)(-1)) { + scavenge_large_srt_bitmap( (StgLargeSRT *)srt ); + return; + } + + while (bitmap != 0) { + if ((bitmap & 1) != 0) { +#ifdef ENABLE_WIN32_DLL_SUPPORT + // Special-case to handle references to closures hiding out in DLLs, since + // double indirections required to get at those. The code generator knows + // which is which when generating the SRT, so it stores the (indirect) + // reference to the DLL closure in the table by first adding one to it. + // We check for this here, and undo the addition before evacuating it. + // + // If the SRT entry hasn't got bit 0 set, the SRT entry points to a + // closure that's fixed at link-time, and no extra magic is required. + if ( (unsigned long)(*srt) & 0x1 ) { + evacuate(*stgCast(StgClosure**,(stgCast(unsigned long, *srt) & ~0x1))); + } else { + evacuate(*p); + } +#else + evacuate(*p); +#endif + } + p++; + bitmap = bitmap >> 1; + } +} + + +STATIC_INLINE void +scavenge_thunk_srt(const StgInfoTable *info) +{ + StgThunkInfoTable *thunk_info; + + thunk_info = itbl_to_thunk_itbl(info); + scavenge_srt((StgClosure **)GET_SRT(thunk_info), thunk_info->i.srt_bitmap); +} + +STATIC_INLINE void +scavenge_fun_srt(const StgInfoTable *info) +{ + StgFunInfoTable *fun_info; + + fun_info = itbl_to_fun_itbl(info); + scavenge_srt((StgClosure **)GET_FUN_SRT(fun_info), fun_info->i.srt_bitmap); +} + +STATIC_INLINE void +scavenge_ret_srt(const StgInfoTable *info) +{ + StgRetInfoTable *ret_info; + + ret_info = itbl_to_ret_itbl(info); + scavenge_srt((StgClosure **)GET_SRT(ret_info), ret_info->i.srt_bitmap); } /* ----------------------------------------------------------------------------- @@ -1785,24 +2312,111 @@ scavenge_srt(const StgInfoTable *info) static void scavengeTSO (StgTSO *tso) { - /* chase the link field for any TSOs on the same queue */ - (StgClosure *)tso->link = evacuate((StgClosure *)tso->link); - if ( tso->why_blocked == BlockedOnMVar - || tso->why_blocked == BlockedOnBlackHole - || tso->why_blocked == BlockedOnException + // chase the link field for any TSOs on the same queue + tso->link = (StgTSO *)evacuate((StgClosure *)tso->link); + if ( tso->why_blocked == BlockedOnMVar + || tso->why_blocked == BlockedOnBlackHole + || tso->why_blocked == BlockedOnException #if defined(PAR) - || tso->why_blocked == BlockedOnGA - || tso->why_blocked == BlockedOnGA_NoSend + || tso->why_blocked == BlockedOnGA + || tso->why_blocked == BlockedOnGA_NoSend #endif - ) { - tso->block_info.closure = evacuate(tso->block_info.closure); - } - if ( tso->blocked_exceptions != NULL ) { - tso->blocked_exceptions = - (StgTSO *)evacuate((StgClosure *)tso->blocked_exceptions); - } - /* scavenge this thread's stack */ - scavenge_stack(tso->sp, &(tso->stack[tso->stack_size])); + ) { + tso->block_info.closure = evacuate(tso->block_info.closure); + } + if ( tso->blocked_exceptions != NULL ) { + tso->blocked_exceptions = + (StgTSO *)evacuate((StgClosure *)tso->blocked_exceptions); + } + + // scavange current transaction record + tso->trec = (StgTRecHeader *)evacuate((StgClosure *)tso->trec); + + // scavenge this thread's stack + scavenge_stack(tso->sp, &(tso->stack[tso->stack_size])); +} + +/* ----------------------------------------------------------------------------- + Blocks of function args occur on the stack (at the top) and + in PAPs. + -------------------------------------------------------------------------- */ + +STATIC_INLINE StgPtr +scavenge_arg_block (StgFunInfoTable *fun_info, StgClosure **args) +{ + StgPtr p; + StgWord bitmap; + nat size; + + p = (StgPtr)args; + switch (fun_info->f.fun_type) { + case ARG_GEN: + bitmap = BITMAP_BITS(fun_info->f.bitmap); + size = BITMAP_SIZE(fun_info->f.bitmap); + goto small_bitmap; + case ARG_GEN_BIG: + size = GET_FUN_LARGE_BITMAP(fun_info)->size; + scavenge_large_bitmap(p, GET_FUN_LARGE_BITMAP(fun_info), size); + p += size; + break; + default: + bitmap = BITMAP_BITS(stg_arg_bitmaps[fun_info->f.fun_type]); + size = BITMAP_SIZE(stg_arg_bitmaps[fun_info->f.fun_type]); + small_bitmap: + while (size > 0) { + if ((bitmap & 1) == 0) { + *p = (StgWord)(StgPtr)evacuate((StgClosure *)*p); + } + p++; + bitmap = bitmap >> 1; + size--; + } + break; + } + return p; +} + +STATIC_INLINE StgPtr +scavenge_PAP (StgPAP *pap) +{ + StgPtr p; + StgWord bitmap, size; + StgFunInfoTable *fun_info; + + pap->fun = evacuate(pap->fun); + fun_info = get_fun_itbl(pap->fun); + ASSERT(fun_info->i.type != PAP); + + p = (StgPtr)pap->payload; + size = pap->n_args; + + switch (fun_info->f.fun_type) { + case ARG_GEN: + bitmap = BITMAP_BITS(fun_info->f.bitmap); + goto small_bitmap; + case ARG_GEN_BIG: + scavenge_large_bitmap(p, GET_FUN_LARGE_BITMAP(fun_info), size); + p += size; + break; + case ARG_BCO: + scavenge_large_bitmap((StgPtr)pap->payload, BCO_BITMAP(pap->fun), size); + p += size; + break; + default: + bitmap = BITMAP_BITS(stg_arg_bitmaps[fun_info->f.fun_type]); + small_bitmap: + size = pap->n_args; + while (size > 0) { + if ((bitmap & 1) == 0) { + *p = (StgWord)(StgPtr)evacuate((StgClosure *)*p); + } + p++; + bitmap = bitmap >> 1; + size--; + } + break; + } + return p; } /* ----------------------------------------------------------------------------- @@ -1817,18 +2431,17 @@ scavengeTSO (StgTSO *tso) scavenging a mutable object where early promotion isn't such a good idea. -------------------------------------------------------------------------- */ -//@cindex scavenge static void -scavenge(step *step) +scavenge(step *stp) { StgPtr p, q; - const StgInfoTable *info; + StgInfoTable *info; bdescr *bd; - nat saved_evac_gen = evac_gen; /* used for temporarily changing evac_gen */ + nat saved_evac_gen = evac_gen; - p = step->scan; - bd = step->scan_bd; + p = stp->scan; + bd = stp->scan_bd; failed_to_evac = rtsFalse; @@ -1836,177 +2449,162 @@ scavenge(step *step) * evacuated objects */ - while (bd != step->hp_bd || p < step->hp) { + while (bd != stp->hp_bd || p < stp->hp) { - /* If we're at the end of this block, move on to the next block */ - if (bd != step->hp_bd && p == bd->free) { + // If we're at the end of this block, move on to the next block + if (bd != stp->hp_bd && p == bd->free) { bd = bd->link; p = bd->start; continue; } - q = p; /* save ptr to object */ - - ASSERT(p && (LOOKS_LIKE_GHC_INFO(GET_INFO((StgClosure *)p)) - || IS_HUGS_CONSTR_INFO(GET_INFO((StgClosure *)p)))); - + ASSERT(LOOKS_LIKE_CLOSURE_PTR(p)); info = get_itbl((StgClosure *)p); - /* - if (info->type==RBH) - info = REVERT_INFOPTR(info); - */ - - switch (info -> type) { + + ASSERT(thunk_selector_depth == 0); - case BCO: - { - StgBCO* bco = (StgBCO *)p; - nat i; - for (i = 0; i < bco->n_ptrs; i++) { - bcoConstCPtr(bco,i) = evacuate(bcoConstCPtr(bco,i)); - } - p += bco_sizeW(bco); - break; - } + q = p; + switch (info->type) { case MVAR: - /* treat MVars specially, because we don't want to evacuate the - * mut_link field in the middle of the closure. - */ - { + { StgMVar *mvar = ((StgMVar *)p); evac_gen = 0; - (StgClosure *)mvar->head = evacuate((StgClosure *)mvar->head); - (StgClosure *)mvar->tail = evacuate((StgClosure *)mvar->tail); - (StgClosure *)mvar->value = evacuate((StgClosure *)mvar->value); - p += sizeofW(StgMVar); + mvar->head = (StgTSO *)evacuate((StgClosure *)mvar->head); + mvar->tail = (StgTSO *)evacuate((StgClosure *)mvar->tail); + mvar->value = evacuate((StgClosure *)mvar->value); evac_gen = saved_evac_gen; + failed_to_evac = rtsTrue; // mutable. + p += sizeofW(StgMVar); break; - } + } - case THUNK_2_0: case FUN_2_0: - scavenge_srt(info); - case CONSTR_2_0: - ((StgClosure *)p)->payload[1] = evacuate(((StgClosure *)p)->payload[1]); - ((StgClosure *)p)->payload[0] = evacuate(((StgClosure *)p)->payload[0]); - p += sizeofW(StgHeader) + 2; - break; + scavenge_fun_srt(info); + ((StgClosure *)p)->payload[1] = evacuate(((StgClosure *)p)->payload[1]); + ((StgClosure *)p)->payload[0] = evacuate(((StgClosure *)p)->payload[0]); + p += sizeofW(StgHeader) + 2; + break; + case THUNK_2_0: + scavenge_thunk_srt(info); + case CONSTR_2_0: + ((StgClosure *)p)->payload[1] = evacuate(((StgClosure *)p)->payload[1]); + ((StgClosure *)p)->payload[0] = evacuate(((StgClosure *)p)->payload[0]); + p += sizeofW(StgHeader) + 2; + break; + case THUNK_1_0: - scavenge_srt(info); - ((StgClosure *)p)->payload[0] = evacuate(((StgClosure *)p)->payload[0]); - p += sizeofW(StgHeader) + 2; /* MIN_UPD_SIZE */ - break; - + scavenge_thunk_srt(info); + ((StgClosure *)p)->payload[0] = evacuate(((StgClosure *)p)->payload[0]); + p += sizeofW(StgHeader) + 1; + break; + case FUN_1_0: - scavenge_srt(info); + scavenge_fun_srt(info); case CONSTR_1_0: - ((StgClosure *)p)->payload[0] = evacuate(((StgClosure *)p)->payload[0]); - p += sizeofW(StgHeader) + 1; - break; - + ((StgClosure *)p)->payload[0] = evacuate(((StgClosure *)p)->payload[0]); + p += sizeofW(StgHeader) + 1; + break; + case THUNK_0_1: - scavenge_srt(info); - p += sizeofW(StgHeader) + 2; /* MIN_UPD_SIZE */ - break; - + scavenge_thunk_srt(info); + p += sizeofW(StgHeader) + 1; + break; + case FUN_0_1: - scavenge_srt(info); + scavenge_fun_srt(info); case CONSTR_0_1: - p += sizeofW(StgHeader) + 1; - break; - + p += sizeofW(StgHeader) + 1; + break; + case THUNK_0_2: + scavenge_thunk_srt(info); + p += sizeofW(StgHeader) + 2; + break; + case FUN_0_2: - scavenge_srt(info); + scavenge_fun_srt(info); case CONSTR_0_2: - p += sizeofW(StgHeader) + 2; - break; - + p += sizeofW(StgHeader) + 2; + break; + case THUNK_1_1: + scavenge_thunk_srt(info); + ((StgClosure *)p)->payload[0] = evacuate(((StgClosure *)p)->payload[0]); + p += sizeofW(StgHeader) + 2; + break; + case FUN_1_1: - scavenge_srt(info); + scavenge_fun_srt(info); case CONSTR_1_1: - ((StgClosure *)p)->payload[0] = evacuate(((StgClosure *)p)->payload[0]); - p += sizeofW(StgHeader) + 2; - break; - + ((StgClosure *)p)->payload[0] = evacuate(((StgClosure *)p)->payload[0]); + p += sizeofW(StgHeader) + 2; + break; + case FUN: - case THUNK: - scavenge_srt(info); - /* fall through */ + scavenge_fun_srt(info); + goto gen_obj; + case THUNK: + scavenge_thunk_srt(info); + // fall through + + gen_obj: case CONSTR: case WEAK: case FOREIGN: case STABLE_NAME: - { + { StgPtr end; end = (P_)((StgClosure *)p)->payload + info->layout.payload.ptrs; for (p = (P_)((StgClosure *)p)->payload; p < end; p++) { - (StgClosure *)*p = evacuate((StgClosure *)*p); + *p = (StgWord)(StgPtr)evacuate((StgClosure *)*p); } p += info->layout.payload.nptrs; break; - } + } + + case BCO: { + StgBCO *bco = (StgBCO *)p; + bco->instrs = (StgArrWords *)evacuate((StgClosure *)bco->instrs); + bco->literals = (StgArrWords *)evacuate((StgClosure *)bco->literals); + bco->ptrs = (StgMutArrPtrs *)evacuate((StgClosure *)bco->ptrs); + bco->itbls = (StgArrWords *)evacuate((StgClosure *)bco->itbls); + p += bco_sizeW(bco); + break; + } case IND_PERM: - if (step->gen->no != 0) { - SET_INFO(((StgClosure *)p), &IND_OLDGEN_PERM_info); + if (stp->gen->no != 0) { +#ifdef PROFILING + // @LDV profiling + // No need to call LDV_recordDead_FILL_SLOP_DYNAMIC() because an + // IND_OLDGEN_PERM closure is larger than an IND_PERM closure. + LDV_recordDead((StgClosure *)p, sizeofW(StgInd)); +#endif + // + // Todo: maybe use SET_HDR() and remove LDV_RECORD_CREATE()? + // + SET_INFO(((StgClosure *)p), &stg_IND_OLDGEN_PERM_info); + + // We pretend that p has just been created. + LDV_RECORD_CREATE((StgClosure *)p); } - /* fall through */ + // fall through case IND_OLDGEN_PERM: - ((StgIndOldGen *)p)->indirectee = - evacuate(((StgIndOldGen *)p)->indirectee); - if (failed_to_evac) { - failed_to_evac = rtsFalse; - recordOldToNewPtrs((StgMutClosure *)p); - } - p += sizeofW(StgIndOldGen); - break; - - case CAF_UNENTERED: - { - StgCAF *caf = (StgCAF *)p; - - caf->body = evacuate(caf->body); - if (failed_to_evac) { - failed_to_evac = rtsFalse; - recordOldToNewPtrs((StgMutClosure *)p); - } else { - caf->mut_link = NULL; - } - p += sizeofW(StgCAF); - break; - } - - case CAF_ENTERED: - { - StgCAF *caf = (StgCAF *)p; - - caf->body = evacuate(caf->body); - caf->value = evacuate(caf->value); - if (failed_to_evac) { - failed_to_evac = rtsFalse; - recordOldToNewPtrs((StgMutClosure *)p); - } else { - caf->mut_link = NULL; - } - p += sizeofW(StgCAF); + ((StgInd *)p)->indirectee = evacuate(((StgInd *)p)->indirectee); + p += sizeofW(StgInd); break; - } case MUT_VAR: - /* ignore MUT_CONSs */ - if (((StgMutVar *)p)->header.info != &MUT_CONS_info) { evac_gen = 0; ((StgMutVar *)p)->var = evacuate(((StgMutVar *)p)->var); evac_gen = saved_evac_gen; - } - p += sizeofW(StgMutVar); - break; + failed_to_evac = rtsTrue; // mutable anyhow + p += sizeofW(StgMutVar); + break; case CAF_BLACKHOLE: case SE_CAF_BLACKHOLE: @@ -2016,351 +2614,850 @@ scavenge(step *step) break; case BLACKHOLE_BQ: - { + { StgBlockingQueue *bh = (StgBlockingQueue *)p; - (StgClosure *)bh->blocking_queue = - evacuate((StgClosure *)bh->blocking_queue); - if (failed_to_evac) { - failed_to_evac = rtsFalse; - recordMutable((StgMutClosure *)bh); - } + bh->blocking_queue = + (StgTSO *)evacuate((StgClosure *)bh->blocking_queue); + failed_to_evac = rtsTrue; p += BLACKHOLE_sizeW(); break; - } + } case THUNK_SELECTOR: - { + { StgSelector *s = (StgSelector *)p; s->selectee = evacuate(s->selectee); p += THUNK_SELECTOR_sizeW(); break; - } - - case IND: - case IND_OLDGEN: - barf("scavenge:IND???\n"); + } - case CONSTR_INTLIKE: - case CONSTR_CHARLIKE: - case CONSTR_STATIC: - case CONSTR_NOCAF_STATIC: - case THUNK_STATIC: - case FUN_STATIC: - case IND_STATIC: - /* Shouldn't see a static object here. */ - barf("scavenge: STATIC object\n"); + // A chunk of stack saved in a heap object + case AP_STACK: + { + StgAP_STACK *ap = (StgAP_STACK *)p; - case RET_BCO: - case RET_SMALL: - case RET_VEC_SMALL: - case RET_BIG: - case RET_VEC_BIG: - case RET_DYN: - case UPDATE_FRAME: - case STOP_FRAME: - case CATCH_FRAME: - case SEQ_FRAME: - /* Shouldn't see stack frames here. */ - barf("scavenge: stack frame\n"); + ap->fun = evacuate(ap->fun); + scavenge_stack((StgPtr)ap->payload, (StgPtr)ap->payload + ap->size); + p = (StgPtr)ap->payload + ap->size; + break; + } - case AP_UPD: /* same as PAPs */ case PAP: - /* Treat a PAP just like a section of stack, not forgetting to - * evacuate the function pointer too... - */ - { - StgPAP* pap = (StgPAP *)p; - - pap->fun = evacuate(pap->fun); - scavenge_stack((P_)pap->payload, (P_)pap->payload + pap->n_args); - p += pap_sizeW(pap); + case AP: + p = scavenge_PAP((StgPAP *)p); break; - } - + case ARR_WORDS: - /* nothing to follow */ - p += arr_words_sizeW((StgArrWords *)p); - break; + // nothing to follow + p += arr_words_sizeW((StgArrWords *)p); + break; case MUT_ARR_PTRS: - /* follow everything */ - { + // follow everything + { StgPtr next; - evac_gen = 0; /* repeatedly mutable */ + evac_gen = 0; // repeatedly mutable next = p + mut_arr_ptrs_sizeW((StgMutArrPtrs*)p); for (p = (P_)((StgMutArrPtrs *)p)->payload; p < next; p++) { - (StgClosure *)*p = evacuate((StgClosure *)*p); + *p = (StgWord)(StgPtr)evacuate((StgClosure *)*p); } evac_gen = saved_evac_gen; + failed_to_evac = rtsTrue; // mutable anyhow. break; - } + } case MUT_ARR_PTRS_FROZEN: - /* follow everything */ - { - StgPtr start = p, next; + case MUT_ARR_PTRS_FROZEN0: + // follow everything + { + StgPtr next; next = p + mut_arr_ptrs_sizeW((StgMutArrPtrs*)p); for (p = (P_)((StgMutArrPtrs *)p)->payload; p < next; p++) { - (StgClosure *)*p = evacuate((StgClosure *)*p); - } - if (failed_to_evac) { - /* we can do this easier... */ - recordMutable((StgMutClosure *)start); - failed_to_evac = rtsFalse; + *p = (StgWord)(StgPtr)evacuate((StgClosure *)*p); } + // it's tempting to recordMutable() if failed_to_evac is + // false, but that breaks some assumptions (eg. every + // closure on the mutable list is supposed to have the MUT + // flag set, and MUT_ARR_PTRS_FROZEN doesn't). break; - } + } case TSO: - { + { StgTSO *tso = (StgTSO *)p; evac_gen = 0; scavengeTSO(tso); evac_gen = saved_evac_gen; + failed_to_evac = rtsTrue; // mutable anyhow. p += tso_sizeW(tso); break; - } + } #if defined(PAR) case RBH: // cf. BLACKHOLE_BQ - { - // nat size, ptrs, nonptrs, vhs; - // char str[80]; - // StgInfoTable *rip = get_closure_info(p, &size, &ptrs, &nonptrs, &vhs, str); + { +#if 0 + nat size, ptrs, nonptrs, vhs; + char str[80]; + StgInfoTable *rip = get_closure_info(p, &size, &ptrs, &nonptrs, &vhs, str); +#endif StgRBH *rbh = (StgRBH *)p; (StgClosure *)rbh->blocking_queue = - evacuate((StgClosure *)rbh->blocking_queue); - if (failed_to_evac) { - failed_to_evac = rtsFalse; - recordMutable((StgMutClosure *)rbh); - } + evacuate((StgClosure *)rbh->blocking_queue); + failed_to_evac = rtsTrue; // mutable anyhow. IF_DEBUG(gc, - belch("@@ scavenge: RBH %p (%s) (new blocking_queue link=%p)", + debugBelch("@@ scavenge: RBH %p (%s) (new blocking_queue link=%p)", p, info_type(p), (StgClosure *)rbh->blocking_queue)); // ToDo: use size of reverted closure here! p += BLACKHOLE_sizeW(); break; - } + } case BLOCKED_FETCH: - { + { StgBlockedFetch *bf = (StgBlockedFetch *)p; - /* follow the pointer to the node which is being demanded */ + // follow the pointer to the node which is being demanded (StgClosure *)bf->node = - evacuate((StgClosure *)bf->node); - /* follow the link to the rest of the blocking queue */ + evacuate((StgClosure *)bf->node); + // follow the link to the rest of the blocking queue (StgClosure *)bf->link = - evacuate((StgClosure *)bf->link); - if (failed_to_evac) { - failed_to_evac = rtsFalse; - recordMutable((StgMutClosure *)bf); - } + evacuate((StgClosure *)bf->link); IF_DEBUG(gc, - belch("@@ scavenge: %p (%s); node is now %p; exciting, isn't it", - bf, info_type((StgClosure *)bf), - bf->node, info_type(bf->node))); + debugBelch("@@ scavenge: %p (%s); node is now %p; exciting, isn't it", + bf, info_type((StgClosure *)bf), + bf->node, info_type(bf->node))); p += sizeofW(StgBlockedFetch); break; - } + } +#ifdef DIST + case REMOTE_REF: +#endif case FETCH_ME: - IF_DEBUG(gc, - belch("@@ scavenge: HWL claims nothing to do for %p (%s)", - p, info_type((StgClosure *)p))); - p += sizeofW(StgFetchMe); - break; // nothing to do in this case + p += sizeofW(StgFetchMe); + break; // nothing to do in this case case FETCH_ME_BQ: // cf. BLACKHOLE_BQ - { + { StgFetchMeBlockingQueue *fmbq = (StgFetchMeBlockingQueue *)p; (StgClosure *)fmbq->blocking_queue = - evacuate((StgClosure *)fmbq->blocking_queue); - if (failed_to_evac) { - failed_to_evac = rtsFalse; - recordMutable((StgMutClosure *)fmbq); - } + evacuate((StgClosure *)fmbq->blocking_queue); IF_DEBUG(gc, - belch("@@ scavenge: %p (%s) exciting, isn't it", - p, info_type((StgClosure *)p))); + debugBelch("@@ scavenge: %p (%s) exciting, isn't it", + p, info_type((StgClosure *)p))); p += sizeofW(StgFetchMeBlockingQueue); break; - } + } #endif - case EVACUATED: - barf("scavenge: unimplemented/strange closure type %d @ %p", - info->type, p); + case TVAR_WAIT_QUEUE: + { + StgTVarWaitQueue *wq = ((StgTVarWaitQueue *) p); + evac_gen = 0; + wq->waiting_tso = (StgTSO *)evacuate((StgClosure*)wq->waiting_tso); + wq->next_queue_entry = (StgTVarWaitQueue *)evacuate((StgClosure*)wq->next_queue_entry); + wq->prev_queue_entry = (StgTVarWaitQueue *)evacuate((StgClosure*)wq->prev_queue_entry); + evac_gen = saved_evac_gen; + failed_to_evac = rtsTrue; // mutable + p += sizeofW(StgTVarWaitQueue); + break; + } + + case TVAR: + { + StgTVar *tvar = ((StgTVar *) p); + evac_gen = 0; + tvar->current_value = evacuate((StgClosure*)tvar->current_value); + tvar->first_wait_queue_entry = (StgTVarWaitQueue *)evacuate((StgClosure*)tvar->first_wait_queue_entry); + evac_gen = saved_evac_gen; + failed_to_evac = rtsTrue; // mutable + p += sizeofW(StgTVar); + break; + } + + case TREC_HEADER: + { + StgTRecHeader *trec = ((StgTRecHeader *) p); + evac_gen = 0; + trec->enclosing_trec = (StgTRecHeader *)evacuate((StgClosure*)trec->enclosing_trec); + trec->current_chunk = (StgTRecChunk *)evacuate((StgClosure*)trec->current_chunk); + evac_gen = saved_evac_gen; + failed_to_evac = rtsTrue; // mutable + p += sizeofW(StgTRecHeader); + break; + } + + case TREC_CHUNK: + { + StgWord i; + StgTRecChunk *tc = ((StgTRecChunk *) p); + TRecEntry *e = &(tc -> entries[0]); + evac_gen = 0; + tc->prev_chunk = (StgTRecChunk *)evacuate((StgClosure*)tc->prev_chunk); + for (i = 0; i < tc -> next_entry_idx; i ++, e++ ) { + e->tvar = (StgTVar *)evacuate((StgClosure*)e->tvar); + e->expected_value = evacuate((StgClosure*)e->expected_value); + e->new_value = evacuate((StgClosure*)e->new_value); + } + evac_gen = saved_evac_gen; + failed_to_evac = rtsTrue; // mutable + p += sizeofW(StgTRecChunk); + break; + } default: - barf("scavenge: unimplemented/strange closure type %d @ %p", - info->type, p); + barf("scavenge: unimplemented/strange closure type %d @ %p", + info->type, p); } - /* If we didn't manage to promote all the objects pointed to by - * the current object, then we have to designate this object as - * mutable (because it contains old-to-new generation pointers). + /* + * We need to record the current object on the mutable list if + * (a) It is actually mutable, or + * (b) It contains pointers to a younger generation. + * Case (b) arises if we didn't manage to promote everything that + * the current object points to into the current generation. */ if (failed_to_evac) { - mkMutCons((StgClosure *)q, &generations[evac_gen]); - failed_to_evac = rtsFalse; + failed_to_evac = rtsFalse; + recordMutableGen((StgClosure *)q, stp->gen); } } - step->scan_bd = bd; - step->scan = p; + stp->scan_bd = bd; + stp->scan = p; } /* ----------------------------------------------------------------------------- + Scavenge everything on the mark stack. + + This is slightly different from scavenge(): + - we don't walk linearly through the objects, so the scavenger + doesn't need to advance the pointer on to the next object. + -------------------------------------------------------------------------- */ + +static void +scavenge_mark_stack(void) +{ + StgPtr p, q; + StgInfoTable *info; + nat saved_evac_gen; + + evac_gen = oldest_gen->no; + saved_evac_gen = evac_gen; + +linear_scan: + while (!mark_stack_empty()) { + p = pop_mark_stack(); + + ASSERT(LOOKS_LIKE_CLOSURE_PTR(p)); + info = get_itbl((StgClosure *)p); + + q = p; + switch (info->type) { + + case MVAR: + { + StgMVar *mvar = ((StgMVar *)p); + evac_gen = 0; + mvar->head = (StgTSO *)evacuate((StgClosure *)mvar->head); + mvar->tail = (StgTSO *)evacuate((StgClosure *)mvar->tail); + mvar->value = evacuate((StgClosure *)mvar->value); + evac_gen = saved_evac_gen; + failed_to_evac = rtsTrue; // mutable. + break; + } + + case FUN_2_0: + scavenge_fun_srt(info); + ((StgClosure *)p)->payload[1] = evacuate(((StgClosure *)p)->payload[1]); + ((StgClosure *)p)->payload[0] = evacuate(((StgClosure *)p)->payload[0]); + break; + + case THUNK_2_0: + scavenge_thunk_srt(info); + case CONSTR_2_0: + ((StgClosure *)p)->payload[1] = evacuate(((StgClosure *)p)->payload[1]); + ((StgClosure *)p)->payload[0] = evacuate(((StgClosure *)p)->payload[0]); + break; + + case FUN_1_0: + case FUN_1_1: + scavenge_fun_srt(info); + ((StgClosure *)p)->payload[0] = evacuate(((StgClosure *)p)->payload[0]); + break; + + case THUNK_1_0: + case THUNK_1_1: + scavenge_thunk_srt(info); + case CONSTR_1_0: + case CONSTR_1_1: + ((StgClosure *)p)->payload[0] = evacuate(((StgClosure *)p)->payload[0]); + break; + + case FUN_0_1: + case FUN_0_2: + scavenge_fun_srt(info); + break; + + case THUNK_0_1: + case THUNK_0_2: + scavenge_thunk_srt(info); + break; + + case CONSTR_0_1: + case CONSTR_0_2: + break; + + case FUN: + scavenge_fun_srt(info); + goto gen_obj; + + case THUNK: + scavenge_thunk_srt(info); + // fall through + + gen_obj: + case CONSTR: + case WEAK: + case FOREIGN: + case STABLE_NAME: + { + StgPtr end; + + end = (P_)((StgClosure *)p)->payload + info->layout.payload.ptrs; + for (p = (P_)((StgClosure *)p)->payload; p < end; p++) { + *p = (StgWord)(StgPtr)evacuate((StgClosure *)*p); + } + break; + } + + case BCO: { + StgBCO *bco = (StgBCO *)p; + bco->instrs = (StgArrWords *)evacuate((StgClosure *)bco->instrs); + bco->literals = (StgArrWords *)evacuate((StgClosure *)bco->literals); + bco->ptrs = (StgMutArrPtrs *)evacuate((StgClosure *)bco->ptrs); + bco->itbls = (StgArrWords *)evacuate((StgClosure *)bco->itbls); + break; + } + + case IND_PERM: + // don't need to do anything here: the only possible case + // is that we're in a 1-space compacting collector, with + // no "old" generation. + break; + + case IND_OLDGEN: + case IND_OLDGEN_PERM: + ((StgInd *)p)->indirectee = + evacuate(((StgInd *)p)->indirectee); + break; + + case MUT_VAR: + evac_gen = 0; + ((StgMutVar *)p)->var = evacuate(((StgMutVar *)p)->var); + evac_gen = saved_evac_gen; + failed_to_evac = rtsTrue; + break; + + case CAF_BLACKHOLE: + case SE_CAF_BLACKHOLE: + case SE_BLACKHOLE: + case BLACKHOLE: + case ARR_WORDS: + break; + + case BLACKHOLE_BQ: + { + StgBlockingQueue *bh = (StgBlockingQueue *)p; + bh->blocking_queue = + (StgTSO *)evacuate((StgClosure *)bh->blocking_queue); + failed_to_evac = rtsTrue; + break; + } + + case THUNK_SELECTOR: + { + StgSelector *s = (StgSelector *)p; + s->selectee = evacuate(s->selectee); + break; + } + + // A chunk of stack saved in a heap object + case AP_STACK: + { + StgAP_STACK *ap = (StgAP_STACK *)p; + + ap->fun = evacuate(ap->fun); + scavenge_stack((StgPtr)ap->payload, (StgPtr)ap->payload + ap->size); + break; + } + + case PAP: + case AP: + scavenge_PAP((StgPAP *)p); + break; + + case MUT_ARR_PTRS: + // follow everything + { + StgPtr next; + + evac_gen = 0; // repeatedly mutable + next = p + mut_arr_ptrs_sizeW((StgMutArrPtrs*)p); + for (p = (P_)((StgMutArrPtrs *)p)->payload; p < next; p++) { + *p = (StgWord)(StgPtr)evacuate((StgClosure *)*p); + } + evac_gen = saved_evac_gen; + failed_to_evac = rtsTrue; // mutable anyhow. + break; + } + + case MUT_ARR_PTRS_FROZEN: + case MUT_ARR_PTRS_FROZEN0: + // follow everything + { + StgPtr next; + + next = p + mut_arr_ptrs_sizeW((StgMutArrPtrs*)p); + for (p = (P_)((StgMutArrPtrs *)p)->payload; p < next; p++) { + *p = (StgWord)(StgPtr)evacuate((StgClosure *)*p); + } + break; + } + + case TSO: + { + StgTSO *tso = (StgTSO *)p; + evac_gen = 0; + scavengeTSO(tso); + evac_gen = saved_evac_gen; + failed_to_evac = rtsTrue; + break; + } + +#if defined(PAR) + case RBH: // cf. BLACKHOLE_BQ + { +#if 0 + nat size, ptrs, nonptrs, vhs; + char str[80]; + StgInfoTable *rip = get_closure_info(p, &size, &ptrs, &nonptrs, &vhs, str); +#endif + StgRBH *rbh = (StgRBH *)p; + bh->blocking_queue = + (StgTSO *)evacuate((StgClosure *)bh->blocking_queue); + failed_to_evac = rtsTrue; // mutable anyhow. + IF_DEBUG(gc, + debugBelch("@@ scavenge: RBH %p (%s) (new blocking_queue link=%p)", + p, info_type(p), (StgClosure *)rbh->blocking_queue)); + break; + } + + case BLOCKED_FETCH: + { + StgBlockedFetch *bf = (StgBlockedFetch *)p; + // follow the pointer to the node which is being demanded + (StgClosure *)bf->node = + evacuate((StgClosure *)bf->node); + // follow the link to the rest of the blocking queue + (StgClosure *)bf->link = + evacuate((StgClosure *)bf->link); + IF_DEBUG(gc, + debugBelch("@@ scavenge: %p (%s); node is now %p; exciting, isn't it", + bf, info_type((StgClosure *)bf), + bf->node, info_type(bf->node))); + break; + } + +#ifdef DIST + case REMOTE_REF: +#endif + case FETCH_ME: + break; // nothing to do in this case + + case FETCH_ME_BQ: // cf. BLACKHOLE_BQ + { + StgFetchMeBlockingQueue *fmbq = (StgFetchMeBlockingQueue *)p; + (StgClosure *)fmbq->blocking_queue = + evacuate((StgClosure *)fmbq->blocking_queue); + IF_DEBUG(gc, + debugBelch("@@ scavenge: %p (%s) exciting, isn't it", + p, info_type((StgClosure *)p))); + break; + } +#endif // PAR + + case TVAR_WAIT_QUEUE: + { + StgTVarWaitQueue *wq = ((StgTVarWaitQueue *) p); + evac_gen = 0; + wq->waiting_tso = (StgTSO *)evacuate((StgClosure*)wq->waiting_tso); + wq->next_queue_entry = (StgTVarWaitQueue *)evacuate((StgClosure*)wq->next_queue_entry); + wq->prev_queue_entry = (StgTVarWaitQueue *)evacuate((StgClosure*)wq->prev_queue_entry); + evac_gen = saved_evac_gen; + failed_to_evac = rtsTrue; // mutable + break; + } + + case TVAR: + { + StgTVar *tvar = ((StgTVar *) p); + evac_gen = 0; + tvar->current_value = evacuate((StgClosure*)tvar->current_value); + tvar->first_wait_queue_entry = (StgTVarWaitQueue *)evacuate((StgClosure*)tvar->first_wait_queue_entry); + evac_gen = saved_evac_gen; + failed_to_evac = rtsTrue; // mutable + break; + } + + case TREC_CHUNK: + { + StgWord i; + StgTRecChunk *tc = ((StgTRecChunk *) p); + TRecEntry *e = &(tc -> entries[0]); + evac_gen = 0; + tc->prev_chunk = (StgTRecChunk *)evacuate((StgClosure*)tc->prev_chunk); + for (i = 0; i < tc -> next_entry_idx; i ++, e++ ) { + e->tvar = (StgTVar *)evacuate((StgClosure*)e->tvar); + e->expected_value = evacuate((StgClosure*)e->expected_value); + e->new_value = evacuate((StgClosure*)e->new_value); + } + evac_gen = saved_evac_gen; + failed_to_evac = rtsTrue; // mutable + break; + } + + case TREC_HEADER: + { + StgTRecHeader *trec = ((StgTRecHeader *) p); + evac_gen = 0; + trec->enclosing_trec = (StgTRecHeader *)evacuate((StgClosure*)trec->enclosing_trec); + trec->current_chunk = (StgTRecChunk *)evacuate((StgClosure*)trec->current_chunk); + evac_gen = saved_evac_gen; + failed_to_evac = rtsTrue; // mutable + break; + } + + default: + barf("scavenge_mark_stack: unimplemented/strange closure type %d @ %p", + info->type, p); + } + + if (failed_to_evac) { + failed_to_evac = rtsFalse; + recordMutableGen((StgClosure *)q, &generations[evac_gen]); + } + + // mark the next bit to indicate "scavenged" + mark(q+1, Bdescr(q)); + + } // while (!mark_stack_empty()) + + // start a new linear scan if the mark stack overflowed at some point + if (mark_stack_overflowed && oldgen_scan_bd == NULL) { + IF_DEBUG(gc, debugBelch("scavenge_mark_stack: starting linear scan")); + mark_stack_overflowed = rtsFalse; + oldgen_scan_bd = oldest_gen->steps[0].blocks; + oldgen_scan = oldgen_scan_bd->start; + } + + if (oldgen_scan_bd) { + // push a new thing on the mark stack + loop: + // find a closure that is marked but not scavenged, and start + // from there. + while (oldgen_scan < oldgen_scan_bd->free + && !is_marked(oldgen_scan,oldgen_scan_bd)) { + oldgen_scan++; + } + + if (oldgen_scan < oldgen_scan_bd->free) { + + // already scavenged? + if (is_marked(oldgen_scan+1,oldgen_scan_bd)) { + oldgen_scan += sizeofW(StgHeader) + MIN_NONUPD_SIZE; + goto loop; + } + push_mark_stack(oldgen_scan); + // ToDo: bump the linear scan by the actual size of the object + oldgen_scan += sizeofW(StgHeader) + MIN_NONUPD_SIZE; + goto linear_scan; + } + + oldgen_scan_bd = oldgen_scan_bd->link; + if (oldgen_scan_bd != NULL) { + oldgen_scan = oldgen_scan_bd->start; + goto loop; + } + } +} + +/* ----------------------------------------------------------------------------- Scavenge one object. This is used for objects that are temporarily marked as mutable because they contain old-to-new generation pointers. Only certain objects can have this property. -------------------------------------------------------------------------- */ -//@cindex scavenge_one static rtsBool -scavenge_one(StgClosure *p) +scavenge_one(StgPtr p) { - const StgInfoTable *info; - rtsBool no_luck; + const StgInfoTable *info; + nat saved_evac_gen = evac_gen; + rtsBool no_luck; + + ASSERT(LOOKS_LIKE_CLOSURE_PTR(p)); + info = get_itbl((StgClosure *)p); + + switch (info->type) { + + case MVAR: + { + StgMVar *mvar = ((StgMVar *)p); + evac_gen = 0; + mvar->head = (StgTSO *)evacuate((StgClosure *)mvar->head); + mvar->tail = (StgTSO *)evacuate((StgClosure *)mvar->tail); + mvar->value = evacuate((StgClosure *)mvar->value); + evac_gen = saved_evac_gen; + failed_to_evac = rtsTrue; // mutable. + break; + } - ASSERT(p && (LOOKS_LIKE_GHC_INFO(GET_INFO(p)) - || IS_HUGS_CONSTR_INFO(GET_INFO(p)))); + case FUN: + case FUN_1_0: // hardly worth specialising these guys + case FUN_0_1: + case FUN_1_1: + case FUN_0_2: + case FUN_2_0: + case THUNK: + case THUNK_1_0: + case THUNK_0_1: + case THUNK_1_1: + case THUNK_0_2: + case THUNK_2_0: + case CONSTR: + case CONSTR_1_0: + case CONSTR_0_1: + case CONSTR_1_1: + case CONSTR_0_2: + case CONSTR_2_0: + case WEAK: + case FOREIGN: + case IND_PERM: + { + StgPtr q, end; + + end = (StgPtr)((StgClosure *)p)->payload + info->layout.payload.ptrs; + for (q = (StgPtr)((StgClosure *)p)->payload; q < end; q++) { + *q = (StgWord)(StgPtr)evacuate((StgClosure *)*q); + } + break; + } + + case MUT_VAR: + evac_gen = 0; + ((StgMutVar *)p)->var = evacuate(((StgMutVar *)p)->var); + evac_gen = saved_evac_gen; + failed_to_evac = rtsTrue; // mutable anyhow + break; - info = get_itbl(p); + case CAF_BLACKHOLE: + case SE_CAF_BLACKHOLE: + case SE_BLACKHOLE: + case BLACKHOLE: + break; + + case BLACKHOLE_BQ: + { + StgBlockingQueue *bh = (StgBlockingQueue *)p; + evac_gen = 0; // repeatedly mutable + bh->blocking_queue = + (StgTSO *)evacuate((StgClosure *)bh->blocking_queue); + failed_to_evac = rtsTrue; + break; + } - /* ngoq moHqu'! - if (info->type==RBH) - info = REVERT_INFOPTR(info); // if it's an RBH, look at the orig closure - */ + case THUNK_SELECTOR: + { + StgSelector *s = (StgSelector *)p; + s->selectee = evacuate(s->selectee); + break; + } + + case AP_STACK: + { + StgAP_STACK *ap = (StgAP_STACK *)p; - switch (info -> type) { + ap->fun = evacuate(ap->fun); + scavenge_stack((StgPtr)ap->payload, (StgPtr)ap->payload + ap->size); + p = (StgPtr)ap->payload + ap->size; + break; + } - case FUN: - case FUN_1_0: /* hardly worth specialising these guys */ - case FUN_0_1: - case FUN_1_1: - case FUN_0_2: - case FUN_2_0: - case THUNK: - case THUNK_1_0: - case THUNK_0_1: - case THUNK_1_1: - case THUNK_0_2: - case THUNK_2_0: - case CONSTR: - case CONSTR_1_0: - case CONSTR_0_1: - case CONSTR_1_1: - case CONSTR_0_2: - case CONSTR_2_0: - case WEAK: - case FOREIGN: - case IND_PERM: - case IND_OLDGEN_PERM: - case CAF_UNENTERED: + case PAP: + case AP: + p = scavenge_PAP((StgPAP *)p); + break; + + case ARR_WORDS: + // nothing to follow + break; + + case MUT_ARR_PTRS: { - StgPtr q, end; + // follow everything + StgPtr next; - end = (P_)p->payload + info->layout.payload.ptrs; - for (q = (P_)p->payload; q < end; q++) { - (StgClosure *)*q = evacuate((StgClosure *)*q); - } - break; + evac_gen = 0; // repeatedly mutable + next = p + mut_arr_ptrs_sizeW((StgMutArrPtrs*)p); + for (p = (P_)((StgMutArrPtrs *)p)->payload; p < next; p++) { + *p = (StgWord)(StgPtr)evacuate((StgClosure *)*p); + } + evac_gen = saved_evac_gen; + failed_to_evac = rtsTrue; + break; } - case CAF_BLACKHOLE: - case SE_CAF_BLACKHOLE: - case SE_BLACKHOLE: - case BLACKHOLE: - break; + case MUT_ARR_PTRS_FROZEN: + case MUT_ARR_PTRS_FROZEN0: + { + // follow everything + StgPtr next; + + next = p + mut_arr_ptrs_sizeW((StgMutArrPtrs*)p); + for (p = (P_)((StgMutArrPtrs *)p)->payload; p < next; p++) { + *p = (StgWord)(StgPtr)evacuate((StgClosure *)*p); + } + break; + } - case THUNK_SELECTOR: - { - StgSelector *s = (StgSelector *)p; - s->selectee = evacuate(s->selectee); - break; + case TSO: + { + StgTSO *tso = (StgTSO *)p; + + evac_gen = 0; // repeatedly mutable + scavengeTSO(tso); + evac_gen = saved_evac_gen; + failed_to_evac = rtsTrue; + break; } - - case AP_UPD: /* same as PAPs */ - case PAP: - /* Treat a PAP just like a section of stack, not forgetting to - * evacuate the function pointer too... - */ + +#if defined(PAR) + case RBH: // cf. BLACKHOLE_BQ { - StgPAP* pap = (StgPAP *)p; - - pap->fun = evacuate(pap->fun); - scavenge_stack((P_)pap->payload, (P_)pap->payload + pap->n_args); - break; +#if 0 + nat size, ptrs, nonptrs, vhs; + char str[80]; + StgInfoTable *rip = get_closure_info(p, &size, &ptrs, &nonptrs, &vhs, str); +#endif + StgRBH *rbh = (StgRBH *)p; + (StgClosure *)rbh->blocking_queue = + evacuate((StgClosure *)rbh->blocking_queue); + failed_to_evac = rtsTrue; // mutable anyhow. + IF_DEBUG(gc, + debugBelch("@@ scavenge: RBH %p (%s) (new blocking_queue link=%p)", + p, info_type(p), (StgClosure *)rbh->blocking_queue)); + // ToDo: use size of reverted closure here! + break; } - case IND_OLDGEN: - /* This might happen if for instance a MUT_CONS was pointing to a - * THUNK which has since been updated. The IND_OLDGEN will - * be on the mutable list anyway, so we don't need to do anything - * here. - */ - break; - - default: - barf("scavenge_one: strange object %d", (int)(info->type)); - } - - no_luck = failed_to_evac; - failed_to_evac = rtsFalse; - return (no_luck); -} - + case BLOCKED_FETCH: + { + StgBlockedFetch *bf = (StgBlockedFetch *)p; + // follow the pointer to the node which is being demanded + (StgClosure *)bf->node = + evacuate((StgClosure *)bf->node); + // follow the link to the rest of the blocking queue + (StgClosure *)bf->link = + evacuate((StgClosure *)bf->link); + IF_DEBUG(gc, + debugBelch("@@ scavenge: %p (%s); node is now %p; exciting, isn't it", + bf, info_type((StgClosure *)bf), + bf->node, info_type(bf->node))); + break; + } -/* ----------------------------------------------------------------------------- - Scavenging mutable lists. +#ifdef DIST + case REMOTE_REF: +#endif + case FETCH_ME: + break; // nothing to do in this case - We treat the mutable list of each generation > N (i.e. all the - generations older than the one being collected) as roots. We also - remove non-mutable objects from the mutable list at this point. - -------------------------------------------------------------------------- */ -//@cindex scavenge_mut_once_list + case FETCH_ME_BQ: // cf. BLACKHOLE_BQ + { + StgFetchMeBlockingQueue *fmbq = (StgFetchMeBlockingQueue *)p; + (StgClosure *)fmbq->blocking_queue = + evacuate((StgClosure *)fmbq->blocking_queue); + IF_DEBUG(gc, + debugBelch("@@ scavenge: %p (%s) exciting, isn't it", + p, info_type((StgClosure *)p))); + break; + } +#endif -static void -scavenge_mut_once_list(generation *gen) -{ - const StgInfoTable *info; - StgMutClosure *p, *next, *new_list; + case TVAR_WAIT_QUEUE: + { + StgTVarWaitQueue *wq = ((StgTVarWaitQueue *) p); + evac_gen = 0; + wq->waiting_tso = (StgTSO *)evacuate((StgClosure*)wq->waiting_tso); + wq->next_queue_entry = (StgTVarWaitQueue *)evacuate((StgClosure*)wq->next_queue_entry); + wq->prev_queue_entry = (StgTVarWaitQueue *)evacuate((StgClosure*)wq->prev_queue_entry); + evac_gen = saved_evac_gen; + failed_to_evac = rtsTrue; // mutable + break; + } - p = gen->mut_once_list; - new_list = END_MUT_LIST; - next = p->mut_link; + case TVAR: + { + StgTVar *tvar = ((StgTVar *) p); + evac_gen = 0; + tvar->current_value = evacuate((StgClosure*)tvar->current_value); + tvar->first_wait_queue_entry = (StgTVarWaitQueue *)evacuate((StgClosure*)tvar->first_wait_queue_entry); + evac_gen = saved_evac_gen; + failed_to_evac = rtsTrue; // mutable + break; + } - evac_gen = gen->no; - failed_to_evac = rtsFalse; + case TREC_HEADER: + { + StgTRecHeader *trec = ((StgTRecHeader *) p); + evac_gen = 0; + trec->enclosing_trec = (StgTRecHeader *)evacuate((StgClosure*)trec->enclosing_trec); + trec->current_chunk = (StgTRecChunk *)evacuate((StgClosure*)trec->current_chunk); + evac_gen = saved_evac_gen; + failed_to_evac = rtsTrue; // mutable + break; + } - for (; p != END_MUT_LIST; p = next, next = p->mut_link) { + case TREC_CHUNK: + { + StgWord i; + StgTRecChunk *tc = ((StgTRecChunk *) p); + TRecEntry *e = &(tc -> entries[0]); + evac_gen = 0; + tc->prev_chunk = (StgTRecChunk *)evacuate((StgClosure*)tc->prev_chunk); + for (i = 0; i < tc -> next_entry_idx; i ++, e++ ) { + e->tvar = (StgTVar *)evacuate((StgClosure*)e->tvar); + e->expected_value = evacuate((StgClosure*)e->expected_value); + e->new_value = evacuate((StgClosure*)e->new_value); + } + evac_gen = saved_evac_gen; + failed_to_evac = rtsTrue; // mutable + break; + } - /* make sure the info pointer is into text space */ - ASSERT(p && (LOOKS_LIKE_GHC_INFO(GET_INFO(p)) - || IS_HUGS_CONSTR_INFO(GET_INFO(p)))); - - info = get_itbl(p); - /* - if (info->type==RBH) - info = REVERT_INFOPTR(info); // if it's an RBH, look at the orig closure - */ - switch(info->type) { - case IND_OLDGEN: case IND_OLDGEN_PERM: case IND_STATIC: /* Try to pull the indirectee into this generation, so we can * remove the indirection from the mutable list. */ - ((StgIndOldGen *)p)->indirectee = - evacuate(((StgIndOldGen *)p)->indirectee); + ((StgInd *)p)->indirectee = evacuate(((StgInd *)p)->indirectee); -#ifdef DEBUG +#if 0 && defined(DEBUG) if (RtsFlags.DebugFlags.gc) /* Debugging code to print out the size of the thing we just * promoted @@ -2382,278 +3479,55 @@ scavenge_mut_once_list(generation *gen) } else { size = gen->steps[0].scan - start; } - fprintf(stderr,"evac IND_OLDGEN: %d bytes\n", size * sizeof(W_)); + debugBelch("evac IND_OLDGEN: %ld bytes", size * sizeof(W_)); } #endif - - /* failed_to_evac might happen if we've got more than two - * generations, we're collecting only generation 0, the - * indirection resides in generation 2 and the indirectee is - * in generation 1. - */ - if (failed_to_evac) { - failed_to_evac = rtsFalse; - p->mut_link = new_list; - new_list = p; - } else { - /* the mut_link field of an IND_STATIC is overloaded as the - * static link field too (it just so happens that we don't need - * both at the same time), so we need to NULL it out when - * removing this object from the mutable list because the static - * link fields are all assumed to be NULL before doing a major - * collection. - */ - p->mut_link = NULL; - } - continue; - - case MUT_VAR: - /* MUT_CONS is a kind of MUT_VAR, except it that we try to remove - * it from the mutable list if possible by promoting whatever it - * points to. - */ - ASSERT(p->header.info == &MUT_CONS_info); - if (scavenge_one(((StgMutVar *)p)->var) == rtsTrue) { - /* didn't manage to promote everything, so put the - * MUT_CONS back on the list. - */ - p->mut_link = new_list; - new_list = p; - } - continue; - - case CAF_ENTERED: - { - StgCAF *caf = (StgCAF *)p; - caf->body = evacuate(caf->body); - caf->value = evacuate(caf->value); - if (failed_to_evac) { - failed_to_evac = rtsFalse; - p->mut_link = new_list; - new_list = p; - } else { - p->mut_link = NULL; - } - } - continue; - - case CAF_UNENTERED: - { - StgCAF *caf = (StgCAF *)p; - caf->body = evacuate(caf->body); - if (failed_to_evac) { - failed_to_evac = rtsFalse; - p->mut_link = new_list; - new_list = p; - } else { - p->mut_link = NULL; - } - } - continue; + break; default: - /* shouldn't have anything else on the mutables list */ - barf("scavenge_mut_once_list: strange object? %d", (int)(info->type)); - } - } + barf("scavenge_one: strange object %d", (int)(info->type)); + } - gen->mut_once_list = new_list; + no_luck = failed_to_evac; + failed_to_evac = rtsFalse; + return (no_luck); } -//@cindex scavenge_mutable_list +/* ----------------------------------------------------------------------------- + Scavenging mutable lists. + + We treat the mutable list of each generation > N (i.e. all the + generations older than the one being collected) as roots. We also + remove non-mutable objects from the mutable list at this point. + -------------------------------------------------------------------------- */ static void scavenge_mutable_list(generation *gen) { - const StgInfoTable *info; - StgMutClosure *p, *next; - - p = gen->saved_mut_list; - next = p->mut_link; - - evac_gen = 0; - failed_to_evac = rtsFalse; - - for (; p != END_MUT_LIST; p = next, next = p->mut_link) { - - /* make sure the info pointer is into text space */ - ASSERT(p && (LOOKS_LIKE_GHC_INFO(GET_INFO(p)) - || IS_HUGS_CONSTR_INFO(GET_INFO(p)))); - - info = get_itbl(p); - /* - if (info->type==RBH) - info = REVERT_INFOPTR(info); // if it's an RBH, look at the orig closure - */ - switch(info->type) { - - case MUT_ARR_PTRS_FROZEN: - /* remove this guy from the mutable list, but follow the ptrs - * anyway (and make sure they get promoted to this gen). - */ - { - StgPtr end, q; - - end = (P_)p + mut_arr_ptrs_sizeW((StgMutArrPtrs*)p); - evac_gen = gen->no; - for (q = (P_)((StgMutArrPtrs *)p)->payload; q < end; q++) { - (StgClosure *)*q = evacuate((StgClosure *)*q); - } - evac_gen = 0; - - if (failed_to_evac) { - failed_to_evac = rtsFalse; - p->mut_link = gen->mut_list; - gen->mut_list = p; - } - continue; - } - - case MUT_ARR_PTRS: - /* follow everything */ - p->mut_link = gen->mut_list; - gen->mut_list = p; - { - StgPtr end, q; - - end = (P_)p + mut_arr_ptrs_sizeW((StgMutArrPtrs*)p); - for (q = (P_)((StgMutArrPtrs *)p)->payload; q < end; q++) { - (StgClosure *)*q = evacuate((StgClosure *)*q); - } - continue; - } - - case MUT_VAR: - /* MUT_CONS is a kind of MUT_VAR, except that we try to remove - * it from the mutable list if possible by promoting whatever it - * points to. - */ - ASSERT(p->header.info != &MUT_CONS_info); - ((StgMutVar *)p)->var = evacuate(((StgMutVar *)p)->var); - p->mut_link = gen->mut_list; - gen->mut_list = p; - continue; - - case MVAR: - { - StgMVar *mvar = (StgMVar *)p; - (StgClosure *)mvar->head = evacuate((StgClosure *)mvar->head); - (StgClosure *)mvar->tail = evacuate((StgClosure *)mvar->tail); - (StgClosure *)mvar->value = evacuate((StgClosure *)mvar->value); - p->mut_link = gen->mut_list; - gen->mut_list = p; - continue; - } - - case TSO: - { - StgTSO *tso = (StgTSO *)p; - - scavengeTSO(tso); - - /* Don't take this TSO off the mutable list - it might still - * point to some younger objects (because we set evac_gen to 0 - * above). - */ - tso->mut_link = gen->mut_list; - gen->mut_list = (StgMutClosure *)tso; - continue; - } - - case BLACKHOLE_BQ: - { - StgBlockingQueue *bh = (StgBlockingQueue *)p; - (StgClosure *)bh->blocking_queue = - evacuate((StgClosure *)bh->blocking_queue); - p->mut_link = gen->mut_list; - gen->mut_list = p; - continue; - } - - /* Happens if a BLACKHOLE_BQ in the old generation is updated: - */ - case IND_OLDGEN: - case IND_OLDGEN_PERM: - /* Try to pull the indirectee into this generation, so we can - * remove the indirection from the mutable list. - */ - evac_gen = gen->no; - ((StgIndOldGen *)p)->indirectee = - evacuate(((StgIndOldGen *)p)->indirectee); - evac_gen = 0; - - if (failed_to_evac) { - failed_to_evac = rtsFalse; - p->mut_link = gen->mut_once_list; - gen->mut_once_list = p; - } else { - p->mut_link = NULL; - } - continue; - -#if defined(PAR) - // HWL: check whether all of these are necessary - - case RBH: // cf. BLACKHOLE_BQ - { - // nat size, ptrs, nonptrs, vhs; - // char str[80]; - // StgInfoTable *rip = get_closure_info(p, &size, &ptrs, &nonptrs, &vhs, str); - StgRBH *rbh = (StgRBH *)p; - (StgClosure *)rbh->blocking_queue = - evacuate((StgClosure *)rbh->blocking_queue); - if (failed_to_evac) { - failed_to_evac = rtsFalse; - recordMutable((StgMutClosure *)rbh); - } - // ToDo: use size of reverted closure here! - p += BLACKHOLE_sizeW(); - break; - } - - case BLOCKED_FETCH: - { - StgBlockedFetch *bf = (StgBlockedFetch *)p; - /* follow the pointer to the node which is being demanded */ - (StgClosure *)bf->node = - evacuate((StgClosure *)bf->node); - /* follow the link to the rest of the blocking queue */ - (StgClosure *)bf->link = - evacuate((StgClosure *)bf->link); - if (failed_to_evac) { - failed_to_evac = rtsFalse; - recordMutable((StgMutClosure *)bf); - } - p += sizeofW(StgBlockedFetch); - break; - } - - case FETCH_ME: - p += sizeofW(StgFetchMe); - break; // nothing to do in this case - - case FETCH_ME_BQ: // cf. BLACKHOLE_BQ - { - StgFetchMeBlockingQueue *fmbq = (StgFetchMeBlockingQueue *)p; - (StgClosure *)fmbq->blocking_queue = - evacuate((StgClosure *)fmbq->blocking_queue); - if (failed_to_evac) { - failed_to_evac = rtsFalse; - recordMutable((StgMutClosure *)fmbq); + bdescr *bd; + StgPtr p, q; + + bd = gen->saved_mut_list; + + evac_gen = gen->no; + for (; bd != NULL; bd = bd->link) { + for (q = bd->start; q < bd->free; q++) { + p = (StgPtr)*q; + ASSERT(LOOKS_LIKE_CLOSURE_PTR(p)); + if (scavenge_one(p)) { + /* didn't manage to promote everything, so put the + * object back on the list. + */ + recordMutableGen((StgClosure *)p,gen); + } } - p += sizeofW(StgFetchMeBlockingQueue); - break; - } -#endif - - default: - /* shouldn't have anything else on the mutables list */ - barf("scavenge_mutable_list: strange object? %d", (int)(info->type)); } - } + + // free the old mut_list + freeChain(gen->saved_mut_list); + gen->saved_mut_list = NULL; } -//@cindex scavenge_static static void scavenge_static(void) @@ -2669,14 +3543,13 @@ scavenge_static(void) list... */ while (p != END_OF_STATIC_LIST) { + ASSERT(LOOKS_LIKE_CLOSURE_PTR(p)); info = get_itbl(p); /* if (info->type==RBH) info = REVERT_INFOPTR(info); // if it's an RBH, look at the orig closure */ - /* make sure the info pointer is into text space */ - ASSERT(p && (LOOKS_LIKE_GHC_INFO(GET_INFO(p)) - || IS_HUGS_CONSTR_INFO(GET_INFO(p)))); + // make sure the info pointer is into text space /* Take this object *off* the static_objects list, * and put it on the scavenged_static_objects list. @@ -2693,32 +3566,33 @@ scavenge_static(void) ind->indirectee = evacuate(ind->indirectee); /* might fail to evacuate it, in which case we have to pop it - * back on the mutable list (and take it off the - * scavenged_static list because the static link and mut link - * pointers are one and the same). + * back on the mutable list of the oldest generation. We + * leave it *on* the scavenged_static_objects list, though, + * in case we visit this object again. */ if (failed_to_evac) { failed_to_evac = rtsFalse; - scavenged_static_objects = STATIC_LINK(info,p); - ((StgMutClosure *)ind)->mut_link = oldest_gen->mut_once_list; - oldest_gen->mut_once_list = (StgMutClosure *)ind; + recordMutableGen((StgClosure *)p,oldest_gen); } break; } case THUNK_STATIC: + scavenge_thunk_srt(info); + break; + case FUN_STATIC: - scavenge_srt(info); - /* fall through */ + scavenge_fun_srt(info); + break; case CONSTR_STATIC: { StgPtr q, next; next = (P_)p->payload + info->layout.payload.ptrs; - /* evacuate the pointers */ + // evacuate the pointers for (q = (P_)p->payload; q < next; q++) { - (StgClosure *)*q = evacuate((StgClosure *)*q); + *q = (StgWord)(StgPtr)evacuate((StgClosure *)*q); } break; } @@ -2738,194 +3612,166 @@ scavenge_static(void) } /* ----------------------------------------------------------------------------- - scavenge_stack walks over a section of stack and evacuates all the - objects pointed to by it. We can use the same code for walking - PAPs, since these are just sections of copied stack. + scavenge a chunk of memory described by a bitmap -------------------------------------------------------------------------- */ -//@cindex scavenge_stack static void -scavenge_stack(StgPtr p, StgPtr stack_end) +scavenge_large_bitmap( StgPtr p, StgLargeBitmap *large_bitmap, nat size ) { - StgPtr q; - const StgInfoTable* info; - StgWord32 bitmap; - - //IF_DEBUG(sanity, belch(" scavenging stack between %p and %p", p, stack_end)); - - /* - * Each time around this loop, we are looking at a chunk of stack - * that starts with either a pending argument section or an - * activation record. - */ - - while (p < stack_end) { - q = *(P_ *)p; - - /* If we've got a tag, skip over that many words on the stack */ - if (IS_ARG_TAG((W_)q)) { - p += ARG_SIZE(q); - p++; continue; + nat i, b; + StgWord bitmap; + + b = 0; + bitmap = large_bitmap->bitmap[b]; + for (i = 0; i < size; ) { + if ((bitmap & 1) == 0) { + *p = (StgWord)(StgPtr)evacuate((StgClosure *)*p); + } + i++; + p++; + if (i % BITS_IN(W_) == 0) { + b++; + bitmap = large_bitmap->bitmap[b]; + } else { + bitmap = bitmap >> 1; + } } - - /* Is q a pointer to a closure? - */ - if (! LOOKS_LIKE_GHC_INFO(q) ) { -#ifdef DEBUG - if ( 0 && LOOKS_LIKE_STATIC_CLOSURE(q) ) { /* Is it a static closure? */ - ASSERT(closure_STATIC((StgClosure *)q)); - } - /* otherwise, must be a pointer into the allocation space. */ -#endif +} - (StgClosure *)*p = evacuate((StgClosure *)q); - p++; - continue; +STATIC_INLINE StgPtr +scavenge_small_bitmap (StgPtr p, nat size, StgWord bitmap) +{ + while (size > 0) { + if ((bitmap & 1) == 0) { + *p = (StgWord)(StgPtr)evacuate((StgClosure *)*p); + } + p++; + bitmap = bitmap >> 1; + size--; } - - /* - * Otherwise, q must be the info pointer of an activation - * record. All activation records have 'bitmap' style layout - * info. - */ - info = get_itbl((StgClosure *)p); - - switch (info->type) { - - /* Dynamic bitmap: the mask is stored on the stack */ - case RET_DYN: - bitmap = ((StgRetDyn *)p)->liveness; - p = (P_)&((StgRetDyn *)p)->payload[0]; - goto small_bitmap; + return p; +} - /* probably a slow-entry point return address: */ - case FUN: - case FUN_STATIC: - { -#if 0 - StgPtr old_p = p; - p++; p++; - IF_DEBUG(sanity, - belch("HWL: scavenge_stack: FUN(_STATIC) adjusting p from %p to %p (instead of %p)", - old_p, p, old_p+1)); -#else - p++; /* what if FHS!=1 !? -- HWL */ -#endif - goto follow_srt; - } +/* ----------------------------------------------------------------------------- + scavenge_stack walks over a section of stack and evacuates all the + objects pointed to by it. We can use the same code for walking + AP_STACK_UPDs, since these are just sections of copied stack. + -------------------------------------------------------------------------- */ - /* Specialised code for update frames, since they're so common. - * We *know* the updatee points to a BLACKHOLE, CAF_BLACKHOLE, - * or BLACKHOLE_BQ, so just inline the code to evacuate it here. - */ - case UPDATE_FRAME: - { - StgUpdateFrame *frame = (StgUpdateFrame *)p; - StgClosure *to; - nat type = get_itbl(frame->updatee)->type; - p += sizeofW(StgUpdateFrame); - if (type == EVACUATED) { - frame->updatee = evacuate(frame->updatee); - continue; - } else { - bdescr *bd = Bdescr((P_)frame->updatee); - step *step; - if (bd->gen->no > N) { - if (bd->gen->no < evac_gen) { - failed_to_evac = rtsTrue; - } - continue; - } +static void +scavenge_stack(StgPtr p, StgPtr stack_end) +{ + const StgRetInfoTable* info; + StgWord bitmap; + nat size; - /* Don't promote blackholes */ - step = bd->step; - if (!(step->gen->no == 0 && - step->no != 0 && - step->no == step->gen->n_steps-1)) { - step = step->to; - } + //IF_DEBUG(sanity, debugBelch(" scavenging stack between %p and %p", p, stack_end)); - switch (type) { - case BLACKHOLE: - case CAF_BLACKHOLE: - to = copyPart(frame->updatee, BLACKHOLE_sizeW(), - sizeofW(StgHeader), step); - frame->updatee = to; - continue; - case BLACKHOLE_BQ: - to = copy(frame->updatee, BLACKHOLE_sizeW(), step); - frame->updatee = to; - recordMutable((StgMutClosure *)to); - continue; - default: - /* will never be SE_{,CAF_}BLACKHOLE, since we - don't push an update frame for single-entry thunks. KSW 1999-01. */ - barf("scavenge_stack: UPDATE_FRAME updatee"); - } - } - } + /* + * Each time around this loop, we are looking at a chunk of stack + * that starts with an activation record. + */ + + while (p < stack_end) { + info = get_ret_itbl((StgClosure *)p); + + switch (info->i.type) { + + case UPDATE_FRAME: + ((StgUpdateFrame *)p)->updatee + = evacuate(((StgUpdateFrame *)p)->updatee); + p += sizeofW(StgUpdateFrame); + continue; - /* small bitmap (< 32 entries, or 64 on a 64-bit machine) */ + // small bitmap (< 32 entries, or 64 on a 64-bit machine) + case CATCH_STM_FRAME: + case CATCH_RETRY_FRAME: + case ATOMICALLY_FRAME: case STOP_FRAME: case CATCH_FRAME: - case SEQ_FRAME: - case RET_BCO: case RET_SMALL: case RET_VEC_SMALL: - bitmap = info->layout.bitmap; - p++; - /* this assumes that the payload starts immediately after the info-ptr */ - small_bitmap: - while (bitmap != 0) { - if ((bitmap & 1) == 0) { - (StgClosure *)*p = evacuate((StgClosure *)*p); - } + bitmap = BITMAP_BITS(info->i.layout.bitmap); + size = BITMAP_SIZE(info->i.layout.bitmap); + // NOTE: the payload starts immediately after the info-ptr, we + // don't have an StgHeader in the same sense as a heap closure. p++; - bitmap = bitmap >> 1; - } - + p = scavenge_small_bitmap(p, size, bitmap); + follow_srt: - scavenge_srt(info); - continue; + scavenge_srt((StgClosure **)GET_SRT(info), info->i.srt_bitmap); + continue; + + case RET_BCO: { + StgBCO *bco; + nat size; - /* large bitmap (> 32 entries) */ + p++; + *p = (StgWord)(StgPtr)evacuate((StgClosure *)*p); + bco = (StgBCO *)*p; + p++; + size = BCO_BITMAP_SIZE(bco); + scavenge_large_bitmap(p, BCO_BITMAP(bco), size); + p += size; + continue; + } + + // large bitmap (> 32 entries, or > 64 on a 64-bit machine) case RET_BIG: case RET_VEC_BIG: - { - StgPtr q; - StgLargeBitmap *large_bitmap; - nat i; + { + nat size; - large_bitmap = info->layout.large_bitmap; + size = GET_LARGE_BITMAP(&info->i)->size; p++; + scavenge_large_bitmap(p, GET_LARGE_BITMAP(&info->i), size); + p += size; + // and don't forget to follow the SRT + goto follow_srt; + } - for (i=0; isize; i++) { - bitmap = large_bitmap->bitmap[i]; - q = p + sizeof(W_) * 8; - while (bitmap != 0) { - if ((bitmap & 1) == 0) { - (StgClosure *)*p = evacuate((StgClosure *)*p); - } + // Dynamic bitmap: the mask is stored on the stack, and + // there are a number of non-pointers followed by a number + // of pointers above the bitmapped area. (see StgMacros.h, + // HEAP_CHK_GEN). + case RET_DYN: + { + StgWord dyn; + dyn = ((StgRetDyn *)p)->liveness; + + // traverse the bitmap first + bitmap = RET_DYN_LIVENESS(dyn); + p = (P_)&((StgRetDyn *)p)->payload[0]; + size = RET_DYN_BITMAP_SIZE; + p = scavenge_small_bitmap(p, size, bitmap); + + // skip over the non-ptr words + p += RET_DYN_NONPTRS(dyn) + RET_DYN_NONPTR_REGS_SIZE; + + // follow the ptr words + for (size = RET_DYN_PTRS(dyn); size > 0; size--) { + *p = (StgWord)(StgPtr)evacuate((StgClosure *)*p); p++; - bitmap = bitmap >> 1; - } - if (i+1 < large_bitmap->size) { - while (p < q) { - (StgClosure *)*p = evacuate((StgClosure *)*p); - p++; - } - } } + continue; + } + + case RET_FUN: + { + StgRetFun *ret_fun = (StgRetFun *)p; + StgFunInfoTable *fun_info; - /* and don't forget to follow the SRT */ + ret_fun->fun = evacuate(ret_fun->fun); + fun_info = get_fun_itbl(ret_fun->fun); + p = scavenge_arg_block(fun_info, ret_fun->payload); goto follow_srt; - } + } default: - barf("scavenge_stack: weird activation record found on stack: %d", (int)(info->type)); + barf("scavenge_stack: weird activation record found on stack: %d", (int)(info->i.type)); } - } + } } /*----------------------------------------------------------------------------- @@ -2936,104 +3782,38 @@ scavenge_stack(StgPtr p, StgPtr stack_end) objects are (repeatedly) mutable, so most of the time evac_gen will be zero. --------------------------------------------------------------------------- */ -//@cindex scavenge_large static void -scavenge_large(step *step) +scavenge_large(step *stp) { bdescr *bd; StgPtr p; - const StgInfoTable* info; - nat saved_evac_gen = evac_gen; /* used for temporarily changing evac_gen */ - evac_gen = 0; /* most objects are mutable */ - bd = step->new_large_objects; + bd = stp->new_large_objects; - for (; bd != NULL; bd = step->new_large_objects) { + for (; bd != NULL; bd = stp->new_large_objects) { /* take this object *off* the large objects list and put it on * the scavenged large objects list. This is so that we can * treat new_large_objects as a stack and push new objects on * the front when evacuating. */ - step->new_large_objects = bd->link; - dbl_link_onto(bd, &step->scavenged_large_objects); - - p = bd->start; - info = get_itbl((StgClosure *)p); - - switch (info->type) { - - /* only certain objects can be "large"... */ - - case ARR_WORDS: - /* nothing to follow */ - continue; - - case MUT_ARR_PTRS: - /* follow everything */ - { - StgPtr next; - - next = p + mut_arr_ptrs_sizeW((StgMutArrPtrs*)p); - for (p = (P_)((StgMutArrPtrs *)p)->payload; p < next; p++) { - (StgClosure *)*p = evacuate((StgClosure *)*p); - } - continue; - } - - case MUT_ARR_PTRS_FROZEN: - /* follow everything */ - { - StgPtr start = p, next; - - evac_gen = saved_evac_gen; /* not really mutable */ - next = p + mut_arr_ptrs_sizeW((StgMutArrPtrs*)p); - for (p = (P_)((StgMutArrPtrs *)p)->payload; p < next; p++) { - (StgClosure *)*p = evacuate((StgClosure *)*p); - } - evac_gen = 0; - if (failed_to_evac) { - recordMutable((StgMutClosure *)start); - } - continue; - } - - case BCO: - { - StgBCO* bco = (StgBCO *)p; - nat i; - evac_gen = saved_evac_gen; - for (i = 0; i < bco->n_ptrs; i++) { - bcoConstCPtr(bco,i) = evacuate(bcoConstCPtr(bco,i)); - } - evac_gen = 0; - continue; - } + stp->new_large_objects = bd->link; + dbl_link_onto(bd, &stp->scavenged_large_objects); - case TSO: - scavengeTSO((StgTSO *)p); - continue; - - case AP_UPD: - case PAP: - { - StgPAP* pap = (StgPAP *)p; - - evac_gen = saved_evac_gen; /* not really mutable */ - pap->fun = evacuate(pap->fun); - scavenge_stack((P_)pap->payload, (P_)pap->payload + pap->n_args); - evac_gen = 0; - continue; - } + // update the block count in this step. + stp->n_scavenged_large_blocks += bd->blocks; - default: - barf("scavenge_large: unknown/strange object %d", (int)(info->type)); + p = bd->start; + if (scavenge_one(p)) { + recordMutableGen((StgClosure *)p, stp->gen); } } } -//@cindex zero_static_object_list +/* ----------------------------------------------------------------------------- + Initialising the static object & mutable lists + -------------------------------------------------------------------------- */ static void zero_static_object_list(StgClosure* first_static) @@ -3049,64 +3829,36 @@ zero_static_object_list(StgClosure* first_static) } } -/* This function is only needed because we share the mutable link - * field with the static link field in an IND_STATIC, so we have to - * zero the mut_link field before doing a major GC, which needs the - * static link field. - * - * It doesn't do any harm to zero all the mutable link fields on the - * mutable list. - */ -//@cindex zero_mutable_list - -static void -zero_mutable_list( StgMutClosure *first ) -{ - StgMutClosure *next, *c; - - for (c = first; c != END_MUT_LIST; c = next) { - next = c->mut_link; - c->mut_link = NULL; - } -} - -//@node Reverting CAFs, Sanity code for CAF garbage collection, Scavenging -//@subsection Reverting CAFs - /* ----------------------------------------------------------------------------- Reverting CAFs -------------------------------------------------------------------------- */ -//@cindex RevertCAFs -void RevertCAFs(void) +void +revertCAFs( void ) { -#ifdef INTERPRETER - StgInt i; - - /* Deal with CAFs created by compiled code. */ - for (i = 0; i < usedECafTable; i++) { - SET_INFO( (StgInd*)(ecafTable[i].closure), ecafTable[i].origItbl ); - ((StgInd*)(ecafTable[i].closure))->indirectee = 0; - } - - /* Deal with CAFs created by the interpreter. */ - while (ecafList != END_ECAF_LIST) { - StgCAF* caf = ecafList; - ecafList = caf->link; - ASSERT(get_itbl(caf)->type == CAF_ENTERED); - SET_INFO(caf,&CAF_UNENTERED_info); - caf->value = (StgClosure *)0xdeadbeef; - caf->link = (StgCAF *)0xdeadbeef; - } - - /* Empty out both the table and the list. */ - clearECafTable(); - ecafList = END_ECAF_LIST; -#endif + StgIndStatic *c; + + for (c = (StgIndStatic *)caf_list; c != NULL; + c = (StgIndStatic *)c->static_link) + { + SET_INFO(c, c->saved_info); + c->saved_info = NULL; + // could, but not necessary: c->static_link = NULL; + } + caf_list = NULL; } -//@node Sanity code for CAF garbage collection, Lazy black holing, Reverting CAFs -//@subsection Sanity code for CAF garbage collection +void +markCAFs( evac_fn evac ) +{ + StgIndStatic *c; + + for (c = (StgIndStatic *)caf_list; c != NULL; + c = (StgIndStatic *)c->static_link) + { + evac(&c->indirectee); + } +} /* ----------------------------------------------------------------------------- Sanity code for CAF garbage collection. @@ -3120,8 +3872,7 @@ void RevertCAFs(void) time. -------------------------------------------------------------------------- */ -#ifdef DEBUG -//@cindex gcCAFs +#if 0 && defined(DEBUG) static void gcCAFs(void) @@ -3142,9 +3893,9 @@ gcCAFs(void) ASSERT(info->type == IND_STATIC); if (STATIC_LINK(info,p) == NULL) { - IF_DEBUG(gccafs, fprintf(stderr, "CAF gc'd at 0x%04x\n", (int)p)); - /* black hole it */ - SET_INFO(p,&BLACKHOLE_info); + IF_DEBUG(gccafs, debugBelch("CAF gc'd at 0x%04lx", (long)p)); + // black hole it + SET_INFO(p,&stg_BLACKHOLE_info); p = STATIC_LINK2(info,p); *pp = p; } @@ -3156,12 +3907,10 @@ gcCAFs(void) } - /* fprintf(stderr, "%d CAFs live\n", i); */ + // debugBelch("%d CAFs live", i); } #endif -//@node Lazy black holing, Stack squeezing, Sanity code for CAF garbage collection -//@subsection Lazy black holing /* ----------------------------------------------------------------------------- Lazy black holing. @@ -3170,64 +3919,67 @@ gcCAFs(void) some work, we have to run down the stack and black-hole all the closures referred to by update frames. -------------------------------------------------------------------------- */ -//@cindex threadLazyBlackHole static void threadLazyBlackHole(StgTSO *tso) { - StgUpdateFrame *update_frame; - StgBlockingQueue *bh; - StgPtr stack_end; - - stack_end = &tso->stack[tso->stack_size]; - update_frame = tso->su; - - while (1) { - switch (get_itbl(update_frame)->type) { - - case CATCH_FRAME: - update_frame = ((StgCatchFrame *)update_frame)->link; - break; - - case UPDATE_FRAME: - bh = (StgBlockingQueue *)update_frame->updatee; - - /* if the thunk is already blackholed, it means we've also - * already blackholed the rest of the thunks on this stack, - * so we can stop early. - * - * The blackhole made for a CAF is a CAF_BLACKHOLE, so they - * don't interfere with this optimisation. - */ - if (bh->header.info == &BLACKHOLE_info) { - return; - } + StgClosure *frame; + StgRetInfoTable *info; + StgBlockingQueue *bh; + StgPtr stack_end; + + stack_end = &tso->stack[tso->stack_size]; + + frame = (StgClosure *)tso->sp; - if (bh->header.info != &BLACKHOLE_BQ_info && - bh->header.info != &CAF_BLACKHOLE_info) { + while (1) { + info = get_ret_itbl(frame); + + switch (info->i.type) { + + case UPDATE_FRAME: + bh = (StgBlockingQueue *)((StgUpdateFrame *)frame)->updatee; + + /* if the thunk is already blackholed, it means we've also + * already blackholed the rest of the thunks on this stack, + * so we can stop early. + * + * The blackhole made for a CAF is a CAF_BLACKHOLE, so they + * don't interfere with this optimisation. + */ + if (bh->header.info == &stg_BLACKHOLE_info) { + return; + } + + if (bh->header.info != &stg_BLACKHOLE_BQ_info && + bh->header.info != &stg_CAF_BLACKHOLE_info) { #if (!defined(LAZY_BLACKHOLING)) && defined(DEBUG) - fprintf(stderr,"Unexpected lazy BHing required at 0x%04x\n",(int)bh); + debugBelch("Unexpected lazy BHing required at 0x%04x",(int)bh); #endif - SET_INFO(bh,&BLACKHOLE_info); - } - - update_frame = update_frame->link; - break; - - case SEQ_FRAME: - update_frame = ((StgSeqFrame *)update_frame)->link; - break; +#ifdef PROFILING + // @LDV profiling + // We pretend that bh is now dead. + LDV_recordDead_FILL_SLOP_DYNAMIC((StgClosure *)bh); +#endif + SET_INFO(bh,&stg_BLACKHOLE_info); - case STOP_FRAME: - return; - default: - barf("threadPaused"); + // We pretend that bh has just been created. + LDV_RECORD_CREATE(bh); + } + + frame = (StgClosure *) ((StgUpdateFrame *)frame + 1); + break; + + case STOP_FRAME: + return; + + // normal stack frames; do nothing except advance the pointer + default: + frame = (StgClosure *)((StgPtr)frame + stack_frame_sizeW(frame)); + } } - } } -//@node Stack squeezing, Pausing a thread, Lazy black holing -//@subsection Stack squeezing /* ----------------------------------------------------------------------------- * Stack squeezing @@ -3236,236 +3988,204 @@ threadLazyBlackHole(StgTSO *tso) * lazy black holing here. * * -------------------------------------------------------------------------- */ -//@cindex threadSqueezeStack + +struct stack_gap { StgWord gap_size; struct stack_gap *next_gap; }; static void threadSqueezeStack(StgTSO *tso) { - lnat displacement = 0; - StgUpdateFrame *frame; - StgUpdateFrame *next_frame; /* Temporally next */ - StgUpdateFrame *prev_frame; /* Temporally previous */ - StgPtr bottom; - rtsBool prev_was_update_frame; -#if DEBUG - StgUpdateFrame *top_frame; - nat upd_frames=0, stop_frames=0, catch_frames=0, seq_frames=0, - bhs=0, squeezes=0; - void printObj( StgClosure *obj ); // from Printer.c + StgPtr frame; + rtsBool prev_was_update_frame; + StgClosure *updatee = NULL; + StgPtr bottom; + StgRetInfoTable *info; + StgWord current_gap_size; + struct stack_gap *gap; - top_frame = tso->su; -#endif - - bottom = &(tso->stack[tso->stack_size]); - frame = tso->su; + // Stage 1: + // Traverse the stack upwards, replacing adjacent update frames + // with a single update frame and a "stack gap". A stack gap + // contains two values: the size of the gap, and the distance + // to the next gap (or the stack top). - /* There must be at least one frame, namely the STOP_FRAME. - */ - ASSERT((P_)frame < bottom); + bottom = &(tso->stack[tso->stack_size]); - /* Walk down the stack, reversing the links between frames so that - * we can walk back up as we squeeze from the bottom. Note that - * next_frame and prev_frame refer to next and previous as they were - * added to the stack, rather than the way we see them in this - * walk. (It makes the next loop less confusing.) - * - * Stop if we find an update frame pointing to a black hole - * (see comment in threadLazyBlackHole()). - */ - - next_frame = NULL; - /* bottom - sizeof(StgStopFrame) is the STOP_FRAME */ - while ((P_)frame < bottom - sizeofW(StgStopFrame)) { - prev_frame = frame->link; - frame->link = next_frame; - next_frame = frame; - frame = prev_frame; -#if DEBUG - IF_DEBUG(sanity, - if (!(frame>=top_frame && frame<=(StgUpdateFrame *)bottom)) { - printObj((StgClosure *)prev_frame); - barf("threadSqueezeStack: current frame is rubbish %p; previous was %p\n", - frame, prev_frame); - }) - switch (get_itbl(frame)->type) { - case UPDATE_FRAME: upd_frames++; - if (frame->updatee->header.info == &BLACKHOLE_info) - bhs++; - break; - case STOP_FRAME: stop_frames++; - break; - case CATCH_FRAME: catch_frames++; - break; - case SEQ_FRAME: seq_frames++; - break; - default: - barf("Found non-frame during stack squeezing at %p (prev frame was %p)\n", - frame, prev_frame); - printObj((StgClosure *)prev_frame); - } -#endif - if (get_itbl(frame)->type == UPDATE_FRAME - && frame->updatee->header.info == &BLACKHOLE_info) { - break; - } - } + frame = tso->sp; - /* Now, we're at the bottom. Frame points to the lowest update - * frame on the stack, and its link actually points to the frame - * above. We have to walk back up the stack, squeezing out empty - * update frames and turning the pointers back around on the way - * back up. - * - * The bottom-most frame (the STOP_FRAME) has not been altered, and - * we never want to eliminate it anyway. Just walk one step up - * before starting to squeeze. When you get to the topmost frame, - * remember that there are still some words above it that might have - * to be moved. - */ - - prev_frame = frame; - frame = next_frame; + ASSERT(frame < bottom); + + prev_was_update_frame = rtsFalse; + current_gap_size = 0; + gap = (struct stack_gap *) (tso->sp - sizeofW(StgUpdateFrame)); - prev_was_update_frame = (get_itbl(prev_frame)->type == UPDATE_FRAME); + while (frame < bottom) { + + info = get_ret_itbl((StgClosure *)frame); + switch (info->i.type) { - /* - * Loop through all of the frames (everything except the very - * bottom). Things are complicated by the fact that we have - * CATCH_FRAMEs and SEQ_FRAMEs interspersed with the update frames. - * We can only squeeze when there are two consecutive UPDATE_FRAMEs. - */ - while (frame != NULL) { - StgPtr sp; - StgPtr frame_bottom = (P_)frame + sizeofW(StgUpdateFrame); - rtsBool is_update_frame; - - next_frame = frame->link; - is_update_frame = (get_itbl(frame)->type == UPDATE_FRAME); + case UPDATE_FRAME: + { + StgUpdateFrame *upd = (StgUpdateFrame *)frame; - /* Check to see if - * 1. both the previous and current frame are update frames - * 2. the current frame is empty - */ - if (prev_was_update_frame && is_update_frame && - (P_)prev_frame == frame_bottom + displacement) { - - /* Now squeeze out the current frame */ - StgClosure *updatee_keep = prev_frame->updatee; - StgClosure *updatee_bypass = frame->updatee; - -#if DEBUG - IF_DEBUG(gc, fprintf(stderr, "@@ squeezing frame at %p\n", frame)); - squeezes++; -#endif + if (upd->updatee->header.info == &stg_BLACKHOLE_info) { - /* Deal with blocking queues. If both updatees have blocked - * threads, then we should merge the queues into the update - * frame that we're keeping. - * - * Alternatively, we could just wake them up: they'll just go - * straight to sleep on the proper blackhole! This is less code - * and probably less bug prone, although it's probably much - * slower --SDM - */ -#if 0 /* do it properly... */ -# if (!defined(LAZY_BLACKHOLING)) && defined(DEBUG) -# error Unimplemented lazy BH warning. (KSW 1999-01) -# endif - if (GET_INFO(updatee_bypass) == BLACKHOLE_BQ_info - || GET_INFO(updatee_bypass) == CAF_BLACKHOLE_info - ) { - /* Sigh. It has one. Don't lose those threads! */ - if (GET_INFO(updatee_keep) == BLACKHOLE_BQ_info) { - /* Urgh. Two queues. Merge them. */ - P_ keep_tso = ((StgBlockingQueue *)updatee_keep)->blocking_queue; - - while (keep_tso->link != END_TSO_QUEUE) { - keep_tso = keep_tso->link; - } - keep_tso->link = ((StgBlockingQueue *)updatee_bypass)->blocking_queue; + // found a BLACKHOLE'd update frame; we've been here + // before, in a previous GC, so just break out. - } else { - /* For simplicity, just swap the BQ for the BH */ - P_ temp = updatee_keep; - - updatee_keep = updatee_bypass; - updatee_bypass = temp; - - /* Record the swap in the kept frame (below) */ - prev_frame->updatee = updatee_keep; - } - } -#endif + // Mark the end of the gap, if we're in one. + if (current_gap_size != 0) { + gap = (struct stack_gap *)(frame-sizeofW(StgUpdateFrame)); + } + + frame += sizeofW(StgUpdateFrame); + goto done_traversing; + } - TICK_UPD_SQUEEZED(); - /* wasn't there something about update squeezing and ticky to be - * sorted out? oh yes: we aren't counting each enter properly - * in this case. See the log somewhere. KSW 1999-04-21 - */ - UPD_IND_NOLOCK(updatee_bypass, updatee_keep); /* this wakes the threads up */ - - sp = (P_)frame - 1; /* sp = stuff to slide */ - displacement += sizeofW(StgUpdateFrame); - - } else { - /* No squeeze for this frame */ - sp = frame_bottom - 1; /* Keep the current frame */ - - /* Do lazy black-holing. - */ - if (is_update_frame) { - StgBlockingQueue *bh = (StgBlockingQueue *)frame->updatee; - if (bh->header.info != &BLACKHOLE_info && - bh->header.info != &BLACKHOLE_BQ_info && - bh->header.info != &CAF_BLACKHOLE_info) { + if (prev_was_update_frame) { + + TICK_UPD_SQUEEZED(); + /* wasn't there something about update squeezing and ticky to be + * sorted out? oh yes: we aren't counting each enter properly + * in this case. See the log somewhere. KSW 1999-04-21 + * + * Check two things: that the two update frames don't point to + * the same object, and that the updatee_bypass isn't already an + * indirection. Both of these cases only happen when we're in a + * block hole-style loop (and there are multiple update frames + * on the stack pointing to the same closure), but they can both + * screw us up if we don't check. + */ + if (upd->updatee != updatee && !closure_IND(upd->updatee)) { + // this wakes the threads up + UPD_IND_NOLOCK(upd->updatee, updatee); + } + + // now mark this update frame as a stack gap. The gap + // marker resides in the bottom-most update frame of + // the series of adjacent frames, and covers all the + // frames in this series. + current_gap_size += sizeofW(StgUpdateFrame); + ((struct stack_gap *)frame)->gap_size = current_gap_size; + ((struct stack_gap *)frame)->next_gap = gap; + + frame += sizeofW(StgUpdateFrame); + continue; + } + + // single update frame, or the topmost update frame in a series + else { + StgBlockingQueue *bh = (StgBlockingQueue *)upd->updatee; + + // Do lazy black-holing + if (bh->header.info != &stg_BLACKHOLE_info && + bh->header.info != &stg_BLACKHOLE_BQ_info && + bh->header.info != &stg_CAF_BLACKHOLE_info) { #if (!defined(LAZY_BLACKHOLING)) && defined(DEBUG) - fprintf(stderr,"Unexpected lazy BHing required at 0x%04x\n",(int)bh); + debugBelch("Unexpected lazy BHing required at 0x%04x",(int)bh); +#endif +#ifdef DEBUG + /* zero out the slop so that the sanity checker can tell + * where the next closure is. + */ + { + StgInfoTable *bh_info = get_itbl(bh); + nat np = bh_info->layout.payload.ptrs, + nw = bh_info->layout.payload.nptrs, i; + /* don't zero out slop for a THUNK_SELECTOR, + * because its layout info is used for a + * different purpose, and it's exactly the + * same size as a BLACKHOLE in any case. + */ + if (bh_info->type != THUNK_SELECTOR) { + for (i = 0; i < np + nw; i++) { + ((StgClosure *)bh)->payload[i] = INVALID_OBJECT; + } + } + } +#endif +#ifdef PROFILING + // We pretend that bh is now dead. + LDV_recordDead_FILL_SLOP_DYNAMIC((StgClosure *)bh); #endif - SET_INFO(bh,&BLACKHOLE_info); + // Todo: maybe use SET_HDR() and remove LDV_RECORD_CREATE()? + SET_INFO(bh,&stg_BLACKHOLE_info); + + // We pretend that bh has just been created. + LDV_RECORD_CREATE(bh); + } + + prev_was_update_frame = rtsTrue; + updatee = upd->updatee; + frame += sizeofW(StgUpdateFrame); + continue; + } } - } + + default: + prev_was_update_frame = rtsFalse; + + // we're not in a gap... check whether this is the end of a gap + // (an update frame can't be the end of a gap). + if (current_gap_size != 0) { + gap = (struct stack_gap *) (frame - sizeofW(StgUpdateFrame)); + } + current_gap_size = 0; - /* Fix the link in the current frame (should point to the frame below) */ - frame->link = prev_frame; - prev_was_update_frame = is_update_frame; + frame += stack_frame_sizeW((StgClosure *)frame); + continue; + } } - - /* Now slide all words from sp up to the next frame */ - - if (displacement > 0) { - P_ next_frame_bottom; - if (next_frame != NULL) - next_frame_bottom = (P_)next_frame + sizeofW(StgUpdateFrame); - else - next_frame_bottom = tso->sp - 1; - -#if DEBUG - IF_DEBUG(gc, - fprintf(stderr, "sliding [%p, %p] by %ld\n", sp, next_frame_bottom, - displacement)) -#endif - - while (sp >= next_frame_bottom) { - sp[displacement] = *sp; - sp -= 1; - } - } - (P_)prev_frame = (P_)frame + displacement; - frame = next_frame; - } +done_traversing: + + // Now we have a stack with gaps in it, and we have to walk down + // shoving the stack up to fill in the gaps. A diagram might + // help: + // + // +| ********* | + // | ********* | <- sp + // | | + // | | <- gap_start + // | ......... | | + // | stack_gap | <- gap | chunk_size + // | ......... | | + // | ......... | <- gap_end v + // | ********* | + // | ********* | + // | ********* | + // -| ********* | + // + // 'sp' points the the current top-of-stack + // 'gap' points to the stack_gap structure inside the gap + // ***** indicates real stack data + // ..... indicates gap + // indicates unused + // + { + void *sp; + void *gap_start, *next_gap_start, *gap_end; + nat chunk_size; - tso->sp += displacement; - tso->su = prev_frame; -#if DEBUG - IF_DEBUG(gc, - fprintf(stderr, "@@ threadSqueezeStack: squeezed %d update-frames; found %d BHs; found %d update-, %d stop-, %d catch, %d seq-frames\n", - squeezes, bhs, upd_frames, stop_frames, catch_frames, seq_frames)) -#endif -} + next_gap_start = (void *)((unsigned char*)gap + sizeof(StgUpdateFrame)); + sp = next_gap_start; + + while ((StgPtr)gap > tso->sp) { + + // we're working in *bytes* now... + gap_start = next_gap_start; + gap_end = (void*) ((unsigned char*)gap_start - gap->gap_size * sizeof(W_)); -//@node Pausing a thread, Index, Stack squeezing -//@subsection Pausing a thread + gap = gap->next_gap; + next_gap_start = (void *)((unsigned char*)gap + sizeof(StgUpdateFrame)); + + chunk_size = (unsigned char*)gap_end - (unsigned char*)next_gap_start; + sp -= chunk_size; + memmove(sp, next_gap_start, chunk_size); + } + + tso->sp = (StgPtr)sp; + } +} /* ----------------------------------------------------------------------------- * Pausing a thread @@ -3474,12 +4194,11 @@ threadSqueezeStack(StgTSO *tso) * here. We also take the opportunity to do stack squeezing if it's * turned on. * -------------------------------------------------------------------------- */ -//@cindex threadPaused void threadPaused(StgTSO *tso) { if ( RtsFlags.GcFlags.squeezeUpdFrames == rtsTrue ) - threadSqueezeStack(tso); /* does black holing too */ + threadSqueezeStack(tso); // does black holing too else threadLazyBlackHole(tso); } @@ -3489,42 +4208,23 @@ threadPaused(StgTSO *tso) * -------------------------------------------------------------------------- */ #if DEBUG -//@cindex printMutOnceList -void -printMutOnceList(generation *gen) -{ - StgMutClosure *p, *next; - - p = gen->mut_once_list; - next = p->mut_link; - - fprintf(stderr, "@@ Mut once list %p: ", gen->mut_once_list); - for (; p != END_MUT_LIST; p = next, next = p->mut_link) { - fprintf(stderr, "%p (%s), ", - p, info_type((StgClosure *)p)); - } - fputc('\n', stderr); -} - -//@cindex printMutableList void printMutableList(generation *gen) { - StgMutClosure *p, *next; + bdescr *bd; + StgPtr p; - p = gen->mut_list; - next = p->mut_link; + debugBelch("@@ Mutable list %p: ", gen->mut_list); - fprintf(stderr, "@@ Mutable list %p: ", gen->mut_list); - for (; p != END_MUT_LIST; p = next, next = p->mut_link) { - fprintf(stderr, "%p (%s), ", - p, info_type((StgClosure *)p)); - } - fputc('\n', stderr); + for (bd = gen->mut_list; bd != NULL; bd = bd->link) { + for (p = bd->start; p < bd->free; p++) { + debugBelch("%p (%s), ", (void *)*p, info_type((StgClosure *)*p)); + } + } + debugBelch("\n"); } -//@cindex maybeLarge -static inline rtsBool +STATIC_INLINE rtsBool maybeLarge(StgClosure *closure) { StgInfoTable *info = get_itbl(closure); @@ -3533,47 +4233,10 @@ maybeLarge(StgClosure *closure) see scavenge_large */ return (info->type == MUT_ARR_PTRS || info->type == MUT_ARR_PTRS_FROZEN || + info->type == MUT_ARR_PTRS_FROZEN0 || info->type == TSO || - info->type == ARR_WORDS || - info->type == BCO); + info->type == ARR_WORDS); } -#endif /* DEBUG */ - -//@node Index, , Pausing a thread -//@subsection Index - -//@index -//* GarbageCollect:: @cindex\s-+GarbageCollect -//* MarkRoot:: @cindex\s-+MarkRoot -//* RevertCAFs:: @cindex\s-+RevertCAFs -//* addBlock:: @cindex\s-+addBlock -//* cleanup_weak_ptr_list:: @cindex\s-+cleanup_weak_ptr_list -//* copy:: @cindex\s-+copy -//* copyPart:: @cindex\s-+copyPart -//* evacuate:: @cindex\s-+evacuate -//* evacuate_large:: @cindex\s-+evacuate_large -//* gcCAFs:: @cindex\s-+gcCAFs -//* isAlive:: @cindex\s-+isAlive -//* maybeLarge:: @cindex\s-+maybeLarge -//* mkMutCons:: @cindex\s-+mkMutCons -//* printMutOnceList:: @cindex\s-+printMutOnceList -//* printMutableList:: @cindex\s-+printMutableList -//* relocate_TSO:: @cindex\s-+relocate_TSO -//* scavenge:: @cindex\s-+scavenge -//* scavenge_large:: @cindex\s-+scavenge_large -//* scavenge_mut_once_list:: @cindex\s-+scavenge_mut_once_list -//* scavenge_mutable_list:: @cindex\s-+scavenge_mutable_list -//* scavenge_one:: @cindex\s-+scavenge_one -//* scavenge_srt:: @cindex\s-+scavenge_srt -//* scavenge_stack:: @cindex\s-+scavenge_stack -//* scavenge_static:: @cindex\s-+scavenge_static -//* threadLazyBlackHole:: @cindex\s-+threadLazyBlackHole -//* threadPaused:: @cindex\s-+threadPaused -//* threadSqueezeStack:: @cindex\s-+threadSqueezeStack -//* traverse_weak_ptr_list:: @cindex\s-+traverse_weak_ptr_list -//* upd_evacuee:: @cindex\s-+upd_evacuee -//* zero_mutable_list:: @cindex\s-+zero_mutable_list -//* zero_static_object_list:: @cindex\s-+zero_static_object_list -//@end index +#endif // DEBUG