ghc/rts/Sanity.c

   1 /* -----------------------------------------------------------------------------
   2  * $Id: Sanity.c,v 1.4 1999/01/15 17:57:10 simonm Exp $
   3  *
   4  * Sanity checking code for the heap and stack.
   5  *
   6  * Used when debugging: check that the stack looks reasonable.
   7  *
   8  *    - All things that are supposed to be pointers look like pointers.
   9  *
  10  *    - Objects in text space are marked as static closures, those
  11  *      in the heap are dynamic.
  12  *
  13  * ---------------------------------------------------------------------------*/
  14
  15 #include "Rts.h"
  16
  17 #ifdef DEBUG
  18
  19 #include "RtsFlags.h"
  20 #include "RtsUtils.h"
  21 #include "BlockAlloc.h"
  22 #include "Sanity.h"
  23
  24 #define LOOKS_LIKE_PTR(r) \
  25   (IS_DATA_PTR(r) || ((IS_USER_PTR(r) && Bdescr((P_)r)->free != (void *)-1)))
  26
  27 /* -----------------------------------------------------------------------------
  28    Check stack sanity
  29    -------------------------------------------------------------------------- */
  30
  31 StgOffset checkStackClosure( StgClosure* c );
  32
  33 StgOffset checkStackObject( StgPtr sp );
  34
  35 void      checkStackChunk( StgPtr sp, StgPtr stack_end );
  36
  37 static StgOffset checkSmallBitmap(  StgPtr payload, StgNat32 bitmap );
  38
  39 static StgOffset checkLargeBitmap( StgPtr payload,
  40                                    StgLargeBitmap* large_bitmap );
  41
  42 void checkClosureShallow( StgClosure* p );
  43
  44 static StgOffset
  45 checkSmallBitmap( StgPtr payload, StgNat32 bitmap )
  46 {
  47     StgOffset i;
  48
  49     i = 0;
  50     for(; bitmap != 0; ++i, bitmap >>= 1 ) {
  51         if ((bitmap & 1) == 0) {
  52             checkClosure(stgCast(StgClosure*,payload[i]));
  53         }
  54     }
  55     return i;
  56 }
  57
  58
  59 static StgOffset
  60 checkLargeBitmap( StgPtr payload, StgLargeBitmap* large_bitmap )
  61 {
  62     StgNat32 bmp;
  63     StgOffset i;
  64
  65     i = 0;
  66     for (bmp=0; bmp<large_bitmap->size; bmp++) {
  67         StgNat32 bitmap = large_bitmap->bitmap[bmp];
  68         for(; bitmap != 0; ++i, bitmap >>= 1 ) {
  69             if ((bitmap & 1) == 0) {
  70                 checkClosure(stgCast(StgClosure*,payload[i]));
  71             }
  72         }
  73     }
  74     return i;
  75 }
  76
  77 StgOffset
  78 checkStackClosure( StgClosure* c )
  79 {
  80     const StgInfoTable* info = get_itbl(c);
  81
  82     /* All activation records have 'bitmap' style layout info. */
  83     switch (info->type) {
  84     case RET_DYN: /* Dynamic bitmap: the mask is stored on the stack */
  85         {
  86             StgRetDyn* r = stgCast(StgRetDyn*,c);
  87             return sizeofW(StgRetDyn) +
  88                    checkSmallBitmap(r->payload,r->liveness);
  89         }
  90     case RET_BCO: /* small bitmap (<= 32 entries) */
  91     case RET_SMALL:
  92     case RET_VEC_SMALL:
  93     case UPDATE_FRAME:
  94     case CATCH_FRAME:
  95     case STOP_FRAME:
  96     case SEQ_FRAME:
  97             return sizeofW(StgClosure) +
  98                    checkSmallBitmap((StgPtr)c->payload,info->layout.bitmap);
  99     case RET_BIG: /* large bitmap (> 32 entries) */
 100     case RET_VEC_BIG:
 101             return sizeofW(StgClosure) +
 102                    checkLargeBitmap((StgPtr)c->payload,
 103                                     info->layout.large_bitmap);
 104     case FUN:
 105     case FUN_STATIC: /* probably a slow-entry point return address: */
 106             return 1;
 107     default:
 108             /* if none of the above, maybe it's a closure which looks a
 109              * little like an infotable
 110              */
 111             checkClosureShallow(*stgCast(StgClosure**,c));
 112             return 1;
 113             /* barf("checkStackClosure: weird activation record found on stack (%p).",c); */
 114     }
 115 }
 116
 117 /*
 118  * check that it looks like a valid closure - without checking its payload
 119  * used to avoid recursion between checking PAPs and checking stack
 120  * chunks.
 121  */
 122
 123 void
 124 checkClosureShallow( StgClosure* p )
 125 {
 126     ASSERT(LOOKS_LIKE_GHC_INFO(p->header.info));
 127
 128     /* Is it a static closure (i.e. in the data segment)? */
 129     if (LOOKS_LIKE_STATIC(p)) {
 130         ASSERT(closure_STATIC(p));
 131     } else {
 132         ASSERT(!closure_STATIC(p));
 133         ASSERT(LOOKS_LIKE_PTR(p));
 134     }
 135 }
 136
 137 /* check an individual stack object */
 138 StgOffset
 139 checkStackObject( StgPtr sp )
 140 {
 141     if (IS_ARG_TAG(*sp)) {
 142         /* Tagged words might be "stubbed" pointers, so there's no
 143          * point checking to see whether they look like pointers or
 144          * not (some of them will).
 145          */
 146         return ARG_SIZE(*sp) + 1;
 147     } else if (LOOKS_LIKE_GHC_INFO(*stgCast(StgPtr*,sp))) {
 148         return checkStackClosure(stgCast(StgClosure*,sp));
 149     } else { /* must be an untagged closure pointer in the stack */
 150         checkClosureShallow(*stgCast(StgClosure**,sp));
 151         return 1;
 152     }
 153 }
 154
 155 /* check sections of stack between update frames */
 156 void
 157 checkStackChunk( StgPtr sp, StgPtr stack_end )
 158 {
 159     StgPtr p;
 160
 161     p = sp;
 162     while (p < stack_end) {
 163         p += checkStackObject( p );
 164     }
 165     ASSERT( p == stack_end );
 166 }
 167
 168 StgOffset
 169 checkClosure( StgClosure* p )
 170 {
 171     const StgInfoTable *info;
 172
 173 #ifndef INTERPRETER
 174     ASSERT(LOOKS_LIKE_GHC_INFO(p->header.info));
 175 #endif
 176
 177     /* Is it a static closure (i.e. in the data segment)? */
 178     if (LOOKS_LIKE_STATIC(p)) {
 179         ASSERT(closure_STATIC(p));
 180     } else {
 181         ASSERT(!closure_STATIC(p));
 182         ASSERT(LOOKS_LIKE_PTR(p));
 183     }
 184
 185     info = get_itbl(p);
 186     switch (info->type) {
 187     case BCO:
 188         {
 189             StgBCO* bco = stgCast(StgBCO*,p);
 190             nat i;
 191             for(i=0; i < bco->n_ptrs; ++i) {
 192                 ASSERT(LOOKS_LIKE_PTR(bcoConstPtr(bco,i)));
 193             }
 194             return bco_sizeW(bco);
 195         }
 196     case FUN:
 197     case THUNK:
 198     case CONSTR:
 199     case IND_PERM:
 200     case IND_OLDGEN:
 201     case IND_OLDGEN_PERM:
 202     case CAF_UNENTERED:
 203     case CAF_ENTERED:
 204     case CAF_BLACKHOLE:
 205     case BLACKHOLE:
 206     case BLACKHOLE_BQ:
 207     case FOREIGN:
 208     case MVAR:
 209     case MUT_VAR:
 210     case CONSTR_INTLIKE:
 211     case CONSTR_CHARLIKE:
 212     case CONSTR_STATIC:
 213     case CONSTR_NOCAF_STATIC:
 214     case THUNK_STATIC:
 215     case FUN_STATIC:
 216     case IND_STATIC:
 217         {
 218             nat i;
 219             for (i = 0; i < info->layout.payload.ptrs; i++) {
 220                 ASSERT(LOOKS_LIKE_PTR(payloadPtr(p,i)));
 221             }
 222             return sizeW_fromITBL(info);
 223         }
 224
 225     case WEAK:
 226       /* deal with these specially - the info table isn't
 227        * representative of the actual layout.
 228        */
 229       { StgWeak *w = (StgWeak *)p;
 230         ASSERT(LOOKS_LIKE_PTR(w->key));
 231         ASSERT(LOOKS_LIKE_PTR(w->value));
 232         ASSERT(LOOKS_LIKE_PTR(w->finaliser));
 233         if (w->link) {
 234           ASSERT(LOOKS_LIKE_PTR(w->link));
 235         }
 236         return sizeW_fromITBL(info);
 237       }
 238
 239     case THUNK_SELECTOR:
 240             ASSERT(LOOKS_LIKE_PTR(stgCast(StgSelector*,p)->selectee));
 241             return sizeofW(StgHeader) + MIN_UPD_SIZE;
 242
 243     case IND:
 244         {
 245             /* we don't expect to see any of these after GC
 246              * but they might appear during execution
 247              */
 248             P_ q;
 249             StgInd *ind = stgCast(StgInd*,p);
 250             ASSERT(LOOKS_LIKE_PTR(ind->indirectee));
 251             q = (P_)p + sizeofW(StgInd);
 252             while (!*q) { q++; }; /* skip padding words (see GC.c: evacuate())*/
 253             return q - (P_)p;
 254         }
 255
 256     case RET_BCO:
 257     case RET_SMALL:
 258     case RET_VEC_SMALL:
 259     case RET_BIG:
 260     case RET_VEC_BIG:
 261     case RET_DYN:
 262     case UPDATE_FRAME:
 263     case STOP_FRAME:
 264     case CATCH_FRAME:
 265     case SEQ_FRAME:
 266             barf("checkClosure: stack frame");
 267
 268     case AP_UPD: /* we can treat this as being the same as a PAP */
 269     case PAP:
 270         {
 271             StgPAP *pap = stgCast(StgPAP*,p);
 272             ASSERT(LOOKS_LIKE_PTR(pap->fun));
 273             checkStackChunk((StgPtr)pap->payload,
 274                             (StgPtr)pap->payload + pap->n_args
 275                             );
 276             return pap_sizeW(pap);
 277         }
 278
 279     case ARR_WORDS:
 280     case MUT_ARR_WORDS:
 281             return arr_words_sizeW(stgCast(StgArrWords*,p));
 282
 283     case MUT_ARR_PTRS:
 284     case MUT_ARR_PTRS_FROZEN:
 285         {
 286             StgMutArrPtrs* a = stgCast(StgMutArrPtrs*,p);
 287             nat i;
 288             for (i = 0; i < a->ptrs; i++) {
 289                 ASSERT(LOOKS_LIKE_PTR(a->payload[i]));
 290             }
 291             return mut_arr_ptrs_sizeW(a);
 292         }
 293
 294     case TSO:
 295         checkTSO((StgTSO *)p);
 296         return tso_sizeW((StgTSO *)p);
 297
 298     case BLOCKED_FETCH:
 299     case FETCH_ME:
 300     case EVACUATED:
 301             barf("checkClosure: unimplemented/strange closure type");
 302     default:
 303             barf("checkClosure");
 304     }
 305 #undef LOOKS_LIKE_PTR
 306 }
 307
 308 /* -----------------------------------------------------------------------------
 309    Check Heap Sanity
 310
 311    After garbage collection, the live heap is in a state where we can
 312    run through and check that all the pointers point to the right
 313    place.  This function starts at a given position and sanity-checks
 314    all the objects in the remainder of the chain.
 315    -------------------------------------------------------------------------- */
 316
 317 extern void
 318 checkHeap(bdescr *bd, StgPtr start)
 319 {
 320     StgPtr p;
 321
 322     if (start == NULL) {
 323       p = bd->start;
 324     } else {
 325       p = start;
 326     }
 327
 328     while (bd != NULL) {
 329       while (p < bd->free) {
 330         nat size = checkClosure(stgCast(StgClosure*,p));
 331         /* This is the smallest size of closure that can live in the heap. */
 332         ASSERT( size >= MIN_NONUPD_SIZE + sizeofW(StgHeader) );
 333         p += size;
 334         while (*p == 0) { p++; } /* skip over slop */
 335       }
 336       bd = bd->link;
 337       if (bd != NULL) {
 338         p = bd->start;
 339       }
 340     }
 341 }
 342
 343 extern void
 344 checkChain(bdescr *bd)
 345 {
 346   while (bd != NULL) {
 347     checkClosure((StgClosure *)bd->start);
 348     bd = bd->link;
 349   }
 350 }
 351
 352 /* check stack - making sure that update frames are linked correctly */
 353 void
 354 checkStack(StgPtr sp, StgPtr stack_end, StgUpdateFrame* su )
 355 {
 356     /* check everything down to the first update frame */
 357     checkStackChunk( sp, stgCast(StgPtr,su) );
 358     while ( stgCast(StgPtr,su) < stack_end) {
 359         sp = stgCast(StgPtr,su);
 360         switch (get_itbl(su)->type) {
 361         case UPDATE_FRAME:
 362                 su = su->link;
 363                 break;
 364         case SEQ_FRAME:
 365                 su = stgCast(StgSeqFrame*,su)->link;
 366                 break;
 367         case CATCH_FRAME:
 368                 su = stgCast(StgCatchFrame*,su)->link;
 369                 break;
 370         case STOP_FRAME:
 371                 /* not quite: ASSERT(stgCast(StgPtr,su) == stack_end); */
 372                 return;
 373         default:
 374                 barf("checkStack: weird record found on update frame list.");
 375         }
 376         checkStackChunk( sp, stgCast(StgPtr,su) );
 377     }
 378     ASSERT(stgCast(StgPtr,su) == stack_end);
 379 }
 380
 381 extern void
 382 checkTSO(StgTSO *tso)
 383 {
 384     StgPtr sp = tso->sp;
 385     StgPtr stack = tso->stack;
 386     StgUpdateFrame* su = tso->su;
 387     StgOffset stack_size = tso->stack_size;
 388     StgPtr stack_end = stack + stack_size;
 389
 390     if (tso->whatNext == ThreadComplete ||  tso->whatNext == ThreadKilled) {
 391       /* The garbage collector doesn't bother following any pointers
 392        * from dead threads, so don't check sanity here.
 393        */
 394       return;
 395     }
 396
 397     ASSERT(stack <= sp && sp < stack_end);
 398     ASSERT(sp <= stgCast(StgPtr,su));
 399
 400     checkStack(sp, stack_end, su);
 401 }
 402
 403 /* -----------------------------------------------------------------------------
 404    Check Blackhole Sanity
 405
 406    Test whether an object is already on the update list.
 407    It isn't necessarily an rts error if it is - it might be a programming
 408    error.
 409
 410    Future versions might be able to test for a blackhole without traversing
 411    the update frame list.
 412
 413    -------------------------------------------------------------------------- */
 414 rtsBool isBlackhole( StgTSO* tso, StgClosure* p )
 415 {
 416   StgUpdateFrame* su = tso->su;
 417   do {
 418     switch (get_itbl(su)->type) {
 419     case UPDATE_FRAME:
 420       if (su->updatee == p) {
 421         return rtsTrue;
 422       } else {
 423         su = su->link;
 424       }
 425       break;
 426     case SEQ_FRAME:
 427       su = stgCast(StgSeqFrame*,su)->link;
 428       break;
 429     case CATCH_FRAME:
 430       su = stgCast(StgCatchFrame*,su)->link;
 431       break;
 432     case STOP_FRAME:
 433       return rtsFalse;
 434     default:
 435       barf("isBlackhole: weird record found on update frame list.");
 436     }
 437   } while (1);
 438 }
 439
 440 #endif /* DEBUG */