src/lj_gc.h

   1 /*
   2 ** Garbage collector.
   3 ** Copyright (C) 2005-2011 Mike Pall. See Copyright Notice in luajit.h
   4 */
   5
   6 #ifndef _LJ_GC_H
   7 #define _LJ_GC_H
   8
   9 #include "lj_obj.h"
  10
  11 /* Garbage collector states. Order matters. */
  12 enum {
  13   GCSpause, GCSpropagate, GCSatomic, GCSsweepstring, GCSsweep, GCSfinalize
  14 };
  15
  16 /* Bitmasks for marked field of GCobj. */
  17 #define LJ_GC_WHITE0    0x01
  18 #define LJ_GC_WHITE1    0x02
  19 #define LJ_GC_BLACK     0x04
  20 #define LJ_GC_FINALIZED 0x08
  21 #define LJ_GC_WEAKKEY   0x08
  22 #define LJ_GC_WEAKVAL   0x10
  23 #define LJ_GC_CDATA_FIN 0x10
  24 #define LJ_GC_FIXED     0x20
  25 #define LJ_GC_SFIXED    0x40
  26
  27 #define LJ_GC_WHITES    (LJ_GC_WHITE0 | LJ_GC_WHITE1)
  28 #define LJ_GC_COLORS    (LJ_GC_WHITES | LJ_GC_BLACK)
  29 #define LJ_GC_WEAK      (LJ_GC_WEAKKEY | LJ_GC_WEAKVAL)
  30
  31 /* Macros to test and set GCobj colors. */
  32 #define iswhite(x)      ((x)->gch.marked & LJ_GC_WHITES)
  33 #define isblack(x)      ((x)->gch.marked & LJ_GC_BLACK)
  34 #define isgray(x)       (!((x)->gch.marked & (LJ_GC_BLACK|LJ_GC_WHITES)))
  35 #define tviswhite(x)    (tvisgcv(x) && iswhite(gcV(x)))
  36 #define otherwhite(g)   (g->gc.currentwhite ^ LJ_GC_WHITES)
  37 #define isdead(g, v)    ((v)->gch.marked & otherwhite(g) & LJ_GC_WHITES)
  38
  39 #define curwhite(g)     ((g)->gc.currentwhite & LJ_GC_WHITES)
  40 #define newwhite(g, x)  (obj2gco(x)->gch.marked = (uint8_t)curwhite(g))
  41 #define makewhite(g, x) \
  42   ((x)->gch.marked = ((x)->gch.marked & (uint8_t)~LJ_GC_COLORS) | curwhite(g))
  43 #define flipwhite(x)    ((x)->gch.marked ^= LJ_GC_WHITES)
  44 #define black2gray(x)   ((x)->gch.marked &= (uint8_t)~LJ_GC_BLACK)
  45 #define fixstring(s)    ((s)->marked |= LJ_GC_FIXED)
  46
  47 /* Collector. */
  48 LJ_FUNC size_t lj_gc_separateudata(global_State *g, int all);
  49 LJ_FUNC void lj_gc_finalize_udata(lua_State *L);
  50 #if LJ_HASFFI
  51 LJ_FUNC void lj_gc_finalize_cdata(lua_State *L);
  52 #else
  53 #define lj_gc_finalize_cdata(L)         UNUSED(L)
  54 #endif
  55 LJ_FUNC void lj_gc_freeall(global_State *g);
  56 LJ_FUNCA int LJ_FASTCALL lj_gc_step(lua_State *L);
  57 LJ_FUNCA void LJ_FASTCALL lj_gc_step_fixtop(lua_State *L);
  58 #if LJ_HASJIT
  59 LJ_FUNC int LJ_FASTCALL lj_gc_step_jit(global_State *g, MSize steps);
  60 #endif
  61 LJ_FUNC void lj_gc_fullgc(lua_State *L);
  62
  63 /* GC check: drive collector forward if the GC threshold has been reached. */
  64 #define lj_gc_check(L) \
  65   { if (LJ_UNLIKELY(G(L)->gc.total >= G(L)->gc.threshold)) \
  66       lj_gc_step(L); }
  67 #define lj_gc_check_fixtop(L) \
  68   { if (LJ_UNLIKELY(G(L)->gc.total >= G(L)->gc.threshold)) \
  69       lj_gc_step_fixtop(L); }
  70
  71 /* Write barriers. */
  72 LJ_FUNC void lj_gc_barrierf(global_State *g, GCobj *o, GCobj *v);
  73 LJ_FUNCA void LJ_FASTCALL lj_gc_barrieruv(global_State *g, TValue *tv);
  74 LJ_FUNC void lj_gc_closeuv(global_State *g, GCupval *uv);
  75 #if LJ_HASJIT
  76 LJ_FUNC void lj_gc_barriertrace(global_State *g, uint32_t traceno);
  77 #endif
  78
  79 /* Move the GC propagation frontier back for tables (make it gray again). */
  80 static LJ_AINLINE void lj_gc_barrierback(global_State *g, GCtab *t)
  81 {
  82   GCobj *o = obj2gco(t);
  83   lua_assert(isblack(o) && !isdead(g, o));
  84   lua_assert(g->gc.state != GCSfinalize && g->gc.state != GCSpause);
  85   black2gray(o);
  86   setgcrefr(t->gclist, g->gc.grayagain);
  87   setgcref(g->gc.grayagain, o);
  88 }
  89
  90 /* Barrier for stores to table objects. TValue and GCobj variant. */
  91 #define lj_gc_anybarriert(L, t)  \
  92   { if (LJ_UNLIKELY(isblack(obj2gco(t)))) lj_gc_barrierback(G(L), (t)); }
  93 #define lj_gc_barriert(L, t, tv) \
  94   { if (tviswhite(tv) && isblack(obj2gco(t))) \
  95       lj_gc_barrierback(G(L), (t)); }
  96 #define lj_gc_objbarriert(L, t, o)  \
  97   { if (iswhite(obj2gco(o)) && isblack(obj2gco(t))) \
  98       lj_gc_barrierback(G(L), (t)); }
  99
 100 /* Barrier for stores to any other object. TValue and GCobj variant. */
 101 #define lj_gc_barrier(L, p, tv) \
 102   { if (tviswhite(tv) && isblack(obj2gco(p))) \
 103       lj_gc_barrierf(G(L), obj2gco(p), gcV(tv)); }
 104 #define lj_gc_objbarrier(L, p, o) \
 105   { if (iswhite(obj2gco(o)) && isblack(obj2gco(p))) \
 106       lj_gc_barrierf(G(L), obj2gco(p), obj2gco(o)); }
 107
 108 /* Allocator. */
 109 LJ_FUNC void *lj_mem_realloc(lua_State *L, void *p, MSize osz, MSize nsz);
 110 LJ_FUNC void * LJ_FASTCALL lj_mem_newgco(lua_State *L, MSize size);
 111 LJ_FUNC void *lj_mem_grow(lua_State *L, void *p,
 112                           MSize *szp, MSize lim, MSize esz);
 113
 114 #define lj_mem_new(L, s)        lj_mem_realloc(L, NULL, 0, (s))
 115
 116 static LJ_AINLINE void lj_mem_free(global_State *g, void *p, size_t osize)
 117 {
 118   g->gc.total -= (MSize)osize;
 119   g->allocf(g->allocd, p, osize, 0);
 120 }
 121
 122 #define lj_mem_newvec(L, n, t)  ((t *)lj_mem_new(L, (MSize)((n)*sizeof(t))))
 123 #define lj_mem_reallocvec(L, p, on, n, t) \
 124   ((p) = (t *)lj_mem_realloc(L, p, (on)*sizeof(t), (MSize)((n)*sizeof(t))))
 125 #define lj_mem_growvec(L, p, n, m, t) \
 126   ((p) = (t *)lj_mem_grow(L, (p), &(n), (m), (MSize)sizeof(t)))
 127 #define lj_mem_freevec(g, p, n, t)      lj_mem_free(g, (p), (n)*sizeof(t))
 128
 129 #define lj_mem_newobj(L, t)     ((t *)lj_mem_newgco(L, sizeof(t)))
 130 #define lj_mem_newt(L, s, t)    ((t *)lj_mem_new(L, (s)))
 131 #define lj_mem_freet(g, p)      lj_mem_free(g, (p), sizeof(*(p)))
 132
 133 #endif