ghc/rts/Sanity.c

   1 /* -----------------------------------------------------------------------------
   2  * $Id: Sanity.c,v 1.8 1999/02/02 12:37:14 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
 197     case MVAR:
 198       {
 199         StgMVar *mvar = (StgMVar *)p;
 200         ASSERT(LOOKS_LIKE_PTR(mvar->head));
 201         ASSERT(LOOKS_LIKE_PTR(mvar->tail));
 202         ASSERT(LOOKS_LIKE_PTR(mvar->value));
 203         return sizeofW(StgMVar);
 204       }
 205
 206     case THUNK:
 207     case THUNK_1_0:
 208     case THUNK_0_1:
 209     case THUNK_1_1:
 210     case THUNK_0_2:
 211     case THUNK_2_0:
 212       {
 213         nat i;
 214         for (i = 0; i < info->layout.payload.ptrs; i++) {
 215           ASSERT(LOOKS_LIKE_PTR(payloadPtr(p,i)));
 216         }
 217         return stg_max(sizeW_fromITBL(info), sizeofW(StgHeader) + MIN_UPD_SIZE);
 218       }
 219
 220     case FUN:
 221     case FUN_1_0:
 222     case FUN_0_1:
 223     case FUN_1_1:
 224     case FUN_0_2:
 225     case FUN_2_0:
 226     case CONSTR:
 227     case CONSTR_1_0:
 228     case CONSTR_0_1:
 229     case CONSTR_1_1:
 230     case CONSTR_0_2:
 231     case CONSTR_2_0:
 232     case IND_PERM:
 233     case IND_OLDGEN:
 234     case IND_OLDGEN_PERM:
 235     case CAF_UNENTERED:
 236     case CAF_ENTERED:
 237     case CAF_BLACKHOLE:
 238     case BLACKHOLE:
 239     case BLACKHOLE_BQ:
 240     case FOREIGN:
 241     case STABLE_NAME:
 242     case MUT_VAR:
 243     case CONSTR_INTLIKE:
 244     case CONSTR_CHARLIKE:
 245     case CONSTR_STATIC:
 246     case CONSTR_NOCAF_STATIC:
 247     case THUNK_STATIC:
 248     case FUN_STATIC:
 249     case IND_STATIC:
 250         {
 251             nat i;
 252             for (i = 0; i < info->layout.payload.ptrs; i++) {
 253                 ASSERT(LOOKS_LIKE_PTR(payloadPtr(p,i)));
 254             }
 255             return sizeW_fromITBL(info);
 256         }
 257
 258     case WEAK:
 259       /* deal with these specially - the info table isn't
 260        * representative of the actual layout.
 261        */
 262       { StgWeak *w = (StgWeak *)p;
 263         ASSERT(LOOKS_LIKE_PTR(w->key));
 264         ASSERT(LOOKS_LIKE_PTR(w->value));
 265         ASSERT(LOOKS_LIKE_PTR(w->finaliser));
 266         if (w->link) {
 267           ASSERT(LOOKS_LIKE_PTR(w->link));
 268         }
 269         return sizeW_fromITBL(info);
 270       }
 271
 272     case THUNK_SELECTOR:
 273             ASSERT(LOOKS_LIKE_PTR(stgCast(StgSelector*,p)->selectee));
 274             return sizeofW(StgHeader) + MIN_UPD_SIZE;
 275
 276     case IND:
 277         {
 278             /* we don't expect to see any of these after GC
 279              * but they might appear during execution
 280              */
 281             P_ q;
 282             StgInd *ind = stgCast(StgInd*,p);
 283             ASSERT(LOOKS_LIKE_PTR(ind->indirectee));
 284             q = (P_)p + sizeofW(StgInd);
 285             while (!*q) { q++; }; /* skip padding words (see GC.c: evacuate())*/
 286             return q - (P_)p;
 287         }
 288
 289     case RET_BCO:
 290     case RET_SMALL:
 291     case RET_VEC_SMALL:
 292     case RET_BIG:
 293     case RET_VEC_BIG:
 294     case RET_DYN:
 295     case UPDATE_FRAME:
 296     case STOP_FRAME:
 297     case CATCH_FRAME:
 298     case SEQ_FRAME:
 299             barf("checkClosure: stack frame");
 300
 301     case AP_UPD: /* we can treat this as being the same as a PAP */
 302     case PAP:
 303         {
 304             StgPAP *pap = stgCast(StgPAP*,p);
 305             ASSERT(LOOKS_LIKE_PTR(pap->fun));
 306             checkStackChunk((StgPtr)pap->payload,
 307                             (StgPtr)pap->payload + pap->n_args
 308                             );
 309             return pap_sizeW(pap);
 310         }
 311
 312     case ARR_WORDS:
 313     case MUT_ARR_WORDS:
 314             return arr_words_sizeW(stgCast(StgArrWords*,p));
 315
 316     case MUT_ARR_PTRS:
 317     case MUT_ARR_PTRS_FROZEN:
 318         {
 319             StgMutArrPtrs* a = stgCast(StgMutArrPtrs*,p);
 320             nat i;
 321             for (i = 0; i < a->ptrs; i++) {
 322                 ASSERT(LOOKS_LIKE_PTR(a->payload[i]));
 323             }
 324             return mut_arr_ptrs_sizeW(a);
 325         }
 326
 327     case TSO:
 328         checkTSO((StgTSO *)p);
 329         return tso_sizeW((StgTSO *)p);
 330
 331     case BLOCKED_FETCH:
 332     case FETCH_ME:
 333     case EVACUATED:
 334             barf("checkClosure: unimplemented/strange closure type");
 335     default:
 336             barf("checkClosure");
 337     }
 338 #undef LOOKS_LIKE_PTR
 339 }
 340
 341 /* -----------------------------------------------------------------------------
 342    Check Heap Sanity
 343
 344    After garbage collection, the live heap is in a state where we can
 345    run through and check that all the pointers point to the right
 346    place.  This function starts at a given position and sanity-checks
 347    all the objects in the remainder of the chain.
 348    -------------------------------------------------------------------------- */
 349
 350 extern void
 351 checkHeap(bdescr *bd, StgPtr start)
 352 {
 353     StgPtr p;
 354
 355     if (start == NULL) {
 356       p = bd->start;
 357     } else {
 358       p = start;
 359     }
 360
 361     while (bd != NULL) {
 362       while (p < bd->free) {
 363         nat size = checkClosure(stgCast(StgClosure*,p));
 364         /* This is the smallest size of closure that can live in the heap. */
 365         ASSERT( size >= MIN_NONUPD_SIZE + sizeofW(StgHeader) );
 366         p += size;
 367
 368         /* skip over slop */
 369         while (p < bd->free &&
 370                (*p == 0 || !LOOKS_LIKE_GHC_INFO(*p))) { p++; }
 371       }
 372       bd = bd->link;
 373       if (bd != NULL) {
 374         p = bd->start;
 375       }
 376     }
 377 }
 378
 379 extern void
 380 checkChain(bdescr *bd)
 381 {
 382   while (bd != NULL) {
 383     checkClosure((StgClosure *)bd->start);
 384     bd = bd->link;
 385   }
 386 }
 387
 388 /* check stack - making sure that update frames are linked correctly */
 389 void
 390 checkStack(StgPtr sp, StgPtr stack_end, StgUpdateFrame* su )
 391 {
 392     /* check everything down to the first update frame */
 393     checkStackChunk( sp, stgCast(StgPtr,su) );
 394     while ( stgCast(StgPtr,su) < stack_end) {
 395         sp = stgCast(StgPtr,su);
 396         switch (get_itbl(su)->type) {
 397         case UPDATE_FRAME:
 398                 su = su->link;
 399                 break;
 400         case SEQ_FRAME:
 401                 su = stgCast(StgSeqFrame*,su)->link;
 402                 break;
 403         case CATCH_FRAME:
 404                 su = stgCast(StgCatchFrame*,su)->link;
 405                 break;
 406         case STOP_FRAME:
 407                 /* not quite: ASSERT(stgCast(StgPtr,su) == stack_end); */
 408                 return;
 409         default:
 410                 barf("checkStack: weird record found on update frame list.");
 411         }
 412         checkStackChunk( sp, stgCast(StgPtr,su) );
 413     }
 414     ASSERT(stgCast(StgPtr,su) == stack_end);
 415 }
 416
 417 extern void
 418 checkTSO(StgTSO *tso)
 419 {
 420     StgPtr sp = tso->sp;
 421     StgPtr stack = tso->stack;
 422     StgUpdateFrame* su = tso->su;
 423     StgOffset stack_size = tso->stack_size;
 424     StgPtr stack_end = stack + stack_size;
 425
 426     if (tso->whatNext == ThreadComplete ||  tso->whatNext == ThreadKilled) {
 427       /* The garbage collector doesn't bother following any pointers
 428        * from dead threads, so don't check sanity here.
 429        */
 430       return;
 431     }
 432
 433     ASSERT(stack <= sp && sp < stack_end);
 434     ASSERT(sp <= stgCast(StgPtr,su));
 435
 436     checkStack(sp, stack_end, su);
 437 }
 438
 439 /* -----------------------------------------------------------------------------
 440    Check Blackhole Sanity
 441
 442    Test whether an object is already on the update list.
 443    It isn't necessarily an rts error if it is - it might be a programming
 444    error.
 445
 446    Future versions might be able to test for a blackhole without traversing
 447    the update frame list.
 448
 449    -------------------------------------------------------------------------- */
 450 rtsBool isBlackhole( StgTSO* tso, StgClosure* p )
 451 {
 452   StgUpdateFrame* su = tso->su;
 453   do {
 454     switch (get_itbl(su)->type) {
 455     case UPDATE_FRAME:
 456       if (su->updatee == p) {
 457         return rtsTrue;
 458       } else {
 459         su = su->link;
 460       }
 461       break;
 462     case SEQ_FRAME:
 463       su = stgCast(StgSeqFrame*,su)->link;
 464       break;
 465     case CATCH_FRAME:
 466       su = stgCast(StgCatchFrame*,su)->link;
 467       break;
 468     case STOP_FRAME:
 469       return rtsFalse;
 470     default:
 471       barf("isBlackhole: weird record found on update frame list.");
 472     }
 473   } while (1);
 474 }
 475
 476 #endif /* DEBUG */