[e16e8f2] | 1 | /* |
---|
| 2 | ** $Id: lgc.c,v 2.140.1.2 2013/04/26 18:22:05 roberto Exp $ |
---|
| 3 | ** Garbage Collector |
---|
| 4 | ** See Copyright Notice in lua.h |
---|
| 5 | */ |
---|
| 6 | |
---|
| 7 | #include <string.h> |
---|
| 8 | |
---|
| 9 | #define lgc_c |
---|
| 10 | #define LUA_CORE |
---|
| 11 | |
---|
| 12 | #include "lua.h" |
---|
| 13 | |
---|
| 14 | #include "ldebug.h" |
---|
| 15 | #include "ldo.h" |
---|
| 16 | #include "lfunc.h" |
---|
| 17 | #include "lgc.h" |
---|
| 18 | #include "lmem.h" |
---|
| 19 | #include "lobject.h" |
---|
| 20 | #include "lstate.h" |
---|
| 21 | #include "lstring.h" |
---|
| 22 | #include "ltable.h" |
---|
| 23 | #include "ltm.h" |
---|
| 24 | |
---|
| 25 | |
---|
| 26 | |
---|
| 27 | /* |
---|
| 28 | ** cost of sweeping one element (the size of a small object divided |
---|
| 29 | ** by some adjust for the sweep speed) |
---|
| 30 | */ |
---|
| 31 | #define GCSWEEPCOST ((sizeof(TString) + 4) / 4) |
---|
| 32 | |
---|
| 33 | /* maximum number of elements to sweep in each single step */ |
---|
| 34 | #define GCSWEEPMAX (cast_int((GCSTEPSIZE / GCSWEEPCOST) / 4)) |
---|
| 35 | |
---|
| 36 | /* maximum number of finalizers to call in each GC step */ |
---|
| 37 | #define GCFINALIZENUM 4 |
---|
| 38 | |
---|
| 39 | |
---|
| 40 | /* |
---|
| 41 | ** macro to adjust 'stepmul': 'stepmul' is actually used like |
---|
| 42 | ** 'stepmul / STEPMULADJ' (value chosen by tests) |
---|
| 43 | */ |
---|
| 44 | #define STEPMULADJ 200 |
---|
| 45 | |
---|
| 46 | |
---|
| 47 | /* |
---|
| 48 | ** macro to adjust 'pause': 'pause' is actually used like |
---|
| 49 | ** 'pause / PAUSEADJ' (value chosen by tests) |
---|
| 50 | */ |
---|
| 51 | #define PAUSEADJ 100 |
---|
| 52 | |
---|
| 53 | |
---|
| 54 | /* |
---|
| 55 | ** 'makewhite' erases all color bits plus the old bit and then |
---|
| 56 | ** sets only the current white bit |
---|
| 57 | */ |
---|
| 58 | #define maskcolors (~(bit2mask(BLACKBIT, OLDBIT) | WHITEBITS)) |
---|
| 59 | #define makewhite(g,x) \ |
---|
| 60 | (gch(x)->marked = cast_byte((gch(x)->marked & maskcolors) | luaC_white(g))) |
---|
| 61 | |
---|
| 62 | #define white2gray(x) resetbits(gch(x)->marked, WHITEBITS) |
---|
| 63 | #define black2gray(x) resetbit(gch(x)->marked, BLACKBIT) |
---|
| 64 | |
---|
| 65 | |
---|
| 66 | #define isfinalized(x) testbit(gch(x)->marked, FINALIZEDBIT) |
---|
| 67 | |
---|
| 68 | #define checkdeadkey(n) lua_assert(!ttisdeadkey(gkey(n)) || ttisnil(gval(n))) |
---|
| 69 | |
---|
| 70 | |
---|
| 71 | #define checkconsistency(obj) \ |
---|
| 72 | lua_longassert(!iscollectable(obj) || righttt(obj)) |
---|
| 73 | |
---|
| 74 | |
---|
| 75 | #define markvalue(g,o) { checkconsistency(o); \ |
---|
| 76 | if (valiswhite(o)) reallymarkobject(g,gcvalue(o)); } |
---|
| 77 | |
---|
| 78 | #define markobject(g,t) { if ((t) && iswhite(obj2gco(t))) \ |
---|
| 79 | reallymarkobject(g, obj2gco(t)); } |
---|
| 80 | |
---|
| 81 | static void reallymarkobject (global_State *g, GCObject *o); |
---|
| 82 | |
---|
| 83 | |
---|
| 84 | /* |
---|
| 85 | ** {====================================================== |
---|
| 86 | ** Generic functions |
---|
| 87 | ** ======================================================= |
---|
| 88 | */ |
---|
| 89 | |
---|
| 90 | |
---|
| 91 | /* |
---|
| 92 | ** one after last element in a hash array |
---|
| 93 | */ |
---|
| 94 | #define gnodelast(h) gnode(h, cast(size_t, sizenode(h))) |
---|
| 95 | |
---|
| 96 | |
---|
| 97 | /* |
---|
| 98 | ** link table 'h' into list pointed by 'p' |
---|
| 99 | */ |
---|
| 100 | #define linktable(h,p) ((h)->gclist = *(p), *(p) = obj2gco(h)) |
---|
| 101 | |
---|
| 102 | |
---|
| 103 | /* |
---|
| 104 | ** if key is not marked, mark its entry as dead (therefore removing it |
---|
| 105 | ** from the table) |
---|
| 106 | */ |
---|
| 107 | static void removeentry (Node *n) { |
---|
| 108 | lua_assert(ttisnil(gval(n))); |
---|
| 109 | if (valiswhite(gkey(n))) |
---|
| 110 | setdeadvalue(gkey(n)); /* unused and unmarked key; remove it */ |
---|
| 111 | } |
---|
| 112 | |
---|
| 113 | |
---|
| 114 | /* |
---|
| 115 | ** tells whether a key or value can be cleared from a weak |
---|
| 116 | ** table. Non-collectable objects are never removed from weak |
---|
| 117 | ** tables. Strings behave as `values', so are never removed too. for |
---|
| 118 | ** other objects: if really collected, cannot keep them; for objects |
---|
| 119 | ** being finalized, keep them in keys, but not in values |
---|
| 120 | */ |
---|
| 121 | static int iscleared (global_State *g, const TValue *o) { |
---|
| 122 | if (!iscollectable(o)) return 0; |
---|
| 123 | else if (ttisstring(o)) { |
---|
| 124 | markobject(g, rawtsvalue(o)); /* strings are `values', so are never weak */ |
---|
| 125 | return 0; |
---|
| 126 | } |
---|
| 127 | else return iswhite(gcvalue(o)); |
---|
| 128 | } |
---|
| 129 | |
---|
| 130 | |
---|
| 131 | /* |
---|
| 132 | ** barrier that moves collector forward, that is, mark the white object |
---|
| 133 | ** being pointed by a black object. |
---|
| 134 | */ |
---|
| 135 | void luaC_barrier_ (lua_State *L, GCObject *o, GCObject *v) { |
---|
| 136 | global_State *g = G(L); |
---|
| 137 | lua_assert(isblack(o) && iswhite(v) && !isdead(g, v) && !isdead(g, o)); |
---|
| 138 | lua_assert(g->gcstate != GCSpause); |
---|
| 139 | lua_assert(gch(o)->tt != LUA_TTABLE); |
---|
| 140 | if (keepinvariantout(g)) /* must keep invariant? */ |
---|
| 141 | reallymarkobject(g, v); /* restore invariant */ |
---|
| 142 | else { /* sweep phase */ |
---|
| 143 | lua_assert(issweepphase(g)); |
---|
| 144 | makewhite(g, o); /* mark main obj. as white to avoid other barriers */ |
---|
| 145 | } |
---|
| 146 | } |
---|
| 147 | |
---|
| 148 | |
---|
| 149 | /* |
---|
| 150 | ** barrier that moves collector backward, that is, mark the black object |
---|
| 151 | ** pointing to a white object as gray again. (Current implementation |
---|
| 152 | ** only works for tables; access to 'gclist' is not uniform across |
---|
| 153 | ** different types.) |
---|
| 154 | */ |
---|
| 155 | void luaC_barrierback_ (lua_State *L, GCObject *o) { |
---|
| 156 | global_State *g = G(L); |
---|
| 157 | lua_assert(isblack(o) && !isdead(g, o) && gch(o)->tt == LUA_TTABLE); |
---|
| 158 | black2gray(o); /* make object gray (again) */ |
---|
| 159 | gco2t(o)->gclist = g->grayagain; |
---|
| 160 | g->grayagain = o; |
---|
| 161 | } |
---|
| 162 | |
---|
| 163 | |
---|
| 164 | /* |
---|
| 165 | ** barrier for prototypes. When creating first closure (cache is |
---|
| 166 | ** NULL), use a forward barrier; this may be the only closure of the |
---|
| 167 | ** prototype (if it is a "regular" function, with a single instance) |
---|
| 168 | ** and the prototype may be big, so it is better to avoid traversing |
---|
| 169 | ** it again. Otherwise, use a backward barrier, to avoid marking all |
---|
| 170 | ** possible instances. |
---|
| 171 | */ |
---|
| 172 | LUAI_FUNC void luaC_barrierproto_ (lua_State *L, Proto *p, Closure *c) { |
---|
| 173 | global_State *g = G(L); |
---|
| 174 | lua_assert(isblack(obj2gco(p))); |
---|
| 175 | if (p->cache == NULL) { /* first time? */ |
---|
| 176 | luaC_objbarrier(L, p, c); |
---|
| 177 | } |
---|
| 178 | else { /* use a backward barrier */ |
---|
| 179 | black2gray(obj2gco(p)); /* make prototype gray (again) */ |
---|
| 180 | p->gclist = g->grayagain; |
---|
| 181 | g->grayagain = obj2gco(p); |
---|
| 182 | } |
---|
| 183 | } |
---|
| 184 | |
---|
| 185 | |
---|
| 186 | /* |
---|
| 187 | ** check color (and invariants) for an upvalue that was closed, |
---|
| 188 | ** i.e., moved into the 'allgc' list |
---|
| 189 | */ |
---|
| 190 | void luaC_checkupvalcolor (global_State *g, UpVal *uv) { |
---|
| 191 | GCObject *o = obj2gco(uv); |
---|
| 192 | lua_assert(!isblack(o)); /* open upvalues are never black */ |
---|
| 193 | if (isgray(o)) { |
---|
| 194 | if (keepinvariant(g)) { |
---|
| 195 | resetoldbit(o); /* see MOVE OLD rule */ |
---|
| 196 | gray2black(o); /* it is being visited now */ |
---|
| 197 | markvalue(g, uv->v); |
---|
| 198 | } |
---|
| 199 | else { |
---|
| 200 | lua_assert(issweepphase(g)); |
---|
| 201 | makewhite(g, o); |
---|
| 202 | } |
---|
| 203 | } |
---|
| 204 | } |
---|
| 205 | |
---|
| 206 | |
---|
| 207 | /* |
---|
| 208 | ** create a new collectable object (with given type and size) and link |
---|
| 209 | ** it to '*list'. 'offset' tells how many bytes to allocate before the |
---|
| 210 | ** object itself (used only by states). |
---|
| 211 | */ |
---|
| 212 | GCObject *luaC_newobj (lua_State *L, int tt, size_t sz, GCObject **list, |
---|
| 213 | int offset) { |
---|
| 214 | global_State *g = G(L); |
---|
| 215 | char *raw = cast(char *, luaM_newobject(L, novariant(tt), sz)); |
---|
| 216 | GCObject *o = obj2gco(raw + offset); |
---|
| 217 | if (list == NULL) |
---|
| 218 | list = &g->allgc; /* standard list for collectable objects */ |
---|
| 219 | gch(o)->marked = luaC_white(g); |
---|
| 220 | gch(o)->tt = tt; |
---|
| 221 | gch(o)->next = *list; |
---|
| 222 | *list = o; |
---|
| 223 | return o; |
---|
| 224 | } |
---|
| 225 | |
---|
| 226 | /* }====================================================== */ |
---|
| 227 | |
---|
| 228 | |
---|
| 229 | |
---|
| 230 | /* |
---|
| 231 | ** {====================================================== |
---|
| 232 | ** Mark functions |
---|
| 233 | ** ======================================================= |
---|
| 234 | */ |
---|
| 235 | |
---|
| 236 | |
---|
| 237 | /* |
---|
| 238 | ** mark an object. Userdata, strings, and closed upvalues are visited |
---|
| 239 | ** and turned black here. Other objects are marked gray and added |
---|
| 240 | ** to appropriate list to be visited (and turned black) later. (Open |
---|
| 241 | ** upvalues are already linked in 'headuv' list.) |
---|
| 242 | */ |
---|
| 243 | static void reallymarkobject (global_State *g, GCObject *o) { |
---|
| 244 | lu_mem size; |
---|
| 245 | white2gray(o); |
---|
| 246 | switch (gch(o)->tt) { |
---|
| 247 | case LUA_TSHRSTR: |
---|
| 248 | case LUA_TLNGSTR: { |
---|
| 249 | size = sizestring(gco2ts(o)); |
---|
| 250 | break; /* nothing else to mark; make it black */ |
---|
| 251 | } |
---|
| 252 | case LUA_TUSERDATA: { |
---|
| 253 | Table *mt = gco2u(o)->metatable; |
---|
| 254 | markobject(g, mt); |
---|
| 255 | markobject(g, gco2u(o)->env); |
---|
| 256 | size = sizeudata(gco2u(o)); |
---|
| 257 | break; |
---|
| 258 | } |
---|
| 259 | case LUA_TUPVAL: { |
---|
| 260 | UpVal *uv = gco2uv(o); |
---|
| 261 | markvalue(g, uv->v); |
---|
| 262 | if (uv->v != &uv->u.value) /* open? */ |
---|
| 263 | return; /* open upvalues remain gray */ |
---|
| 264 | size = sizeof(UpVal); |
---|
| 265 | break; |
---|
| 266 | } |
---|
| 267 | case LUA_TLCL: { |
---|
| 268 | gco2lcl(o)->gclist = g->gray; |
---|
| 269 | g->gray = o; |
---|
| 270 | return; |
---|
| 271 | } |
---|
| 272 | case LUA_TCCL: { |
---|
| 273 | gco2ccl(o)->gclist = g->gray; |
---|
| 274 | g->gray = o; |
---|
| 275 | return; |
---|
| 276 | } |
---|
| 277 | case LUA_TTABLE: { |
---|
| 278 | linktable(gco2t(o), &g->gray); |
---|
| 279 | return; |
---|
| 280 | } |
---|
| 281 | case LUA_TTHREAD: { |
---|
| 282 | gco2th(o)->gclist = g->gray; |
---|
| 283 | g->gray = o; |
---|
| 284 | return; |
---|
| 285 | } |
---|
| 286 | case LUA_TPROTO: { |
---|
| 287 | gco2p(o)->gclist = g->gray; |
---|
| 288 | g->gray = o; |
---|
| 289 | return; |
---|
| 290 | } |
---|
| 291 | default: lua_assert(0); return; |
---|
| 292 | } |
---|
| 293 | gray2black(o); |
---|
| 294 | g->GCmemtrav += size; |
---|
| 295 | } |
---|
| 296 | |
---|
| 297 | |
---|
| 298 | /* |
---|
| 299 | ** mark metamethods for basic types |
---|
| 300 | */ |
---|
| 301 | static void markmt (global_State *g) { |
---|
| 302 | int i; |
---|
| 303 | for (i=0; i < LUA_NUMTAGS; i++) |
---|
| 304 | markobject(g, g->mt[i]); |
---|
| 305 | } |
---|
| 306 | |
---|
| 307 | |
---|
| 308 | /* |
---|
| 309 | ** mark all objects in list of being-finalized |
---|
| 310 | */ |
---|
| 311 | static void markbeingfnz (global_State *g) { |
---|
| 312 | GCObject *o; |
---|
| 313 | for (o = g->tobefnz; o != NULL; o = gch(o)->next) { |
---|
| 314 | makewhite(g, o); |
---|
| 315 | reallymarkobject(g, o); |
---|
| 316 | } |
---|
| 317 | } |
---|
| 318 | |
---|
| 319 | |
---|
| 320 | /* |
---|
| 321 | ** mark all values stored in marked open upvalues. (See comment in |
---|
| 322 | ** 'lstate.h'.) |
---|
| 323 | */ |
---|
| 324 | static void remarkupvals (global_State *g) { |
---|
| 325 | UpVal *uv; |
---|
| 326 | for (uv = g->uvhead.u.l.next; uv != &g->uvhead; uv = uv->u.l.next) { |
---|
| 327 | if (isgray(obj2gco(uv))) |
---|
| 328 | markvalue(g, uv->v); |
---|
| 329 | } |
---|
| 330 | } |
---|
| 331 | |
---|
| 332 | |
---|
| 333 | /* |
---|
| 334 | ** mark root set and reset all gray lists, to start a new |
---|
| 335 | ** incremental (or full) collection |
---|
| 336 | */ |
---|
| 337 | static void restartcollection (global_State *g) { |
---|
| 338 | g->gray = g->grayagain = NULL; |
---|
| 339 | g->weak = g->allweak = g->ephemeron = NULL; |
---|
| 340 | markobject(g, g->mainthread); |
---|
| 341 | markvalue(g, &g->l_registry); |
---|
| 342 | markmt(g); |
---|
| 343 | markbeingfnz(g); /* mark any finalizing object left from previous cycle */ |
---|
| 344 | } |
---|
| 345 | |
---|
| 346 | /* }====================================================== */ |
---|
| 347 | |
---|
| 348 | |
---|
| 349 | /* |
---|
| 350 | ** {====================================================== |
---|
| 351 | ** Traverse functions |
---|
| 352 | ** ======================================================= |
---|
| 353 | */ |
---|
| 354 | |
---|
| 355 | static void traverseweakvalue (global_State *g, Table *h) { |
---|
| 356 | Node *n, *limit = gnodelast(h); |
---|
| 357 | /* if there is array part, assume it may have white values (do not |
---|
| 358 | traverse it just to check) */ |
---|
| 359 | int hasclears = (h->sizearray > 0); |
---|
| 360 | for (n = gnode(h, 0); n < limit; n++) { |
---|
| 361 | checkdeadkey(n); |
---|
| 362 | if (ttisnil(gval(n))) /* entry is empty? */ |
---|
| 363 | removeentry(n); /* remove it */ |
---|
| 364 | else { |
---|
| 365 | lua_assert(!ttisnil(gkey(n))); |
---|
| 366 | markvalue(g, gkey(n)); /* mark key */ |
---|
| 367 | if (!hasclears && iscleared(g, gval(n))) /* is there a white value? */ |
---|
| 368 | hasclears = 1; /* table will have to be cleared */ |
---|
| 369 | } |
---|
| 370 | } |
---|
| 371 | if (hasclears) |
---|
| 372 | linktable(h, &g->weak); /* has to be cleared later */ |
---|
| 373 | else /* no white values */ |
---|
| 374 | linktable(h, &g->grayagain); /* no need to clean */ |
---|
| 375 | } |
---|
| 376 | |
---|
| 377 | |
---|
| 378 | static int traverseephemeron (global_State *g, Table *h) { |
---|
| 379 | int marked = 0; /* true if an object is marked in this traversal */ |
---|
| 380 | int hasclears = 0; /* true if table has white keys */ |
---|
| 381 | int prop = 0; /* true if table has entry "white-key -> white-value" */ |
---|
| 382 | Node *n, *limit = gnodelast(h); |
---|
| 383 | int i; |
---|
| 384 | /* traverse array part (numeric keys are 'strong') */ |
---|
| 385 | for (i = 0; i < h->sizearray; i++) { |
---|
| 386 | if (valiswhite(&h->array[i])) { |
---|
| 387 | marked = 1; |
---|
| 388 | reallymarkobject(g, gcvalue(&h->array[i])); |
---|
| 389 | } |
---|
| 390 | } |
---|
| 391 | /* traverse hash part */ |
---|
| 392 | for (n = gnode(h, 0); n < limit; n++) { |
---|
| 393 | checkdeadkey(n); |
---|
| 394 | if (ttisnil(gval(n))) /* entry is empty? */ |
---|
| 395 | removeentry(n); /* remove it */ |
---|
| 396 | else if (iscleared(g, gkey(n))) { /* key is not marked (yet)? */ |
---|
| 397 | hasclears = 1; /* table must be cleared */ |
---|
| 398 | if (valiswhite(gval(n))) /* value not marked yet? */ |
---|
| 399 | prop = 1; /* must propagate again */ |
---|
| 400 | } |
---|
| 401 | else if (valiswhite(gval(n))) { /* value not marked yet? */ |
---|
| 402 | marked = 1; |
---|
| 403 | reallymarkobject(g, gcvalue(gval(n))); /* mark it now */ |
---|
| 404 | } |
---|
| 405 | } |
---|
| 406 | if (prop) |
---|
| 407 | linktable(h, &g->ephemeron); /* have to propagate again */ |
---|
| 408 | else if (hasclears) /* does table have white keys? */ |
---|
| 409 | linktable(h, &g->allweak); /* may have to clean white keys */ |
---|
| 410 | else /* no white keys */ |
---|
| 411 | linktable(h, &g->grayagain); /* no need to clean */ |
---|
| 412 | return marked; |
---|
| 413 | } |
---|
| 414 | |
---|
| 415 | |
---|
| 416 | static void traversestrongtable (global_State *g, Table *h) { |
---|
| 417 | Node *n, *limit = gnodelast(h); |
---|
| 418 | int i; |
---|
| 419 | for (i = 0; i < h->sizearray; i++) /* traverse array part */ |
---|
| 420 | markvalue(g, &h->array[i]); |
---|
| 421 | for (n = gnode(h, 0); n < limit; n++) { /* traverse hash part */ |
---|
| 422 | checkdeadkey(n); |
---|
| 423 | if (ttisnil(gval(n))) /* entry is empty? */ |
---|
| 424 | removeentry(n); /* remove it */ |
---|
| 425 | else { |
---|
| 426 | lua_assert(!ttisnil(gkey(n))); |
---|
| 427 | markvalue(g, gkey(n)); /* mark key */ |
---|
| 428 | markvalue(g, gval(n)); /* mark value */ |
---|
| 429 | } |
---|
| 430 | } |
---|
| 431 | } |
---|
| 432 | |
---|
| 433 | |
---|
| 434 | static lu_mem traversetable (global_State *g, Table *h) { |
---|
| 435 | const char *weakkey, *weakvalue; |
---|
| 436 | const TValue *mode = gfasttm(g, h->metatable, TM_MODE); |
---|
| 437 | markobject(g, h->metatable); |
---|
| 438 | if (mode && ttisstring(mode) && /* is there a weak mode? */ |
---|
| 439 | ((weakkey = strchr(svalue(mode), 'k')), |
---|
| 440 | (weakvalue = strchr(svalue(mode), 'v')), |
---|
| 441 | (weakkey || weakvalue))) { /* is really weak? */ |
---|
| 442 | black2gray(obj2gco(h)); /* keep table gray */ |
---|
| 443 | if (!weakkey) /* strong keys? */ |
---|
| 444 | traverseweakvalue(g, h); |
---|
| 445 | else if (!weakvalue) /* strong values? */ |
---|
| 446 | traverseephemeron(g, h); |
---|
| 447 | else /* all weak */ |
---|
| 448 | linktable(h, &g->allweak); /* nothing to traverse now */ |
---|
| 449 | } |
---|
| 450 | else /* not weak */ |
---|
| 451 | traversestrongtable(g, h); |
---|
| 452 | return sizeof(Table) + sizeof(TValue) * h->sizearray + |
---|
| 453 | sizeof(Node) * cast(size_t, sizenode(h)); |
---|
| 454 | } |
---|
| 455 | |
---|
| 456 | |
---|
| 457 | static int traverseproto (global_State *g, Proto *f) { |
---|
| 458 | int i; |
---|
| 459 | if (f->cache && iswhite(obj2gco(f->cache))) |
---|
| 460 | f->cache = NULL; /* allow cache to be collected */ |
---|
| 461 | markobject(g, f->source); |
---|
| 462 | for (i = 0; i < f->sizek; i++) /* mark literals */ |
---|
| 463 | markvalue(g, &f->k[i]); |
---|
| 464 | for (i = 0; i < f->sizeupvalues; i++) /* mark upvalue names */ |
---|
| 465 | markobject(g, f->upvalues[i].name); |
---|
| 466 | for (i = 0; i < f->sizep; i++) /* mark nested protos */ |
---|
| 467 | markobject(g, f->p[i]); |
---|
| 468 | for (i = 0; i < f->sizelocvars; i++) /* mark local-variable names */ |
---|
| 469 | markobject(g, f->locvars[i].varname); |
---|
| 470 | return sizeof(Proto) + sizeof(Instruction) * f->sizecode + |
---|
| 471 | sizeof(Proto *) * f->sizep + |
---|
| 472 | sizeof(TValue) * f->sizek + |
---|
| 473 | sizeof(int) * f->sizelineinfo + |
---|
| 474 | sizeof(LocVar) * f->sizelocvars + |
---|
| 475 | sizeof(Upvaldesc) * f->sizeupvalues; |
---|
| 476 | } |
---|
| 477 | |
---|
| 478 | |
---|
| 479 | static lu_mem traverseCclosure (global_State *g, CClosure *cl) { |
---|
| 480 | int i; |
---|
| 481 | for (i = 0; i < cl->nupvalues; i++) /* mark its upvalues */ |
---|
| 482 | markvalue(g, &cl->upvalue[i]); |
---|
| 483 | return sizeCclosure(cl->nupvalues); |
---|
| 484 | } |
---|
| 485 | |
---|
| 486 | static lu_mem traverseLclosure (global_State *g, LClosure *cl) { |
---|
| 487 | int i; |
---|
| 488 | markobject(g, cl->p); /* mark its prototype */ |
---|
| 489 | for (i = 0; i < cl->nupvalues; i++) /* mark its upvalues */ |
---|
| 490 | markobject(g, cl->upvals[i]); |
---|
| 491 | return sizeLclosure(cl->nupvalues); |
---|
| 492 | } |
---|
| 493 | |
---|
| 494 | |
---|
| 495 | static lu_mem traversestack (global_State *g, lua_State *th) { |
---|
| 496 | int n = 0; |
---|
| 497 | StkId o = th->stack; |
---|
| 498 | if (o == NULL) |
---|
| 499 | return 1; /* stack not completely built yet */ |
---|
| 500 | for (; o < th->top; o++) /* mark live elements in the stack */ |
---|
| 501 | markvalue(g, o); |
---|
| 502 | if (g->gcstate == GCSatomic) { /* final traversal? */ |
---|
| 503 | StkId lim = th->stack + th->stacksize; /* real end of stack */ |
---|
| 504 | for (; o < lim; o++) /* clear not-marked stack slice */ |
---|
| 505 | setnilvalue(o); |
---|
| 506 | } |
---|
| 507 | else { /* count call infos to compute size */ |
---|
| 508 | CallInfo *ci; |
---|
| 509 | for (ci = &th->base_ci; ci != th->ci; ci = ci->next) |
---|
| 510 | n++; |
---|
| 511 | } |
---|
| 512 | return sizeof(lua_State) + sizeof(TValue) * th->stacksize + |
---|
| 513 | sizeof(CallInfo) * n; |
---|
| 514 | } |
---|
| 515 | |
---|
| 516 | |
---|
| 517 | /* |
---|
| 518 | ** traverse one gray object, turning it to black (except for threads, |
---|
| 519 | ** which are always gray). |
---|
| 520 | */ |
---|
| 521 | static void propagatemark (global_State *g) { |
---|
| 522 | lu_mem size; |
---|
| 523 | GCObject *o = g->gray; |
---|
| 524 | lua_assert(isgray(o)); |
---|
| 525 | gray2black(o); |
---|
| 526 | switch (gch(o)->tt) { |
---|
| 527 | case LUA_TTABLE: { |
---|
| 528 | Table *h = gco2t(o); |
---|
| 529 | g->gray = h->gclist; /* remove from 'gray' list */ |
---|
| 530 | size = traversetable(g, h); |
---|
| 531 | break; |
---|
| 532 | } |
---|
| 533 | case LUA_TLCL: { |
---|
| 534 | LClosure *cl = gco2lcl(o); |
---|
| 535 | g->gray = cl->gclist; /* remove from 'gray' list */ |
---|
| 536 | size = traverseLclosure(g, cl); |
---|
| 537 | break; |
---|
| 538 | } |
---|
| 539 | case LUA_TCCL: { |
---|
| 540 | CClosure *cl = gco2ccl(o); |
---|
| 541 | g->gray = cl->gclist; /* remove from 'gray' list */ |
---|
| 542 | size = traverseCclosure(g, cl); |
---|
| 543 | break; |
---|
| 544 | } |
---|
| 545 | case LUA_TTHREAD: { |
---|
| 546 | lua_State *th = gco2th(o); |
---|
| 547 | g->gray = th->gclist; /* remove from 'gray' list */ |
---|
| 548 | th->gclist = g->grayagain; |
---|
| 549 | g->grayagain = o; /* insert into 'grayagain' list */ |
---|
| 550 | black2gray(o); |
---|
| 551 | size = traversestack(g, th); |
---|
| 552 | break; |
---|
| 553 | } |
---|
| 554 | case LUA_TPROTO: { |
---|
| 555 | Proto *p = gco2p(o); |
---|
| 556 | g->gray = p->gclist; /* remove from 'gray' list */ |
---|
| 557 | size = traverseproto(g, p); |
---|
| 558 | break; |
---|
| 559 | } |
---|
| 560 | default: lua_assert(0); return; |
---|
| 561 | } |
---|
| 562 | g->GCmemtrav += size; |
---|
| 563 | } |
---|
| 564 | |
---|
| 565 | |
---|
| 566 | static void propagateall (global_State *g) { |
---|
| 567 | while (g->gray) propagatemark(g); |
---|
| 568 | } |
---|
| 569 | |
---|
| 570 | |
---|
| 571 | static void propagatelist (global_State *g, GCObject *l) { |
---|
| 572 | lua_assert(g->gray == NULL); /* no grays left */ |
---|
| 573 | g->gray = l; |
---|
| 574 | propagateall(g); /* traverse all elements from 'l' */ |
---|
| 575 | } |
---|
| 576 | |
---|
| 577 | /* |
---|
| 578 | ** retraverse all gray lists. Because tables may be reinserted in other |
---|
| 579 | ** lists when traversed, traverse the original lists to avoid traversing |
---|
| 580 | ** twice the same table (which is not wrong, but inefficient) |
---|
| 581 | */ |
---|
| 582 | static void retraversegrays (global_State *g) { |
---|
| 583 | GCObject *weak = g->weak; /* save original lists */ |
---|
| 584 | GCObject *grayagain = g->grayagain; |
---|
| 585 | GCObject *ephemeron = g->ephemeron; |
---|
| 586 | g->weak = g->grayagain = g->ephemeron = NULL; |
---|
| 587 | propagateall(g); /* traverse main gray list */ |
---|
| 588 | propagatelist(g, grayagain); |
---|
| 589 | propagatelist(g, weak); |
---|
| 590 | propagatelist(g, ephemeron); |
---|
| 591 | } |
---|
| 592 | |
---|
| 593 | |
---|
| 594 | static void convergeephemerons (global_State *g) { |
---|
| 595 | int changed; |
---|
| 596 | do { |
---|
| 597 | GCObject *w; |
---|
| 598 | GCObject *next = g->ephemeron; /* get ephemeron list */ |
---|
| 599 | g->ephemeron = NULL; /* tables will return to this list when traversed */ |
---|
| 600 | changed = 0; |
---|
| 601 | while ((w = next) != NULL) { |
---|
| 602 | next = gco2t(w)->gclist; |
---|
| 603 | if (traverseephemeron(g, gco2t(w))) { /* traverse marked some value? */ |
---|
| 604 | propagateall(g); /* propagate changes */ |
---|
| 605 | changed = 1; /* will have to revisit all ephemeron tables */ |
---|
| 606 | } |
---|
| 607 | } |
---|
| 608 | } while (changed); |
---|
| 609 | } |
---|
| 610 | |
---|
| 611 | /* }====================================================== */ |
---|
| 612 | |
---|
| 613 | |
---|
| 614 | /* |
---|
| 615 | ** {====================================================== |
---|
| 616 | ** Sweep Functions |
---|
| 617 | ** ======================================================= |
---|
| 618 | */ |
---|
| 619 | |
---|
| 620 | |
---|
| 621 | /* |
---|
| 622 | ** clear entries with unmarked keys from all weaktables in list 'l' up |
---|
| 623 | ** to element 'f' |
---|
| 624 | */ |
---|
| 625 | static void clearkeys (global_State *g, GCObject *l, GCObject *f) { |
---|
| 626 | for (; l != f; l = gco2t(l)->gclist) { |
---|
| 627 | Table *h = gco2t(l); |
---|
| 628 | Node *n, *limit = gnodelast(h); |
---|
| 629 | for (n = gnode(h, 0); n < limit; n++) { |
---|
| 630 | if (!ttisnil(gval(n)) && (iscleared(g, gkey(n)))) { |
---|
| 631 | setnilvalue(gval(n)); /* remove value ... */ |
---|
| 632 | removeentry(n); /* and remove entry from table */ |
---|
| 633 | } |
---|
| 634 | } |
---|
| 635 | } |
---|
| 636 | } |
---|
| 637 | |
---|
| 638 | |
---|
| 639 | /* |
---|
| 640 | ** clear entries with unmarked values from all weaktables in list 'l' up |
---|
| 641 | ** to element 'f' |
---|
| 642 | */ |
---|
| 643 | static void clearvalues (global_State *g, GCObject *l, GCObject *f) { |
---|
| 644 | for (; l != f; l = gco2t(l)->gclist) { |
---|
| 645 | Table *h = gco2t(l); |
---|
| 646 | Node *n, *limit = gnodelast(h); |
---|
| 647 | int i; |
---|
| 648 | for (i = 0; i < h->sizearray; i++) { |
---|
| 649 | TValue *o = &h->array[i]; |
---|
| 650 | if (iscleared(g, o)) /* value was collected? */ |
---|
| 651 | setnilvalue(o); /* remove value */ |
---|
| 652 | } |
---|
| 653 | for (n = gnode(h, 0); n < limit; n++) { |
---|
| 654 | if (!ttisnil(gval(n)) && iscleared(g, gval(n))) { |
---|
| 655 | setnilvalue(gval(n)); /* remove value ... */ |
---|
| 656 | removeentry(n); /* and remove entry from table */ |
---|
| 657 | } |
---|
| 658 | } |
---|
| 659 | } |
---|
| 660 | } |
---|
| 661 | |
---|
| 662 | |
---|
| 663 | static void freeobj (lua_State *L, GCObject *o) { |
---|
| 664 | switch (gch(o)->tt) { |
---|
| 665 | case LUA_TPROTO: luaF_freeproto(L, gco2p(o)); break; |
---|
| 666 | case LUA_TLCL: { |
---|
| 667 | luaM_freemem(L, o, sizeLclosure(gco2lcl(o)->nupvalues)); |
---|
| 668 | break; |
---|
| 669 | } |
---|
| 670 | case LUA_TCCL: { |
---|
| 671 | luaM_freemem(L, o, sizeCclosure(gco2ccl(o)->nupvalues)); |
---|
| 672 | break; |
---|
| 673 | } |
---|
| 674 | case LUA_TUPVAL: luaF_freeupval(L, gco2uv(o)); break; |
---|
| 675 | case LUA_TTABLE: luaH_free(L, gco2t(o)); break; |
---|
| 676 | case LUA_TTHREAD: luaE_freethread(L, gco2th(o)); break; |
---|
| 677 | case LUA_TUSERDATA: luaM_freemem(L, o, sizeudata(gco2u(o))); break; |
---|
| 678 | case LUA_TSHRSTR: |
---|
| 679 | G(L)->strt.nuse--; |
---|
| 680 | /* go through */ |
---|
| 681 | case LUA_TLNGSTR: { |
---|
| 682 | luaM_freemem(L, o, sizestring(gco2ts(o))); |
---|
| 683 | break; |
---|
| 684 | } |
---|
| 685 | default: lua_assert(0); |
---|
| 686 | } |
---|
| 687 | } |
---|
| 688 | |
---|
| 689 | |
---|
| 690 | #define sweepwholelist(L,p) sweeplist(L,p,MAX_LUMEM) |
---|
| 691 | static GCObject **sweeplist (lua_State *L, GCObject **p, lu_mem count); |
---|
| 692 | |
---|
| 693 | |
---|
| 694 | /* |
---|
| 695 | ** sweep the (open) upvalues of a thread and resize its stack and |
---|
| 696 | ** list of call-info structures. |
---|
| 697 | */ |
---|
| 698 | static void sweepthread (lua_State *L, lua_State *L1) { |
---|
| 699 | if (L1->stack == NULL) return; /* stack not completely built yet */ |
---|
| 700 | sweepwholelist(L, &L1->openupval); /* sweep open upvalues */ |
---|
| 701 | luaE_freeCI(L1); /* free extra CallInfo slots */ |
---|
| 702 | /* should not change the stack during an emergency gc cycle */ |
---|
| 703 | if (G(L)->gckind != KGC_EMERGENCY) |
---|
| 704 | luaD_shrinkstack(L1); |
---|
| 705 | } |
---|
| 706 | |
---|
| 707 | |
---|
| 708 | /* |
---|
| 709 | ** sweep at most 'count' elements from a list of GCObjects erasing dead |
---|
| 710 | ** objects, where a dead (not alive) object is one marked with the "old" |
---|
| 711 | ** (non current) white and not fixed. |
---|
| 712 | ** In non-generational mode, change all non-dead objects back to white, |
---|
| 713 | ** preparing for next collection cycle. |
---|
| 714 | ** In generational mode, keep black objects black, and also mark them as |
---|
| 715 | ** old; stop when hitting an old object, as all objects after that |
---|
| 716 | ** one will be old too. |
---|
| 717 | ** When object is a thread, sweep its list of open upvalues too. |
---|
| 718 | */ |
---|
| 719 | static GCObject **sweeplist (lua_State *L, GCObject **p, lu_mem count) { |
---|
| 720 | global_State *g = G(L); |
---|
| 721 | int ow = otherwhite(g); |
---|
| 722 | int toclear, toset; /* bits to clear and to set in all live objects */ |
---|
| 723 | int tostop; /* stop sweep when this is true */ |
---|
| 724 | if (isgenerational(g)) { /* generational mode? */ |
---|
| 725 | toclear = ~0; /* clear nothing */ |
---|
| 726 | toset = bitmask(OLDBIT); /* set the old bit of all surviving objects */ |
---|
| 727 | tostop = bitmask(OLDBIT); /* do not sweep old generation */ |
---|
| 728 | } |
---|
| 729 | else { /* normal mode */ |
---|
| 730 | toclear = maskcolors; /* clear all color bits + old bit */ |
---|
| 731 | toset = luaC_white(g); /* make object white */ |
---|
| 732 | tostop = 0; /* do not stop */ |
---|
| 733 | } |
---|
| 734 | while (*p != NULL && count-- > 0) { |
---|
| 735 | GCObject *curr = *p; |
---|
| 736 | int marked = gch(curr)->marked; |
---|
| 737 | if (isdeadm(ow, marked)) { /* is 'curr' dead? */ |
---|
| 738 | *p = gch(curr)->next; /* remove 'curr' from list */ |
---|
| 739 | freeobj(L, curr); /* erase 'curr' */ |
---|
| 740 | } |
---|
| 741 | else { |
---|
| 742 | if (testbits(marked, tostop)) |
---|
| 743 | return NULL; /* stop sweeping this list */ |
---|
| 744 | if (gch(curr)->tt == LUA_TTHREAD) |
---|
| 745 | sweepthread(L, gco2th(curr)); /* sweep thread's upvalues */ |
---|
| 746 | /* update marks */ |
---|
| 747 | gch(curr)->marked = cast_byte((marked & toclear) | toset); |
---|
| 748 | p = &gch(curr)->next; /* go to next element */ |
---|
| 749 | } |
---|
| 750 | } |
---|
| 751 | return (*p == NULL) ? NULL : p; |
---|
| 752 | } |
---|
| 753 | |
---|
| 754 | |
---|
| 755 | /* |
---|
| 756 | ** sweep a list until a live object (or end of list) |
---|
| 757 | */ |
---|
| 758 | static GCObject **sweeptolive (lua_State *L, GCObject **p, int *n) { |
---|
| 759 | GCObject ** old = p; |
---|
| 760 | int i = 0; |
---|
| 761 | do { |
---|
| 762 | i++; |
---|
| 763 | p = sweeplist(L, p, 1); |
---|
| 764 | } while (p == old); |
---|
| 765 | if (n) *n += i; |
---|
| 766 | return p; |
---|
| 767 | } |
---|
| 768 | |
---|
| 769 | /* }====================================================== */ |
---|
| 770 | |
---|
| 771 | |
---|
| 772 | /* |
---|
| 773 | ** {====================================================== |
---|
| 774 | ** Finalization |
---|
| 775 | ** ======================================================= |
---|
| 776 | */ |
---|
| 777 | |
---|
| 778 | static void checkSizes (lua_State *L) { |
---|
| 779 | global_State *g = G(L); |
---|
| 780 | if (g->gckind != KGC_EMERGENCY) { /* do not change sizes in emergency */ |
---|
| 781 | int hs = g->strt.size / 2; /* half the size of the string table */ |
---|
| 782 | if (g->strt.nuse < cast(lu_int32, hs)) /* using less than that half? */ |
---|
| 783 | luaS_resize(L, hs); /* halve its size */ |
---|
| 784 | luaZ_freebuffer(L, &g->buff); /* free concatenation buffer */ |
---|
| 785 | } |
---|
| 786 | } |
---|
| 787 | |
---|
| 788 | |
---|
| 789 | static GCObject *udata2finalize (global_State *g) { |
---|
| 790 | GCObject *o = g->tobefnz; /* get first element */ |
---|
| 791 | lua_assert(isfinalized(o)); |
---|
| 792 | g->tobefnz = gch(o)->next; /* remove it from 'tobefnz' list */ |
---|
| 793 | gch(o)->next = g->allgc; /* return it to 'allgc' list */ |
---|
| 794 | g->allgc = o; |
---|
| 795 | resetbit(gch(o)->marked, SEPARATED); /* mark that it is not in 'tobefnz' */ |
---|
| 796 | lua_assert(!isold(o)); /* see MOVE OLD rule */ |
---|
| 797 | if (!keepinvariantout(g)) /* not keeping invariant? */ |
---|
| 798 | makewhite(g, o); /* "sweep" object */ |
---|
| 799 | return o; |
---|
| 800 | } |
---|
| 801 | |
---|
| 802 | |
---|
| 803 | static void dothecall (lua_State *L, void *ud) { |
---|
| 804 | UNUSED(ud); |
---|
| 805 | luaD_call(L, L->top - 2, 0, 0); |
---|
| 806 | } |
---|
| 807 | |
---|
| 808 | |
---|
| 809 | static void GCTM (lua_State *L, int propagateerrors) { |
---|
| 810 | global_State *g = G(L); |
---|
| 811 | const TValue *tm; |
---|
| 812 | TValue v; |
---|
| 813 | setgcovalue(L, &v, udata2finalize(g)); |
---|
| 814 | tm = luaT_gettmbyobj(L, &v, TM_GC); |
---|
| 815 | if (tm != NULL && ttisfunction(tm)) { /* is there a finalizer? */ |
---|
| 816 | int status; |
---|
| 817 | lu_byte oldah = L->allowhook; |
---|
| 818 | int running = g->gcrunning; |
---|
| 819 | L->allowhook = 0; /* stop debug hooks during GC metamethod */ |
---|
| 820 | g->gcrunning = 0; /* avoid GC steps */ |
---|
| 821 | setobj2s(L, L->top, tm); /* push finalizer... */ |
---|
| 822 | setobj2s(L, L->top + 1, &v); /* ... and its argument */ |
---|
| 823 | L->top += 2; /* and (next line) call the finalizer */ |
---|
| 824 | status = luaD_pcall(L, dothecall, NULL, savestack(L, L->top - 2), 0); |
---|
| 825 | L->allowhook = oldah; /* restore hooks */ |
---|
| 826 | g->gcrunning = running; /* restore state */ |
---|
| 827 | if (status != LUA_OK && propagateerrors) { /* error while running __gc? */ |
---|
| 828 | if (status == LUA_ERRRUN) { /* is there an error object? */ |
---|
| 829 | const char *msg = (ttisstring(L->top - 1)) |
---|
| 830 | ? svalue(L->top - 1) |
---|
| 831 | : "no message"; |
---|
| 832 | luaO_pushfstring(L, "error in __gc metamethod (%s)", msg); |
---|
| 833 | status = LUA_ERRGCMM; /* error in __gc metamethod */ |
---|
| 834 | } |
---|
| 835 | luaD_throw(L, status); /* re-throw error */ |
---|
| 836 | } |
---|
| 837 | } |
---|
| 838 | } |
---|
| 839 | |
---|
| 840 | |
---|
| 841 | /* |
---|
| 842 | ** move all unreachable objects (or 'all' objects) that need |
---|
| 843 | ** finalization from list 'finobj' to list 'tobefnz' (to be finalized) |
---|
| 844 | */ |
---|
| 845 | static void separatetobefnz (lua_State *L, int all) { |
---|
| 846 | global_State *g = G(L); |
---|
| 847 | GCObject **p = &g->finobj; |
---|
| 848 | GCObject *curr; |
---|
| 849 | GCObject **lastnext = &g->tobefnz; |
---|
| 850 | /* find last 'next' field in 'tobefnz' list (to add elements in its end) */ |
---|
| 851 | while (*lastnext != NULL) |
---|
| 852 | lastnext = &gch(*lastnext)->next; |
---|
| 853 | while ((curr = *p) != NULL) { /* traverse all finalizable objects */ |
---|
| 854 | lua_assert(!isfinalized(curr)); |
---|
| 855 | lua_assert(testbit(gch(curr)->marked, SEPARATED)); |
---|
| 856 | if (!(iswhite(curr) || all)) /* not being collected? */ |
---|
| 857 | p = &gch(curr)->next; /* don't bother with it */ |
---|
| 858 | else { |
---|
| 859 | l_setbit(gch(curr)->marked, FINALIZEDBIT); /* won't be finalized again */ |
---|
| 860 | *p = gch(curr)->next; /* remove 'curr' from 'finobj' list */ |
---|
| 861 | gch(curr)->next = *lastnext; /* link at the end of 'tobefnz' list */ |
---|
| 862 | *lastnext = curr; |
---|
| 863 | lastnext = &gch(curr)->next; |
---|
| 864 | } |
---|
| 865 | } |
---|
| 866 | } |
---|
| 867 | |
---|
| 868 | |
---|
| 869 | /* |
---|
| 870 | ** if object 'o' has a finalizer, remove it from 'allgc' list (must |
---|
| 871 | ** search the list to find it) and link it in 'finobj' list. |
---|
| 872 | */ |
---|
| 873 | void luaC_checkfinalizer (lua_State *L, GCObject *o, Table *mt) { |
---|
| 874 | global_State *g = G(L); |
---|
| 875 | if (testbit(gch(o)->marked, SEPARATED) || /* obj. is already separated... */ |
---|
| 876 | isfinalized(o) || /* ... or is finalized... */ |
---|
| 877 | gfasttm(g, mt, TM_GC) == NULL) /* or has no finalizer? */ |
---|
| 878 | return; /* nothing to be done */ |
---|
| 879 | else { /* move 'o' to 'finobj' list */ |
---|
| 880 | GCObject **p; |
---|
| 881 | GCheader *ho = gch(o); |
---|
| 882 | if (g->sweepgc == &ho->next) { /* avoid removing current sweep object */ |
---|
| 883 | lua_assert(issweepphase(g)); |
---|
| 884 | g->sweepgc = sweeptolive(L, g->sweepgc, NULL); |
---|
| 885 | } |
---|
| 886 | /* search for pointer pointing to 'o' */ |
---|
| 887 | for (p = &g->allgc; *p != o; p = &gch(*p)->next) { /* empty */ } |
---|
| 888 | *p = ho->next; /* remove 'o' from root list */ |
---|
| 889 | ho->next = g->finobj; /* link it in list 'finobj' */ |
---|
| 890 | g->finobj = o; |
---|
| 891 | l_setbit(ho->marked, SEPARATED); /* mark it as such */ |
---|
| 892 | if (!keepinvariantout(g)) /* not keeping invariant? */ |
---|
| 893 | makewhite(g, o); /* "sweep" object */ |
---|
| 894 | else |
---|
| 895 | resetoldbit(o); /* see MOVE OLD rule */ |
---|
| 896 | } |
---|
| 897 | } |
---|
| 898 | |
---|
| 899 | /* }====================================================== */ |
---|
| 900 | |
---|
| 901 | |
---|
| 902 | /* |
---|
| 903 | ** {====================================================== |
---|
| 904 | ** GC control |
---|
| 905 | ** ======================================================= |
---|
| 906 | */ |
---|
| 907 | |
---|
| 908 | |
---|
| 909 | /* |
---|
| 910 | ** set a reasonable "time" to wait before starting a new GC cycle; |
---|
| 911 | ** cycle will start when memory use hits threshold |
---|
| 912 | */ |
---|
| 913 | static void setpause (global_State *g, l_mem estimate) { |
---|
| 914 | l_mem debt, threshold; |
---|
| 915 | estimate = estimate / PAUSEADJ; /* adjust 'estimate' */ |
---|
| 916 | threshold = (g->gcpause < MAX_LMEM / estimate) /* overflow? */ |
---|
| 917 | ? estimate * g->gcpause /* no overflow */ |
---|
| 918 | : MAX_LMEM; /* overflow; truncate to maximum */ |
---|
| 919 | debt = -cast(l_mem, threshold - gettotalbytes(g)); |
---|
| 920 | luaE_setdebt(g, debt); |
---|
| 921 | } |
---|
| 922 | |
---|
| 923 | |
---|
| 924 | #define sweepphases \ |
---|
| 925 | (bitmask(GCSsweepstring) | bitmask(GCSsweepudata) | bitmask(GCSsweep)) |
---|
| 926 | |
---|
| 927 | |
---|
| 928 | /* |
---|
| 929 | ** enter first sweep phase (strings) and prepare pointers for other |
---|
| 930 | ** sweep phases. The calls to 'sweeptolive' make pointers point to an |
---|
| 931 | ** object inside the list (instead of to the header), so that the real |
---|
| 932 | ** sweep do not need to skip objects created between "now" and the start |
---|
| 933 | ** of the real sweep. |
---|
| 934 | ** Returns how many objects it swept. |
---|
| 935 | */ |
---|
| 936 | static int entersweep (lua_State *L) { |
---|
| 937 | global_State *g = G(L); |
---|
| 938 | int n = 0; |
---|
| 939 | g->gcstate = GCSsweepstring; |
---|
| 940 | lua_assert(g->sweepgc == NULL && g->sweepfin == NULL); |
---|
| 941 | /* prepare to sweep strings, finalizable objects, and regular objects */ |
---|
| 942 | g->sweepstrgc = 0; |
---|
| 943 | g->sweepfin = sweeptolive(L, &g->finobj, &n); |
---|
| 944 | g->sweepgc = sweeptolive(L, &g->allgc, &n); |
---|
| 945 | return n; |
---|
| 946 | } |
---|
| 947 | |
---|
| 948 | |
---|
| 949 | /* |
---|
| 950 | ** change GC mode |
---|
| 951 | */ |
---|
| 952 | void luaC_changemode (lua_State *L, int mode) { |
---|
| 953 | global_State *g = G(L); |
---|
| 954 | if (mode == g->gckind) return; /* nothing to change */ |
---|
| 955 | if (mode == KGC_GEN) { /* change to generational mode */ |
---|
| 956 | /* make sure gray lists are consistent */ |
---|
| 957 | luaC_runtilstate(L, bitmask(GCSpropagate)); |
---|
| 958 | g->GCestimate = gettotalbytes(g); |
---|
| 959 | g->gckind = KGC_GEN; |
---|
| 960 | } |
---|
| 961 | else { /* change to incremental mode */ |
---|
| 962 | /* sweep all objects to turn them back to white |
---|
| 963 | (as white has not changed, nothing extra will be collected) */ |
---|
| 964 | g->gckind = KGC_NORMAL; |
---|
| 965 | entersweep(L); |
---|
| 966 | luaC_runtilstate(L, ~sweepphases); |
---|
| 967 | } |
---|
| 968 | } |
---|
| 969 | |
---|
| 970 | |
---|
| 971 | /* |
---|
| 972 | ** call all pending finalizers |
---|
| 973 | */ |
---|
| 974 | static void callallpendingfinalizers (lua_State *L, int propagateerrors) { |
---|
| 975 | global_State *g = G(L); |
---|
| 976 | while (g->tobefnz) { |
---|
| 977 | resetoldbit(g->tobefnz); |
---|
| 978 | GCTM(L, propagateerrors); |
---|
| 979 | } |
---|
| 980 | } |
---|
| 981 | |
---|
| 982 | |
---|
| 983 | void luaC_freeallobjects (lua_State *L) { |
---|
| 984 | global_State *g = G(L); |
---|
| 985 | int i; |
---|
| 986 | separatetobefnz(L, 1); /* separate all objects with finalizers */ |
---|
| 987 | lua_assert(g->finobj == NULL); |
---|
| 988 | callallpendingfinalizers(L, 0); |
---|
| 989 | g->currentwhite = WHITEBITS; /* this "white" makes all objects look dead */ |
---|
| 990 | g->gckind = KGC_NORMAL; |
---|
| 991 | sweepwholelist(L, &g->finobj); /* finalizers can create objs. in 'finobj' */ |
---|
| 992 | sweepwholelist(L, &g->allgc); |
---|
| 993 | for (i = 0; i < g->strt.size; i++) /* free all string lists */ |
---|
| 994 | sweepwholelist(L, &g->strt.hash[i]); |
---|
| 995 | lua_assert(g->strt.nuse == 0); |
---|
| 996 | } |
---|
| 997 | |
---|
| 998 | |
---|
| 999 | static l_mem atomic (lua_State *L) { |
---|
| 1000 | global_State *g = G(L); |
---|
| 1001 | l_mem work = -cast(l_mem, g->GCmemtrav); /* start counting work */ |
---|
| 1002 | GCObject *origweak, *origall; |
---|
| 1003 | lua_assert(!iswhite(obj2gco(g->mainthread))); |
---|
| 1004 | markobject(g, L); /* mark running thread */ |
---|
| 1005 | /* registry and global metatables may be changed by API */ |
---|
| 1006 | markvalue(g, &g->l_registry); |
---|
| 1007 | markmt(g); /* mark basic metatables */ |
---|
| 1008 | /* remark occasional upvalues of (maybe) dead threads */ |
---|
| 1009 | remarkupvals(g); |
---|
| 1010 | propagateall(g); /* propagate changes */ |
---|
| 1011 | work += g->GCmemtrav; /* stop counting (do not (re)count grays) */ |
---|
| 1012 | /* traverse objects caught by write barrier and by 'remarkupvals' */ |
---|
| 1013 | retraversegrays(g); |
---|
| 1014 | work -= g->GCmemtrav; /* restart counting */ |
---|
| 1015 | convergeephemerons(g); |
---|
| 1016 | /* at this point, all strongly accessible objects are marked. */ |
---|
| 1017 | /* clear values from weak tables, before checking finalizers */ |
---|
| 1018 | clearvalues(g, g->weak, NULL); |
---|
| 1019 | clearvalues(g, g->allweak, NULL); |
---|
| 1020 | origweak = g->weak; origall = g->allweak; |
---|
| 1021 | work += g->GCmemtrav; /* stop counting (objects being finalized) */ |
---|
| 1022 | separatetobefnz(L, 0); /* separate objects to be finalized */ |
---|
| 1023 | markbeingfnz(g); /* mark objects that will be finalized */ |
---|
| 1024 | propagateall(g); /* remark, to propagate `preserveness' */ |
---|
| 1025 | work -= g->GCmemtrav; /* restart counting */ |
---|
| 1026 | convergeephemerons(g); |
---|
| 1027 | /* at this point, all resurrected objects are marked. */ |
---|
| 1028 | /* remove dead objects from weak tables */ |
---|
| 1029 | clearkeys(g, g->ephemeron, NULL); /* clear keys from all ephemeron tables */ |
---|
| 1030 | clearkeys(g, g->allweak, NULL); /* clear keys from all allweak tables */ |
---|
| 1031 | /* clear values from resurrected weak tables */ |
---|
| 1032 | clearvalues(g, g->weak, origweak); |
---|
| 1033 | clearvalues(g, g->allweak, origall); |
---|
| 1034 | g->currentwhite = cast_byte(otherwhite(g)); /* flip current white */ |
---|
| 1035 | work += g->GCmemtrav; /* complete counting */ |
---|
| 1036 | return work; /* estimate of memory marked by 'atomic' */ |
---|
| 1037 | } |
---|
| 1038 | |
---|
| 1039 | |
---|
| 1040 | static lu_mem singlestep (lua_State *L) { |
---|
| 1041 | global_State *g = G(L); |
---|
| 1042 | switch (g->gcstate) { |
---|
| 1043 | case GCSpause: { |
---|
| 1044 | /* start to count memory traversed */ |
---|
| 1045 | g->GCmemtrav = g->strt.size * sizeof(GCObject*); |
---|
| 1046 | lua_assert(!isgenerational(g)); |
---|
| 1047 | restartcollection(g); |
---|
| 1048 | g->gcstate = GCSpropagate; |
---|
| 1049 | return g->GCmemtrav; |
---|
| 1050 | } |
---|
| 1051 | case GCSpropagate: { |
---|
| 1052 | if (g->gray) { |
---|
| 1053 | lu_mem oldtrav = g->GCmemtrav; |
---|
| 1054 | propagatemark(g); |
---|
| 1055 | return g->GCmemtrav - oldtrav; /* memory traversed in this step */ |
---|
| 1056 | } |
---|
| 1057 | else { /* no more `gray' objects */ |
---|
| 1058 | lu_mem work; |
---|
| 1059 | int sw; |
---|
| 1060 | g->gcstate = GCSatomic; /* finish mark phase */ |
---|
| 1061 | g->GCestimate = g->GCmemtrav; /* save what was counted */; |
---|
| 1062 | work = atomic(L); /* add what was traversed by 'atomic' */ |
---|
| 1063 | g->GCestimate += work; /* estimate of total memory traversed */ |
---|
| 1064 | sw = entersweep(L); |
---|
| 1065 | return work + sw * GCSWEEPCOST; |
---|
| 1066 | } |
---|
| 1067 | } |
---|
| 1068 | case GCSsweepstring: { |
---|
| 1069 | int i; |
---|
| 1070 | for (i = 0; i < GCSWEEPMAX && g->sweepstrgc + i < g->strt.size; i++) |
---|
| 1071 | sweepwholelist(L, &g->strt.hash[g->sweepstrgc + i]); |
---|
| 1072 | g->sweepstrgc += i; |
---|
| 1073 | if (g->sweepstrgc >= g->strt.size) /* no more strings to sweep? */ |
---|
| 1074 | g->gcstate = GCSsweepudata; |
---|
| 1075 | return i * GCSWEEPCOST; |
---|
| 1076 | } |
---|
| 1077 | case GCSsweepudata: { |
---|
| 1078 | if (g->sweepfin) { |
---|
| 1079 | g->sweepfin = sweeplist(L, g->sweepfin, GCSWEEPMAX); |
---|
| 1080 | return GCSWEEPMAX*GCSWEEPCOST; |
---|
| 1081 | } |
---|
| 1082 | else { |
---|
| 1083 | g->gcstate = GCSsweep; |
---|
| 1084 | return 0; |
---|
| 1085 | } |
---|
| 1086 | } |
---|
| 1087 | case GCSsweep: { |
---|
| 1088 | if (g->sweepgc) { |
---|
| 1089 | g->sweepgc = sweeplist(L, g->sweepgc, GCSWEEPMAX); |
---|
| 1090 | return GCSWEEPMAX*GCSWEEPCOST; |
---|
| 1091 | } |
---|
| 1092 | else { |
---|
| 1093 | /* sweep main thread */ |
---|
| 1094 | GCObject *mt = obj2gco(g->mainthread); |
---|
| 1095 | sweeplist(L, &mt, 1); |
---|
| 1096 | checkSizes(L); |
---|
| 1097 | g->gcstate = GCSpause; /* finish collection */ |
---|
| 1098 | return GCSWEEPCOST; |
---|
| 1099 | } |
---|
| 1100 | } |
---|
| 1101 | default: lua_assert(0); return 0; |
---|
| 1102 | } |
---|
| 1103 | } |
---|
| 1104 | |
---|
| 1105 | |
---|
| 1106 | /* |
---|
| 1107 | ** advances the garbage collector until it reaches a state allowed |
---|
| 1108 | ** by 'statemask' |
---|
| 1109 | */ |
---|
| 1110 | void luaC_runtilstate (lua_State *L, int statesmask) { |
---|
| 1111 | global_State *g = G(L); |
---|
| 1112 | while (!testbit(statesmask, g->gcstate)) |
---|
| 1113 | singlestep(L); |
---|
| 1114 | } |
---|
| 1115 | |
---|
| 1116 | |
---|
| 1117 | static void generationalcollection (lua_State *L) { |
---|
| 1118 | global_State *g = G(L); |
---|
| 1119 | lua_assert(g->gcstate == GCSpropagate); |
---|
| 1120 | if (g->GCestimate == 0) { /* signal for another major collection? */ |
---|
| 1121 | luaC_fullgc(L, 0); /* perform a full regular collection */ |
---|
| 1122 | g->GCestimate = gettotalbytes(g); /* update control */ |
---|
| 1123 | } |
---|
| 1124 | else { |
---|
| 1125 | lu_mem estimate = g->GCestimate; |
---|
| 1126 | luaC_runtilstate(L, bitmask(GCSpause)); /* run complete (minor) cycle */ |
---|
| 1127 | g->gcstate = GCSpropagate; /* skip restart */ |
---|
| 1128 | if (gettotalbytes(g) > (estimate / 100) * g->gcmajorinc) |
---|
| 1129 | g->GCestimate = 0; /* signal for a major collection */ |
---|
| 1130 | else |
---|
| 1131 | g->GCestimate = estimate; /* keep estimate from last major coll. */ |
---|
| 1132 | |
---|
| 1133 | } |
---|
| 1134 | setpause(g, gettotalbytes(g)); |
---|
| 1135 | lua_assert(g->gcstate == GCSpropagate); |
---|
| 1136 | } |
---|
| 1137 | |
---|
| 1138 | |
---|
| 1139 | static void incstep (lua_State *L) { |
---|
| 1140 | global_State *g = G(L); |
---|
| 1141 | l_mem debt = g->GCdebt; |
---|
| 1142 | int stepmul = g->gcstepmul; |
---|
| 1143 | if (stepmul < 40) stepmul = 40; /* avoid ridiculous low values (and 0) */ |
---|
| 1144 | /* convert debt from Kb to 'work units' (avoid zero debt and overflows) */ |
---|
| 1145 | debt = (debt / STEPMULADJ) + 1; |
---|
| 1146 | debt = (debt < MAX_LMEM / stepmul) ? debt * stepmul : MAX_LMEM; |
---|
| 1147 | do { /* always perform at least one single step */ |
---|
| 1148 | lu_mem work = singlestep(L); /* do some work */ |
---|
| 1149 | debt -= work; |
---|
| 1150 | } while (debt > -GCSTEPSIZE && g->gcstate != GCSpause); |
---|
| 1151 | if (g->gcstate == GCSpause) |
---|
| 1152 | setpause(g, g->GCestimate); /* pause until next cycle */ |
---|
| 1153 | else { |
---|
| 1154 | debt = (debt / stepmul) * STEPMULADJ; /* convert 'work units' to Kb */ |
---|
| 1155 | luaE_setdebt(g, debt); |
---|
| 1156 | } |
---|
| 1157 | } |
---|
| 1158 | |
---|
| 1159 | |
---|
| 1160 | /* |
---|
| 1161 | ** performs a basic GC step |
---|
| 1162 | */ |
---|
| 1163 | void luaC_forcestep (lua_State *L) { |
---|
| 1164 | global_State *g = G(L); |
---|
| 1165 | int i; |
---|
| 1166 | if (isgenerational(g)) generationalcollection(L); |
---|
| 1167 | else incstep(L); |
---|
| 1168 | /* run a few finalizers (or all of them at the end of a collect cycle) */ |
---|
| 1169 | for (i = 0; g->tobefnz && (i < GCFINALIZENUM || g->gcstate == GCSpause); i++) |
---|
| 1170 | GCTM(L, 1); /* call one finalizer */ |
---|
| 1171 | } |
---|
| 1172 | |
---|
| 1173 | |
---|
| 1174 | /* |
---|
| 1175 | ** performs a basic GC step only if collector is running |
---|
| 1176 | */ |
---|
| 1177 | void luaC_step (lua_State *L) { |
---|
| 1178 | global_State *g = G(L); |
---|
| 1179 | if (g->gcrunning) luaC_forcestep(L); |
---|
| 1180 | else luaE_setdebt(g, -GCSTEPSIZE); /* avoid being called too often */ |
---|
| 1181 | } |
---|
| 1182 | |
---|
| 1183 | |
---|
| 1184 | |
---|
| 1185 | /* |
---|
| 1186 | ** performs a full GC cycle; if "isemergency", does not call |
---|
| 1187 | ** finalizers (which could change stack positions) |
---|
| 1188 | */ |
---|
| 1189 | void luaC_fullgc (lua_State *L, int isemergency) { |
---|
| 1190 | global_State *g = G(L); |
---|
| 1191 | int origkind = g->gckind; |
---|
| 1192 | lua_assert(origkind != KGC_EMERGENCY); |
---|
| 1193 | if (isemergency) /* do not run finalizers during emergency GC */ |
---|
| 1194 | g->gckind = KGC_EMERGENCY; |
---|
| 1195 | else { |
---|
| 1196 | g->gckind = KGC_NORMAL; |
---|
| 1197 | callallpendingfinalizers(L, 1); |
---|
| 1198 | } |
---|
| 1199 | if (keepinvariant(g)) { /* may there be some black objects? */ |
---|
| 1200 | /* must sweep all objects to turn them back to white |
---|
| 1201 | (as white has not changed, nothing will be collected) */ |
---|
| 1202 | entersweep(L); |
---|
| 1203 | } |
---|
| 1204 | /* finish any pending sweep phase to start a new cycle */ |
---|
| 1205 | luaC_runtilstate(L, bitmask(GCSpause)); |
---|
| 1206 | luaC_runtilstate(L, ~bitmask(GCSpause)); /* start new collection */ |
---|
| 1207 | luaC_runtilstate(L, bitmask(GCSpause)); /* run entire collection */ |
---|
| 1208 | if (origkind == KGC_GEN) { /* generational mode? */ |
---|
| 1209 | /* generational mode must be kept in propagate phase */ |
---|
| 1210 | luaC_runtilstate(L, bitmask(GCSpropagate)); |
---|
| 1211 | } |
---|
| 1212 | g->gckind = origkind; |
---|
| 1213 | setpause(g, gettotalbytes(g)); |
---|
| 1214 | if (!isemergency) /* do not run finalizers during emergency GC */ |
---|
| 1215 | callallpendingfinalizers(L, 1); |
---|
| 1216 | } |
---|
| 1217 | |
---|
| 1218 | /* }====================================================== */ |
---|
| 1219 | |
---|
| 1220 | |
---|