lgc.c 23 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834
  1. /*
  2. ** $Id: lgc.c,v 2.43 2008/02/11 15:46:03 roberto Exp roberto $
  3. ** Garbage Collector
  4. ** See Copyright Notice in lua.h
  5. */
  6. #include <string.h>
  7. #define lgc_c
  8. #define LUA_CORE
  9. #include "lua.h"
  10. #include "ldebug.h"
  11. #include "ldo.h"
  12. #include "lfunc.h"
  13. #include "lgc.h"
  14. #include "lmem.h"
  15. #include "lobject.h"
  16. #include "lstate.h"
  17. #include "lstring.h"
  18. #include "ltable.h"
  19. #include "ltm.h"
  20. #define GCSTEPSIZE 1024u
  21. #define GCSWEEPMAX 40
  22. #define GCSWEEPCOST 10
  23. #define GCFINALIZECOST 100
  24. #define maskcolors cast_byte(~(bitmask(BLACKBIT)|WHITEBITS))
  25. #define makewhite(g,x) \
  26. (gch(x)->marked = cast_byte((gch(x)->marked & maskcolors) | luaC_white(g)))
  27. #define white2gray(x) resetbits(gch(x)->marked, WHITEBITS)
  28. #define black2gray(x) resetbit(gch(x)->marked, BLACKBIT)
  29. #define stringmark(s) resetbits((s)->tsv.marked, WHITEBITS)
  30. #define isfinalized(u) testbit((u)->marked, FINALIZEDBIT)
  31. #define markvalue(g,o) { checkconsistency(o); \
  32. if (valiswhite(o)) reallymarkobject(g,gcvalue(o)); }
  33. #define markobject(g,t) { if ((t) && iswhite(obj2gco(t))) \
  34. reallymarkobject(g, obj2gco(t)); }
  35. #define setthreshold(g) (g->GCthreshold = (g->estimate/100) * g->gcpause)
  36. static void linktable (Table *h, GCObject **p) {
  37. h->gclist = *p;
  38. *p = obj2gco(h);
  39. }
  40. static void removeentry (Node *n) {
  41. lua_assert(ttisnil(gval(n)));
  42. if (iscollectable(gkey(n)))
  43. setttype(gkey(n), LUA_TDEADKEY); /* dead key; remove it */
  44. }
  45. /*
  46. ** The next function tells whether a key or value can be cleared from
  47. ** a weak table. Non-collectable objects are never removed from weak
  48. ** tables. Strings behave as `values', so are never removed too. for
  49. ** other objects: if really collected, cannot keep them; for userdata
  50. ** being finalized, keep them in keys, but not in values
  51. */
  52. static int iscleared (const TValue *o, int iskey) {
  53. if (!iscollectable(o)) return 0;
  54. if (ttisstring(o)) {
  55. stringmark(rawtsvalue(o)); /* strings are `values', so are never weak */
  56. return 0;
  57. }
  58. return iswhite(gcvalue(o)) ||
  59. (ttisuserdata(o) && (!iskey && isfinalized(uvalue(o))));
  60. }
  61. static void reallymarkobject (global_State *g, GCObject *o) {
  62. lua_assert(iswhite(o) && !isdead(g, o));
  63. white2gray(o);
  64. switch (gch(o)->tt) {
  65. case LUA_TSTRING: {
  66. return;
  67. }
  68. case LUA_TUSERDATA: {
  69. Table *mt = gco2u(o)->metatable;
  70. gray2black(o); /* udata are never gray */
  71. markobject(g, mt);
  72. markobject(g, gco2u(o)->env);
  73. return;
  74. }
  75. case LUA_TUPVAL: {
  76. UpVal *uv = gco2uv(o);
  77. markvalue(g, uv->v);
  78. if (uv->v == &uv->u.value) /* closed? */
  79. gray2black(o); /* open upvalues are never black */
  80. return;
  81. }
  82. case LUA_TFUNCTION: {
  83. gco2cl(o)->c.gclist = g->gray;
  84. g->gray = o;
  85. break;
  86. }
  87. case LUA_TTABLE: {
  88. linktable(gco2t(o), &g->gray);
  89. break;
  90. }
  91. case LUA_TTHREAD: {
  92. gco2th(o)->gclist = g->gray;
  93. g->gray = o;
  94. break;
  95. }
  96. case LUA_TPROTO: {
  97. gco2p(o)->gclist = g->gray;
  98. g->gray = o;
  99. break;
  100. }
  101. default: lua_assert(0);
  102. }
  103. }
  104. /* move 'dead' udata that need finalization to list 'tobefnz' */
  105. size_t luaC_separateudata (lua_State *L, int all) {
  106. global_State *g = G(L);
  107. size_t deadmem = 0;
  108. GCObject **p = &g->tmudata;
  109. GCObject *curr;
  110. while ((curr = *p) != NULL) {
  111. lua_assert(ttisuserdata(gch(curr)) && !isfinalized(gco2u(curr)));
  112. lua_assert(testbit(gch(curr)->marked, SEPARATED));
  113. if (all) makewhite(g, curr); /* if 'all', collect all objects */
  114. if (!iswhite(curr)) /* not being collected? */
  115. p = &gch(curr)->next; /* don't bother with it */
  116. else {
  117. l_setbit(gch(curr)->marked, FINALIZEDBIT); /* won't be finalized again */
  118. reallymarkobject(g, curr); /* won't be collected now */
  119. deadmem += sizeudata(gco2u(curr));
  120. *p = gch(curr)->next; /* remove 'curr' from 'tmudata' list */
  121. /* link 'curr' at the end of 'tobefnz' list */
  122. if (g->tobefnz == NULL) /* list is empty? */
  123. g->tobefnz = gch(curr)->next = curr; /* creates a circular list */
  124. else {
  125. gch(curr)->next = gch(g->tobefnz)->next;
  126. gch(g->tobefnz)->next = curr;
  127. g->tobefnz = curr;
  128. }
  129. }
  130. }
  131. return deadmem;
  132. }
  133. static void traverseweakvalue (global_State *g, Table *h) {
  134. int i = sizenode(h);
  135. while (i--) {
  136. Node *n = gnode(h, i);
  137. lua_assert(ttype(gkey(n)) != LUA_TDEADKEY || ttisnil(gval(n)));
  138. if (ttisnil(gval(n)))
  139. removeentry(n); /* remove empty entries */
  140. else {
  141. lua_assert(!ttisnil(gkey(n)));
  142. markvalue(g, gkey(n));
  143. }
  144. }
  145. linktable(h, &g->weak);
  146. }
  147. static int traverseephemeron (global_State *g, Table *h) {
  148. int marked = 0;
  149. int hasclears = 0;
  150. int i = h->sizearray;
  151. while (i--) { /* mark array part (numeric keys are 'strong') */
  152. if (valiswhite(&h->array[i])) {
  153. marked = 1;
  154. reallymarkobject(g, gcvalue(&h->array[i]));
  155. }
  156. }
  157. i = sizenode(h);
  158. while (i--) {
  159. Node *n = gnode(h, i);
  160. lua_assert(ttype(gkey(n)) != LUA_TDEADKEY || ttisnil(gval(n)));
  161. if (ttisnil(gval(n))) /* entry is empty? */
  162. removeentry(n); /* remove it */
  163. else if (valiswhite(gval(n))) {
  164. /* value is not marked yet */
  165. if (iscleared(key2tval(n), 1)) /* key is not marked (yet)? */
  166. hasclears = 1; /* may have to propagate mark from key to value */
  167. else { /* mark value only if key is marked */
  168. marked = 1; /* some mark changed status */
  169. reallymarkobject(g, gcvalue(gval(n)));
  170. }
  171. }
  172. }
  173. if (hasclears)
  174. linktable(h, &g->ephemeron);
  175. else /* nothing to propagate */
  176. linktable(h, &g->weak); /* avoid convergence phase */
  177. return marked;
  178. }
  179. static void traversestrongtable (global_State *g, Table *h) {
  180. int i;
  181. i = h->sizearray;
  182. while (i--)
  183. markvalue(g, &h->array[i]);
  184. i = sizenode(h);
  185. while (i--) {
  186. Node *n = gnode(h, i);
  187. lua_assert(ttype(gkey(n)) != LUA_TDEADKEY || ttisnil(gval(n)));
  188. if (ttisnil(gval(n)))
  189. removeentry(n); /* remove empty entries */
  190. else {
  191. lua_assert(!ttisnil(gkey(n)));
  192. markvalue(g, gkey(n));
  193. markvalue(g, gval(n));
  194. }
  195. }
  196. }
  197. static void traversetable (global_State *g, Table *h) {
  198. const TValue *mode = gfasttm(g, h->metatable, TM_MODE);
  199. markobject(g, h->metatable);
  200. if (mode && ttisstring(mode)) { /* is there a weak mode? */
  201. int weakkey = (strchr(svalue(mode), 'k') != NULL);
  202. int weakvalue = (strchr(svalue(mode), 'v') != NULL);
  203. if (weakkey || weakvalue) { /* is really weak? */
  204. black2gray(obj2gco(h)); /* keep table gray */
  205. if (!weakkey) /* strong keys? */
  206. traverseweakvalue(g, h);
  207. else if (!weakvalue) /* strong values? */
  208. traverseephemeron(g, h);
  209. else
  210. linktable(h, &g->allweak); /* nothing to traverse now */
  211. return;
  212. } /* else go through */
  213. }
  214. traversestrongtable(g, h);
  215. }
  216. /*
  217. ** All marks are conditional because a GC may happen while the
  218. ** prototype is still being created
  219. */
  220. static void traverseproto (global_State *g, Proto *f) {
  221. int i;
  222. if (f->source) stringmark(f->source);
  223. for (i=0; i<f->sizek; i++) /* mark literals */
  224. markvalue(g, &f->k[i]);
  225. for (i=0; i<f->sizeupvalues; i++) { /* mark upvalue names */
  226. if (f->upvalues[i])
  227. stringmark(f->upvalues[i]);
  228. }
  229. for (i=0; i<f->sizep; i++) /* mark nested protos */
  230. markobject(g, f->p[i]);
  231. for (i=0; i<f->sizelocvars; i++) { /* mark local-variable names */
  232. if (f->locvars[i].varname)
  233. stringmark(f->locvars[i].varname);
  234. }
  235. }
  236. static void traverseclosure (global_State *g, Closure *cl) {
  237. markobject(g, cl->c.env);
  238. if (cl->c.isC) {
  239. int i;
  240. for (i=0; i<cl->c.nupvalues; i++) /* mark its upvalues */
  241. markvalue(g, &cl->c.upvalue[i]);
  242. }
  243. else {
  244. int i;
  245. lua_assert(cl->l.nupvalues == cl->l.p->nups);
  246. markobject(g, cl->l.p);
  247. for (i=0; i<cl->l.nupvalues; i++) /* mark its upvalues */
  248. markobject(g, cl->l.upvals[i]);
  249. }
  250. }
  251. static void checkstacksizes (lua_State *L, StkId max) {
  252. /* should not change the stack when handling overflow or
  253. during an emergency gc cycle */
  254. if (L->size_ci > LUAI_MAXCALLS || G(L)->gckind == KGC_EMERGENCY)
  255. return; /* do not touch the stacks */
  256. else {
  257. int ci_used = cast_int(L->ci - L->base_ci) + 1; /* number of `ci' in use */
  258. int s_used = cast_int(max - L->stack) + 1; /* part of stack in use */
  259. if (2*ci_used < L->size_ci)
  260. luaD_reallocCI(L, 2*ci_used);
  261. if (2*s_used < (L->stacksize - EXTRA_STACK))
  262. luaD_reallocstack(L, 2*s_used);
  263. }
  264. }
  265. static void traversestack (global_State *g, lua_State *l) {
  266. StkId o, lim;
  267. CallInfo *ci;
  268. if (l->stack == NULL || l->base_ci == NULL)
  269. return; /* stack not completely built yet */
  270. markvalue(g, gt(l));
  271. lim = l->top;
  272. for (ci = l->base_ci; ci <= l->ci; ci++) {
  273. lua_assert(ci->top <= l->stack_last);
  274. if (lim < ci->top) lim = ci->top;
  275. }
  276. for (o = l->stack; o < l->top; o++)
  277. markvalue(g, o);
  278. for (; o <= lim; o++)
  279. setnilvalue(o);
  280. checkstacksizes(l, lim);
  281. }
  282. /*
  283. ** traverse one gray object, turning it to black.
  284. ** Returns `quantity' traversed.
  285. */
  286. static l_mem propagatemark (global_State *g) {
  287. GCObject *o = g->gray;
  288. lua_assert(isgray(o));
  289. gray2black(o);
  290. switch (gch(o)->tt) {
  291. case LUA_TTABLE: {
  292. Table *h = gco2t(o);
  293. g->gray = h->gclist;
  294. traversetable(g, h);
  295. return sizeof(Table) + sizeof(TValue) * h->sizearray +
  296. sizeof(Node) * sizenode(h);
  297. }
  298. case LUA_TFUNCTION: {
  299. Closure *cl = gco2cl(o);
  300. g->gray = cl->c.gclist;
  301. traverseclosure(g, cl);
  302. return (cl->c.isC) ? sizeCclosure(cl->c.nupvalues) :
  303. sizeLclosure(cl->l.nupvalues);
  304. }
  305. case LUA_TTHREAD: {
  306. lua_State *th = gco2th(o);
  307. g->gray = th->gclist;
  308. th->gclist = g->grayagain;
  309. g->grayagain = o;
  310. black2gray(o);
  311. traversestack(g, th);
  312. return sizeof(lua_State) + sizeof(TValue) * th->stacksize +
  313. sizeof(CallInfo) * th->size_ci;
  314. }
  315. case LUA_TPROTO: {
  316. Proto *p = gco2p(o);
  317. g->gray = p->gclist;
  318. traverseproto(g, p);
  319. return sizeof(Proto) + sizeof(Instruction) * p->sizecode +
  320. sizeof(Proto *) * p->sizep +
  321. sizeof(TValue) * p->sizek +
  322. sizeof(int) * p->sizelineinfo +
  323. sizeof(LocVar) * p->sizelocvars +
  324. sizeof(TString *) * p->sizeupvalues;
  325. }
  326. default: lua_assert(0); return 0;
  327. }
  328. }
  329. static size_t propagateall (global_State *g) {
  330. size_t m = 0;
  331. while (g->gray) m += propagatemark(g);
  332. return m;
  333. }
  334. static void convergeephemerons (global_State *g) {
  335. int changed;
  336. do {
  337. GCObject *w;
  338. GCObject *next = g->ephemeron;
  339. g->ephemeron = NULL;
  340. changed = 0;
  341. while ((w = next) != NULL) {
  342. next = gco2t(w)->gclist;
  343. if (traverseephemeron(g, gco2t(w))) {
  344. changed = 1;
  345. propagateall(g);
  346. }
  347. }
  348. } while (changed);
  349. }
  350. /*
  351. ** clear collected entries from weaktables
  352. */
  353. static void cleartable (GCObject *l) {
  354. while (l) {
  355. Table *h = gco2t(l);
  356. int i = h->sizearray;
  357. while (i--) {
  358. TValue *o = &h->array[i];
  359. if (iscleared(o, 0)) /* value was collected? */
  360. setnilvalue(o); /* remove value */
  361. }
  362. i = sizenode(h);
  363. while (i--) {
  364. Node *n = gnode(h, i);
  365. if (!ttisnil(gval(n)) && /* non-empty entry? */
  366. (iscleared(key2tval(n), 1) || iscleared(gval(n), 0))) {
  367. setnilvalue(gval(n)); /* remove value ... */
  368. removeentry(n); /* remove entry from table */
  369. }
  370. }
  371. l = h->gclist;
  372. }
  373. }
  374. static void freeobj (lua_State *L, GCObject *o) {
  375. switch (gch(o)->tt) {
  376. case LUA_TPROTO: luaF_freeproto(L, gco2p(o)); break;
  377. case LUA_TFUNCTION: luaF_freeclosure(L, gco2cl(o)); break;
  378. case LUA_TUPVAL: luaF_freeupval(L, gco2uv(o)); break;
  379. case LUA_TTABLE: luaH_free(L, gco2t(o)); break;
  380. case LUA_TTHREAD: luaE_freethread(L, gco2th(o)); break;
  381. case LUA_TUSERDATA: luaM_freemem(L, o, sizeudata(gco2u(o))); break;
  382. case LUA_TSTRING: {
  383. G(L)->strt.nuse--;
  384. luaM_freemem(L, o, sizestring(gco2ts(o)));
  385. break;
  386. }
  387. default: lua_assert(0);
  388. }
  389. }
  390. #define sweepwholelist(L,p) sweeplist(L,p,MAX_LUMEM)
  391. static GCObject **sweeplist (lua_State *L, GCObject **p, lu_mem count) {
  392. GCObject *curr;
  393. global_State *g = G(L);
  394. int deadmask = otherwhite(g);
  395. while ((curr = *p) != NULL && count-- > 0) {
  396. if (ttisthread(gch(curr))) /* sweep open upvalues of each thread */
  397. sweepwholelist(L, &gco2th(curr)->openupval);
  398. if ((gch(curr)->marked ^ WHITEBITS) & deadmask) { /* not dead? */
  399. lua_assert(!isdead(g, curr) || testbit(gch(curr)->marked, FIXEDBIT));
  400. makewhite(g, curr); /* make it white (for next cycle) */
  401. p = &gch(curr)->next;
  402. }
  403. else { /* must erase `curr' */
  404. lua_assert(isdead(g, curr) || deadmask == bitmask(SFIXEDBIT));
  405. *p = gch(curr)->next; /* remove 'curr' from list */
  406. freeobj(L, curr);
  407. }
  408. }
  409. return p;
  410. }
  411. static GCObject **unmarklist (global_State *g, GCObject **p, lu_mem count) {
  412. for (; *p != NULL && count-- > 0; p = &gch(*p)->next) {
  413. lua_assert(ttisuserdata(gch(*p)) && !isdead(g, *p));
  414. makewhite(g, *p);
  415. }
  416. return p;
  417. }
  418. static void checkSizes (lua_State *L) {
  419. global_State *g = G(L);
  420. if (g->strt.nuse < cast(lu_int32, g->strt.size))
  421. luaS_resize(L, 1 << luaO_ceillog2(g->strt.nuse));
  422. luaZ_freebuffer(L, &g->buff);
  423. }
  424. static Udata *udata2finalize (global_State *g) {
  425. GCObject *o = gch(g->tobefnz)->next; /* get first element */
  426. Udata *udata = rawgco2u(o);
  427. /* remove udata from `tobefnz' */
  428. if (o == g->tobefnz) /* last element? */
  429. g->tobefnz = NULL;
  430. else
  431. gch(g->tobefnz)->next = udata->uv.next;
  432. udata->uv.next = g->mainthread->next; /* return it to `root' list */
  433. g->mainthread->next = o;
  434. resetbit(udata->uv.marked, SEPARATED); /* mark it as such */
  435. makewhite(g, o);
  436. return udata;
  437. }
  438. static void GCTM (lua_State *L) {
  439. global_State *g = G(L);
  440. Udata *udata = udata2finalize(g);
  441. const TValue *tm = gfasttm(g, udata->uv.metatable, TM_GC);
  442. if (tm != NULL && ttisfunction(tm)) {
  443. lu_byte oldah = L->allowhook;
  444. lu_mem oldt = g->GCthreshold;
  445. L->allowhook = 0; /* stop debug hooks during GC tag method */
  446. g->GCthreshold = 2*g->totalbytes; /* avoid GC steps */
  447. setobj2s(L, L->top, tm);
  448. setuvalue(L, L->top+1, udata);
  449. L->top += 2;
  450. luaD_call(L, L->top - 2, 0);
  451. L->allowhook = oldah; /* restore hooks */
  452. g->GCthreshold = oldt; /* restore threshold */
  453. }
  454. }
  455. /*
  456. ** Call all GC tag methods
  457. */
  458. void luaC_callGCTM (lua_State *L) {
  459. global_State *g = G(L);
  460. GCObject *last = g->tobefnz;
  461. GCObject *curr;
  462. if (last == NULL) return; /* empty list? */
  463. do {
  464. curr = gch(g->tobefnz)->next; /* element to be collected */
  465. GCTM(L);
  466. } while (curr != last); /* go only until original last */
  467. /* do not finalize new udata created during previous finalizations */
  468. while (g->tobefnz)
  469. udata2finalize(g); /* simply remove them from list */
  470. }
  471. void luaC_freeall (lua_State *L) {
  472. global_State *g = G(L);
  473. int i;
  474. lua_assert(g->tobefnz == NULL);
  475. /* mask to collect all elements */
  476. g->currentwhite = WHITEBITS | bitmask(SFIXEDBIT);
  477. sweepwholelist(L, &g->rootgc);
  478. lua_assert(g->rootgc == obj2gco(L));
  479. sweepwholelist(L, &g->tmudata);
  480. lua_assert(g->tmudata == NULL);
  481. for (i = 0; i < g->strt.size; i++) /* free all string lists */
  482. sweepwholelist(L, &g->strt.hash[i]);
  483. lua_assert(g->strt.nuse == 0);
  484. }
  485. static void markmt (global_State *g) {
  486. int i;
  487. for (i=0; i<NUM_TAGS; i++)
  488. markobject(g, g->mt[i]);
  489. }
  490. static void markbeingfnz (global_State *g) {
  491. GCObject *u = g->tobefnz;
  492. if (u) {
  493. do {
  494. u = gch(u)->next;
  495. lua_assert(testbit(gch(u)->marked, SEPARATED));
  496. lua_assert(!iswhite(u)); /* must be marked, if left from previous GC */
  497. makewhite(g, u);
  498. reallymarkobject(g, u);
  499. } while (u != g->tobefnz);
  500. }
  501. }
  502. /* mark root set */
  503. static void markroot (lua_State *L) {
  504. global_State *g = G(L);
  505. g->gray = NULL;
  506. g->grayagain = NULL;
  507. g->weak = g->ephemeron = g->allweak = NULL;
  508. markobject(g, g->mainthread);
  509. /* make global table be traversed before main stack */
  510. markvalue(g, gt(g->mainthread));
  511. markvalue(g, registry(L));
  512. markmt(g);
  513. markbeingfnz(g); /* mark any finalizing userdata left from previous cycle */
  514. g->gcstate = GCSpropagate;
  515. }
  516. static void remarkupvals (global_State *g) {
  517. UpVal *uv;
  518. for (uv = g->uvhead.u.l.next; uv != &g->uvhead; uv = uv->u.l.next) {
  519. lua_assert(uv->u.l.next->u.l.prev == uv && uv->u.l.prev->u.l.next == uv);
  520. if (isgray(obj2gco(uv)))
  521. markvalue(g, uv->v);
  522. }
  523. }
  524. static void marklistofgrays (global_State *g, GCObject **l) {
  525. lua_assert(g->gray == NULL); /* no grays left */
  526. g->gray = *l; /* now 'l' is new gray list */
  527. *l = NULL;
  528. propagateall(g);
  529. }
  530. static void atomic (lua_State *L) {
  531. global_State *g = G(L);
  532. size_t udsize; /* total size of userdata to be finalized */
  533. /* remark occasional upvalues of (maybe) dead threads */
  534. remarkupvals(g);
  535. /* traverse objects cautch by write barrier and by 'remarkupvals' */
  536. propagateall(g);
  537. /* remark weak tables */
  538. g->gray = g->weak;
  539. g->weak = NULL;
  540. lua_assert(!iswhite(obj2gco(g->mainthread)));
  541. markobject(g, L); /* mark running thread */
  542. markmt(g); /* mark basic metatables (again) */
  543. propagateall(g);
  544. marklistofgrays(g, &g->ephemeron); /* remark ephemeron tables */
  545. marklistofgrays(g, &g->grayagain); /* remark gray again */
  546. convergeephemerons(g);
  547. udsize = luaC_separateudata(L, 0); /* separate userdata to be finalized */
  548. udsize += propagateall(g); /* remark, to propagate `preserveness' */
  549. convergeephemerons(g);
  550. /* remove collected objects from weak tables */
  551. cleartable(g->weak);
  552. cleartable(g->ephemeron);
  553. cleartable(g->allweak);
  554. /* flip current white */
  555. g->currentwhite = cast_byte(otherwhite(g));
  556. g->sweepstrgc = 0;
  557. g->gcstate = GCSsweepstring;
  558. g->estimate = g->totalbytes - udsize; /* first estimate */
  559. }
  560. #define correctestimate(g,s) {lu_mem old = g->totalbytes; s; \
  561. lua_assert(old >= g->totalbytes); g->estimate -= old - g->totalbytes;}
  562. static l_mem singlestep (lua_State *L) {
  563. global_State *g = G(L);
  564. /*lua_checkmemory(L);*/
  565. switch (g->gcstate) {
  566. case GCSpause: {
  567. markroot(L); /* start a new collection */
  568. return 0;
  569. }
  570. case GCSpropagate: {
  571. if (g->gray)
  572. return propagatemark(g);
  573. else { /* no more `gray' objects */
  574. atomic(L); /* finish mark phase */
  575. return 0;
  576. }
  577. }
  578. case GCSsweepstring: {
  579. correctestimate(g, sweepwholelist(L, &g->strt.hash[g->sweepstrgc++]));
  580. if (g->sweepstrgc >= g->strt.size) { /* nothing more to sweep? */
  581. g->sweepgc = &g->tmudata;
  582. g->gcstate = GCSsweeptmu; /* end sweep-string phase */
  583. }
  584. return GCSWEEPCOST;
  585. }
  586. case GCSsweeptmu: {
  587. g->sweepgc = unmarklist(g, g->sweepgc, GCSWEEPMAX);
  588. if (*g->sweepgc == NULL) { /* nothing more to sweep? */
  589. g->sweepgc = &g->rootgc;
  590. g->gcstate = GCSsweep; /* sweep all other objects */
  591. }
  592. return GCSWEEPMAX*GCSWEEPCOST;
  593. }
  594. case GCSsweep: {
  595. correctestimate(g, g->sweepgc = sweeplist(L, g->sweepgc, GCSWEEPMAX));
  596. if (*g->sweepgc == NULL) /* nothing more to sweep? */
  597. g->gcstate = GCSfinalize; /* end sweep phase */
  598. return GCSWEEPMAX*GCSWEEPCOST;
  599. }
  600. case GCSfinalize: {
  601. if (g->tobefnz) {
  602. GCTM(L);
  603. if (g->estimate > GCFINALIZECOST)
  604. g->estimate -= GCFINALIZECOST;
  605. return GCFINALIZECOST;
  606. }
  607. else {
  608. correctestimate(g, checkSizes(L));
  609. g->gcstate = GCSpause; /* end collection */
  610. g->gcdept = 0;
  611. return 0;
  612. }
  613. }
  614. default: lua_assert(0); return 0;
  615. }
  616. }
  617. void luaC_step (lua_State *L) {
  618. global_State *g = G(L);
  619. l_mem lim = (GCSTEPSIZE/100) * g->gcstepmul;
  620. lua_assert(g->gckind == KGC_NORMAL);
  621. if (lim == 0)
  622. lim = (MAX_LUMEM-1)/2; /* no limit */
  623. g->gcdept += g->totalbytes - g->GCthreshold;
  624. do {
  625. lim -= singlestep(L);
  626. if (g->gcstate == GCSpause)
  627. break;
  628. } while (lim > 0);
  629. if (g->gcstate != GCSpause) {
  630. if (g->gcdept < GCSTEPSIZE)
  631. g->GCthreshold = g->totalbytes + GCSTEPSIZE; /* - lim/g->gcstepmul;*/
  632. else {
  633. g->gcdept -= GCSTEPSIZE;
  634. g->GCthreshold = g->totalbytes;
  635. }
  636. }
  637. else {
  638. lua_assert(g->totalbytes >= g->estimate);
  639. setthreshold(g);
  640. }
  641. }
  642. void luaC_fullgc (lua_State *L, int isemergency) {
  643. global_State *g = G(L);
  644. lua_assert(g->gckind == KGC_NORMAL);
  645. g->gckind = isemergency ? KGC_EMERGENCY : KGC_FORCED;
  646. if (g->gcstate <= GCSpropagate) {
  647. /* reset other collector lists */
  648. g->gray = NULL;
  649. g->grayagain = NULL;
  650. g->weak = g->ephemeron = g->allweak = NULL;
  651. g->sweepstrgc = 0;
  652. g->gcstate = GCSsweepstring;
  653. }
  654. lua_assert(g->gcstate != GCSpause && g->gcstate != GCSpropagate);
  655. /* finish any pending sweep phase */
  656. while (g->gcstate != GCSfinalize) {
  657. lua_assert(issweep(g));
  658. singlestep(L);
  659. }
  660. markroot(L);
  661. /* run collector up to finalizers */
  662. while (g->gcstate != GCSfinalize)
  663. singlestep(L);
  664. g->gckind = KGC_NORMAL;
  665. if (!isemergency) { /* do not run finalizers during emergency GC */
  666. while (g->gcstate != GCSpause)
  667. singlestep(L);
  668. }
  669. setthreshold(g);
  670. }
  671. void luaC_barrierf (lua_State *L, GCObject *o, GCObject *v) {
  672. global_State *g = G(L);
  673. lua_assert(isblack(o) && iswhite(v) && !isdead(g, v) && !isdead(g, o));
  674. lua_assert(g->gcstate != GCSfinalize && g->gcstate != GCSpause);
  675. lua_assert(ttype(gch(o)) != LUA_TTABLE);
  676. /* must keep invariant? */
  677. if (g->gcstate == GCSpropagate)
  678. reallymarkobject(g, v); /* restore invariant */
  679. else /* don't mind */
  680. makewhite(g, o); /* mark as white just to avoid other barriers */
  681. }
  682. void luaC_barrierback (lua_State *L, Table *t) {
  683. global_State *g = G(L);
  684. GCObject *o = obj2gco(t);
  685. lua_assert(isblack(o) && !isdead(g, o));
  686. lua_assert(g->gcstate != GCSfinalize && g->gcstate != GCSpause);
  687. black2gray(o); /* make table gray (again) */
  688. t->gclist = g->grayagain;
  689. g->grayagain = o;
  690. }
  691. void luaC_link (lua_State *L, GCObject *o, lu_byte tt) {
  692. global_State *g = G(L);
  693. gch(o)->marked = luaC_white(g);
  694. gch(o)->tt = tt;
  695. gch(o)->next = g->rootgc;
  696. g->rootgc = o;
  697. }
  698. void luaC_linkupval (lua_State *L, UpVal *uv) {
  699. global_State *g = G(L);
  700. GCObject *o = obj2gco(uv);
  701. gch(o)->next = g->rootgc; /* link upvalue into `rootgc' list */
  702. g->rootgc = o;
  703. if (isgray(o)) {
  704. if (g->gcstate == GCSpropagate) {
  705. gray2black(o); /* closed upvalues need barrier */
  706. luaC_barrier(L, uv, uv->v);
  707. }
  708. else { /* sweep phase: sweep it (turning it into white) */
  709. makewhite(g, o);
  710. lua_assert(g->gcstate != GCSfinalize && g->gcstate != GCSpause);
  711. }
  712. }
  713. }
  714. void luaC_checkfinalizer (lua_State *L, Udata *u) {
  715. global_State *g = G(L);
  716. if (testbit(u->uv.marked, SEPARATED) || /* userdata is already separated... */
  717. isfinalized(&u->uv) || /* ... or is finalized... */
  718. gfasttm(g, u->uv.metatable, TM_GC) == NULL) /* or has no finalization? */
  719. return; /* nothing to be done */
  720. else { /* move 'u' from root list to tobefnz list */
  721. GCObject **p;
  722. for (p = &g->rootgc; *p != obj2gco(u); p = &gch(*p)->next)
  723. lua_assert(*p != NULL); /* 'u' must be in this list */
  724. *p = u->uv.next; /* remove 'u' from root list */
  725. u->uv.next = g->tmudata; /* link it in tobefnz list */
  726. g->tmudata = obj2gco(u);
  727. l_setbit(u->uv.marked, SEPARATED); /* mark it as such */
  728. }
  729. }