rts/sm/Storage.h

   1 /* -----------------------------------------------------------------------------
   2  *
   3  * (c) The GHC Team, 1998-2009
   4  *
   5  * External Storage Manger Interface
   6  *
   7  * ---------------------------------------------------------------------------*/
   8
   9 #ifndef SM_STORAGE_H
  10 #define SM_STORAGE_H
  11
  12 #include "Capability.h"
  13
  14 #include "BeginPrivate.h"
  15
  16 /* -----------------------------------------------------------------------------
  17    Initialisation / De-initialisation
  18    -------------------------------------------------------------------------- */
  19
  20 void initStorage(void);
  21 void exitStorage(void);
  22 void freeStorage(rtsBool free_heap);
  23
  24 /* -----------------------------------------------------------------------------
  25    Storage manager state
  26    -------------------------------------------------------------------------- */
  27
  28 INLINE_HEADER rtsBool
  29 doYouWantToGC( Capability *cap )
  30 {
  31   return (cap->r.rCurrentNursery->link == NULL ||
  32           g0->n_new_large_words >= large_alloc_lim);
  33 }
  34
  35 /* for splitting blocks groups in two */
  36 bdescr * splitLargeBlock (bdescr *bd, nat blocks);
  37
  38 /* -----------------------------------------------------------------------------
  39    Generational garbage collection support
  40
  41    recordMutable(StgPtr p)       Informs the garbage collector that a
  42                                  previously immutable object has
  43                                  become (permanently) mutable.  Used
  44                                  by thawArray and similar.
  45
  46    updateWithIndirection(p1,p2)  Updates the object at p1 with an
  47                                  indirection pointing to p2.  This is
  48                                  normally called for objects in an old
  49                                  generation (>0) when they are updated.
  50
  51    updateWithPermIndirection(p1,p2)  As above but uses a permanent indir.
  52
  53    -------------------------------------------------------------------------- */
  54
  55 /*
  56  * Storage manager mutex
  57  */
  58 #if defined(THREADED_RTS)
  59 extern Mutex sm_mutex;
  60 #endif
  61
  62 #if defined(THREADED_RTS)
  63 #define ACQUIRE_SM_LOCK   ACQUIRE_LOCK(&sm_mutex);
  64 #define RELEASE_SM_LOCK   RELEASE_LOCK(&sm_mutex);
  65 #define ASSERT_SM_LOCK()  ASSERT_LOCK_HELD(&sm_mutex);
  66 #else
  67 #define ACQUIRE_SM_LOCK
  68 #define RELEASE_SM_LOCK
  69 #define ASSERT_SM_LOCK()
  70 #endif
  71
  72 INLINE_HEADER void
  73 recordMutableGen(StgClosure *p, nat gen_no)
  74 {
  75     bdescr *bd;
  76
  77     bd = generations[gen_no].mut_list;
  78     if (bd->free >= bd->start + BLOCK_SIZE_W) {
  79         bdescr *new_bd;
  80         new_bd = allocBlock();
  81         new_bd->link = bd;
  82         bd = new_bd;
  83         generations[gen_no].mut_list = bd;
  84     }
  85     *bd->free++ = (StgWord)p;
  86
  87 }
  88
  89 INLINE_HEADER void
  90 recordMutableGenLock(StgClosure *p, nat gen_no)
  91 {
  92     ACQUIRE_SM_LOCK;
  93     recordMutableGen(p,gen_no);
  94     RELEASE_SM_LOCK;
  95 }
  96
  97 INLINE_HEADER void
  98 recordMutable(StgClosure *p)
  99 {
 100     bdescr *bd;
 101     ASSERT(closure_MUTABLE(p));
 102     bd = Bdescr((P_)p);
 103     if (bd->gen_no > 0) recordMutableGen(p, bd->gen_no);
 104 }
 105
 106 INLINE_HEADER void
 107 recordMutableLock(StgClosure *p)
 108 {
 109     ACQUIRE_SM_LOCK;
 110     recordMutable(p);
 111     RELEASE_SM_LOCK;
 112 }
 113
 114 /* -----------------------------------------------------------------------------
 115    The write barrier for MVARs
 116    -------------------------------------------------------------------------- */
 117
 118 void dirty_MVAR(StgRegTable *reg, StgClosure *p);
 119
 120 /* -----------------------------------------------------------------------------
 121    Nursery manipulation
 122    -------------------------------------------------------------------------- */
 123
 124 extern nursery *nurseries;
 125
 126 void     resetNurseries       ( void );
 127 lnat     clearNurseries       ( void );
 128 void     resizeNurseries      ( nat blocks );
 129 void     resizeNurseriesFixed ( nat blocks );
 130 lnat     countNurseryBlocks   ( void );
 131
 132 /* -----------------------------------------------------------------------------
 133    Stats 'n' DEBUG stuff
 134    -------------------------------------------------------------------------- */
 135
 136 lnat    calcAllocated  (rtsBool count_nurseries);
 137 lnat    calcLiveBlocks (void);
 138 lnat    calcLiveWords  (void);
 139 lnat    countOccupied  (bdescr *bd);
 140 lnat    calcNeeded     (void);
 141
 142 /* ----------------------------------------------------------------------------
 143    Storage manager internal APIs and globals
 144    ------------------------------------------------------------------------- */
 145
 146 extern bdescr *exec_block;
 147
 148 #define END_OF_STATIC_LIST ((StgClosure*)1)
 149
 150 void move_STACK  (StgStack *src, StgStack *dest);
 151
 152 extern StgClosure * caf_list;
 153 extern StgClosure * revertible_caf_list;
 154
 155 #include "EndPrivate.h"
 156
 157 #endif /* SM_STORAGE_H */