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