lgc.c 22 KB

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