lgc.c 23 KB

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