/*
** open parts that may cause memory-allocation errors
*/
static void f_luaopen (lua_State *L, void *ud) {
/* create a new global state */
global_State *g = luaM_new(NULL, global_State);
UNUSED(ud);
if (g == NULL) luaD_throw(L, LUA_ERRMEM);
L->l_G = g;
g->mainthread = L;
g->GCthreshold = 0; /* mark it as unfinished state */
g->strt.size = 0;
g->strt.nuse = 0;
g->strt.hash = NULL;
setnilvalue(defaultmeta(L));
setnilvalue(registry(L));
luaZ_initbuffer(L, &g->buff);
g->panic = default_panic;
g->rootgc = NULL;
g->rootudata = NULL;
g->tmudata = NULL;
setnilvalue(gkey(g->dummynode));
setnilvalue(gval(g->dummynode));
g->dummynode->next = NULL;
g->nblocks = sizeof(lua_State) + sizeof(global_State);
stack_init(L, L); /* init stack */
/* create default meta table with a dummy table, and then close the loop */
defaultmeta(L)->tt = LUA_TTABLE;
sethvalue(defaultmeta(L), luaH_new(L, 0, 0));
hvalue(defaultmeta(L))->metatable = hvalue(defaultmeta(L));
sethvalue(gt(L), luaH_new(L, 0, 4)); /* table of globals */
sethvalue(registry(L), luaH_new(L, 4, 4)); /* registry */
luaS_resize(L, MINSTRTABSIZE); /* initial size of string table */
luaT_init(L);
luaX_init(L);
luaS_fix(luaS_newliteral(L, MEMERRMSG));
g->GCthreshold = 4*G(L)->nblocks;
}
LUA_API int lua_setmetatable (lua_State *L, int objindex) {
TObject *obj, *mt;
int res = 1;
lua_lock(L);
api_checknelems(L, 1);
obj = luaA_index(L, objindex);
mt = (!ttisnil(L->top - 1)) ? L->top - 1 : defaultmeta(L);
api_check(L, ttistable(mt));
switch (ttype(obj)) {
case LUA_TTABLE: {
hvalue(obj)->metatable = hvalue(mt); /* write barrier */
break;
}
case LUA_TUSERDATA: {
uvalue(obj)->uv.metatable = hvalue(mt); /* write barrier */
break;
}
default: {
res = 0; /* cannot set */
break;
}
}
L->top--;
lua_unlock(L);
return res;
}
LUA_API int lua_getmetatable (lua_State *L, int objindex) {
const TObject *obj;
Table *mt = NULL;
int res;
lua_lock(L);
obj = luaA_indexAcceptable(L, objindex);
if (obj != NULL) {
switch (ttype(obj)) {
case LUA_TTABLE:
mt = hvalue(obj)->metatable;
break;
case LUA_TUSERDATA:
mt = uvalue(obj)->uv.metatable;
break;
}
}
if (mt == NULL || mt == hvalue(defaultmeta(L)))
res = 0;
else {
sethvalue(L->top, mt);
api_incr_top(L);
res = 1;
}
lua_unlock(L);
return res;
}
/* mark root set */
static void markroot (GCState *st, lua_State *L) {
global_State *g = st->g;
markobject(st, defaultmeta(L));
markobject(st, registry(L));
traversestack(st, g->mainthread);
if (L != g->mainthread) /* another thread is running? */
markvalue(st, L); /* cannot collect it */
}
LUA_API int lua_setmetatable (lua_State *L, int objindex) {
TObject *obj, *mt;
int res = 1;
lua_lock(L);
api_checknelems(L, 1);
obj = luaA_index(L, objindex);
mt = (!ttisnil(L->top - 1)) ? L->top - 1 : defaultmeta(L);
api_check(L, ttistable(mt));
switch (ttype(obj)) {
case LUA_TTABLE: {
#if LUA_REFCOUNT
__AddRefDirect(hvalue(mt));
__ReleaseDirect(L, hvalue(obj)->metatable);
#endif LUA_REFCOUNT
hvalue(obj)->metatable = hvalue(mt); /* write barrier */
break;
}
case LUA_TUSERDATA: {
#if LUA_REFCOUNT
__AddRefDirect(hvalue(mt));
__ReleaseDirect(L, &uvalue(obj)->uv.metatable);
#endif LUA_REFCOUNT
uvalue(obj)->uv.metatable = hvalue(mt); /* write barrier */
break;
}
default: {
res = 0; /* cannot set */
break;
}
}
#if LUA_REFCOUNT
setnilvalue(--L->top);
#else !LUA_REFCOUNT
L->top--;
#endif LUA_REFCOUNT
lua_unlock(L);
return res;
}
void luaC_callGCTM (lua_State *L) {
lu_byte oldah = L->allowhook;
L->allowhook = 0; /* stop debug hooks during GC tag methods */
L->top++; /* reserve space to keep udata while runs its gc method */
#if LUA_REFCOUNT
while (G(L)->tmudata_head.next != (GCObject*)&G(L)->tmudata_tail) {
GCObject *o = G(L)->tmudata_head.next;
Udata *udata = gcotou(o);
G(L)->tmudata_head.next = udata->uv.next; /* remove udata from `tmudata' */
udata->uv.prev = (GCObject*)&G(L)->rootudata_head;
udata->uv.next = G(L)->rootudata_head.next;
if (udata->uv.next)
udata->uv.next->uv.prev = o;
G(L)->rootudata_head.next = o;
#else !LUA_REFCOUNT
while (G(L)->tmudata != NULL) {
GCObject *o = G(L)->tmudata;
Udata *udata = gcotou(o);
G(L)->tmudata = udata->uv.next; /* remove udata from `tmudata' */
udata->uv.next = G(L)->rootudata; /* return it to `root' list */
G(L)->rootudata = o;
#endif LUA_REFCOUNT
setuvalue(L->top - 1, udata); /* keep a reference to it */
unmark(o);
markfinalized(udata);
do1gcTM(L, udata);
}
L->top--;
L->allowhook = oldah; /* restore hooks */
}
void luaC_sweep (lua_State *L, int all) {
if (all) all = 256; /* larger than any mark */
#if LUA_REFCOUNT
sweeplist(L, &G(L)->rootudata_head.next, (GCObject*)&G(L)->rootudata_tail, all);
#else !LUA_REFCOUNT
sweeplist(L, &G(L)->rootudata, NULL, all);
#endif LUA_REFCOUNT
sweepstrings(L, all);
#if LUA_REFCOUNT
sweeplist(L, &G(L)->rootgc_head.next, (GCObject*)&G(L)->rootgc_tail, all);
#else !LUA_REFCOUNT
sweeplist(L, &G(L)->rootgc, NULL, all);
#endif LUA_REFCOUNT
}
/* mark root set */
static void markroot (GCState *st, lua_State *L) {
int i;
global_State *g = st->g;
markobject(st, defaultmeta(L));
for (i = 0; i < LUA_NTYPES; i++)
{
markobject(st, defaultmetatypes(L, i));
}
markobject(st, registry(L));
traversestack(st, g->mainthread);
if (L != g->mainthread) /* another thread is running? */
markvalue(st, L); /* cannot collect it */
if (G(L)->userGCFunction)
{
st->L = g->mainthread;
G(L)->userGCFunction(st);
}
}
/*
** open parts that may cause memory-allocation errors
*/
static void f_luaopen (lua_State *L, void *ud) {
int i;
global_State globalState;
lua_State luaState;
global_State *g;
#ifdef _DEBUG
luaState.allocName = "Lua_global_State";
#endif _DEBUG
luaState.l_G = &globalState;
globalState.reallocFunc = luaHelper_Realloc;
globalState.freeFunc = luaHelper_Free;
globalState.memData = luaHelper_memData;
globalState.nblocks = sizeof(lua_State) + sizeof(global_State); // Bogus.
/* create a new global state */
g = luaM_new(&luaState, global_State);
UNUSED(ud);
if (g == NULL) luaD_throw(L, LUA_ERRMEM);
L->l_G = g;
g->mainthread = L;
g->GCthreshold = 0; /* mark it as unfinished state */
g->strt.size = 0;
g->strt.nuse = 0;
g->strt.hash = NULL;
setnilvalue2n(defaultmeta(L));
setnilvalue2n(registry(L));
luaZ_initbuffer(L, &g->buff);
g->panic = default_panic;
#if !LUA_REFCOUNT
g->rootgc = NULL;
g->rootudata = NULL;
g->tmudata = NULL;
#else LUA_REFCOUNT
g->rootgc_head.next = (GCObject*)&g->rootgc_tail;
g->rootgc_head.prev = NULL;
g->rootgc_tail.next = NULL;
g->rootgc_tail.prev = (GCObject*)&g->rootgc_head;
g->rootgc_head.tt = LUA_TNIL;
g->rootgc_head.marked = 0;
g->rootgc_head.ref = 0;
g->rootgc_tail.tt = LUA_TNIL;
g->rootgc_tail.marked = 0;
g->rootgc_tail.ref = 0;
g->rootudata_head.next = (GCObject*)&g->rootudata_tail;
g->rootudata_head.prev = NULL;
g->rootudata_tail.next = NULL;
g->rootudata_tail.prev = (GCObject*)&g->rootudata_head;
g->rootudata_head.tt = LUA_TNIL;
g->rootudata_head.marked = 0;
g->rootudata_head.ref = 0;
g->rootudata_tail.tt = LUA_TNIL;
g->rootudata_tail.marked = 0;
g->rootudata_tail.ref = 0;
g->tmudata_head.next = (GCObject*)&g->tmudata_tail;
g->tmudata_head.prev = NULL;
g->tmudata_tail.next = NULL;
g->tmudata_tail.prev = (GCObject*)&g->tmudata_head;
g->tmudata_head.tt = LUA_TNIL;
g->tmudata_head.marked = 0;
g->tmudata_head.ref = 0;
g->tmudata_tail.tt = LUA_TNIL;
g->tmudata_tail.marked = 0;
g->tmudata_tail.ref = 0;
#endif LUA_REFCOUNT
setnilvalue2n(gkey(g->dummynode));
setnilvalue2n(gval(g->dummynode));
g->dummynode->next = NULL;
g->nblocks = sizeof(lua_State) + sizeof(global_State);
g->reallocFunc = luaHelper_Realloc;
g->freeFunc = luaHelper_Free;
g->memData = luaHelper_memData;
g->fatalErrorFunc = defaultFatalErrorFunc;
#ifdef LUA_MTSUPPORT
g->lockData = NULL;
g->lockFunc = NULL;
g->unlockFunc = NULL;
#endif LUA_MTSUPPORT
g->userGCFunction = NULL;
g->globalUserData = NULL;
stack_init(L, L); /* init stack */
for (i = 0; i < LUA_NTYPES; i++)
{
defaultmetatypes(L, i)->value.gc = NULL;
}
/* create default meta table with a dummy table, and then close the loop */
defaultmeta(L)->tt = LUA_TNUMBER;
defaultmeta(L)->value.gc = NULL;
sethvalue2n(defaultmeta(L), luaH_new(L, 0, 0));
__AddRefDirect(hvalue(defaultmeta(L)));
hvalue(defaultmeta(L))->metatable = hvalue(defaultmeta(L));
__AddRefDirect(hvalue(defaultmeta(L))->metatable);
/* build meta tables */
for (i = 0; i < LUA_NTYPES; i++)
{
luaM_setname(L, "Lua_defaultMetaTypes");
sethvalue2n(defaultmetatypes(L, i), luaH_new(L, 0, 0));
hvalue(defaultmetatypes(L, i))->metatable = hvalue(defaultmeta(L));
}
luaM_setname(L, "Lua_Globals");
sethvalue(gt(L), luaH_new(L, 0, 4)); /* table of globals */
__AddRefDirect(hvalue(gt(L)));
luaM_setname(L, "Lua_Registry");
sethvalue(registry(L), luaH_new(L, 4, 4)); /* registry */
__AddRef(registry(L));
g->minimumstrings = lua_minimumnumstrings;
luaS_resize(L, MINSTRTABSIZE); /* initial size of string table */
luaT_init(L);
luaX_init(L);
luaS_fix(luaS_newliteral(L, MEMERRMSG));
g->GCthreshold = 4*G(L)->nblocks;
luaZ_openspace(L, &g->buff, lua_minimumauxspace);
}