nik/lush
1
1
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

distinct functions to create/destroy states and threads

Browse Source
This commit is contained in:
Roberto Ierusalimschy
2002-01-11 18:26:52 -02:00
parent b7ae43d457
commit d56d4cf776
7 changed files with 144 additions and 97 deletions
Download Patch File Download Diff File Expand all files Collapse all files

203
lstate.c
View File

@@ -26,7 +26,26 @@ struct Sopen {
};
static void close_state (lua_State *L, lua_State *OL);
static void close_state (lua_State *L);
static void stack_init (lua_State *L, lua_State *OL, int stacksize) {
if (stacksize == 0)
stacksize = DEFAULT_STACK_SIZE;
else
stacksize += LUA_MINSTACK;
stacksize += EXTRA_STACK;
L->stack = luaM_newvector(OL, stacksize, TObject);
L->stacksize = stacksize;
L->top = L->stack + RESERVED_STACK_PREFIX;
L->stack_last = L->stack+(L->stacksize-EXTRA_STACK)-1;
L->base_ci = luaM_newvector(OL, 10, CallInfo);
L->ci = L->base_ci;
L->ci->base = L->top;
L->ci->savedpc = NULL;
L->size_ci = 10;
L->end_ci = L->base_ci + L->size_ci;
}
/*
@@ -34,93 +53,102 @@ static void close_state (lua_State *L, lua_State *OL);
*/
static void f_luaopen (lua_State *L, void *ud) {
struct Sopen *so = cast(struct Sopen *, ud);
if (so->stacksize == 0)
so->stacksize = DEFAULT_STACK_SIZE;
else
so->stacksize += LUA_MINSTACK;
if (so->L != NULL) { /* shared global state? */
L->_G = G(so->L);
so->L->next->previous = L; /* insert L into linked list */
L->next = so->L->next;
so->L->next = L;
L->previous = so->L;
luaD_init(L, so->stacksize); /* init stack */
setobj(defaultet(L), defaultet(so->L)); /* share default event table */
setobj(gt(L), gt(so->L)); /* share table of globals */
setobj(registry(L), registry(so->L)); /* share registry */
}
else { /* create a new global state */
L->_G = luaM_new(L, global_State);
G(L)->strt.size = 0;
G(L)->strt.nuse = 0;
G(L)->strt.hash = NULL;
G(L)->Mbuffer = NULL;
G(L)->Mbuffsize = 0;
G(L)->rootproto = NULL;
G(L)->rootcl = NULL;
G(L)->roottable = NULL;
G(L)->rootupval = NULL;
G(L)->rootudata = NULL;
G(L)->tmudata = NULL;
G(L)->nblocks = sizeof(lua_State) + sizeof(global_State);
luaD_init(L, so->stacksize); /* init stack */
/* create default event table with a dummy table, and then close the loop */
sethvalue(defaultet(L), NULL);
sethvalue(defaultet(L), luaH_new(L, 0, 4));
hvalue(defaultet(L))->eventtable = hvalue(defaultet(L));
sethvalue(gt(L), luaH_new(L, 0, 4)); /* table of globals */
sethvalue(registry(L), luaH_new(L, 0, 0)); /* registry */
luaS_resize(L, 4); /* initial size of string table */
luaT_init(L);
luaX_init(L);
G(L)->GCthreshold = 4*G(L)->nblocks;
}
/* create a new global state */
L->_G = luaM_new(L, global_State);
G(L)->strt.size = 0;
G(L)->strt.nuse = 0;
G(L)->strt.hash = NULL;
G(L)->Mbuffer = NULL;
G(L)->Mbuffsize = 0;
G(L)->rootproto = NULL;
G(L)->rootcl = NULL;
G(L)->roottable = NULL;
G(L)->rootupval = NULL;
G(L)->rootudata = NULL;
G(L)->tmudata = NULL;
G(L)->nblocks = sizeof(lua_State) + sizeof(global_State);
stack_init(L, L, so->stacksize); /* init stack */
/* create default event table with a dummy table, and then close the loop */
sethvalue(defaultet(L), NULL);
sethvalue(defaultet(L), luaH_new(L, 0, 4));
hvalue(defaultet(L))->eventtable = hvalue(defaultet(L));
sethvalue(gt(L), luaH_new(L, 0, 4)); /* table of globals */
sethvalue(registry(L), luaH_new(L, 0, 0)); /* registry */
luaS_resize(L, 4); /* initial size of string table */
luaT_init(L);
luaX_init(L);
G(L)->GCthreshold = 4*G(L)->nblocks;
}
static void preinit_state (lua_State *L) {
L->stack = NULL;
L->stacksize = 0;
L->errorJmp = NULL;
L->callhook = NULL;
L->linehook = NULL;
L->openupval = NULL;
L->size_ci = 0;
L->base_ci = NULL;
L->allowhooks = 1;
}
LUA_API lua_State *lua_newthread (lua_State *OL, int stacksize) {
struct Sopen so;
lua_State *L;
if (OL) lua_lock(OL);
lua_lock(OL);
L = luaM_new(OL, lua_State);
if (L) { /* allocation OK? */
L->_G = NULL;
L->stack = NULL;
L->stacksize = 0;
L->errorJmp = NULL;
L->callhook = NULL;
L->linehook = NULL;
L->openupval = NULL;
L->size_ci = 0;
L->base_ci = NULL;
L->allowhooks = 1;
L->next = L->previous = L;
so.stacksize = stacksize;
so.L = OL;
if (luaD_runprotected(L, f_luaopen, &so) != 0) {
/* memory allocation error: free partial state */
close_state(L, OL);
L = NULL;
}
}
if (OL) lua_unlock(OL);
preinit_state(L);
L->_G = OL->_G;
OL->next->previous = L; /* insert L into linked list */
L->next = OL->next;
OL->next = L;
L->previous = OL;
stack_init(L, OL, stacksize); /* init stack */
setobj(defaultet(L), defaultet(OL)); /* share default event table */
setobj(gt(L), gt(OL)); /* share table of globals */
setobj(registry(L), registry(OL)); /* share registry */
lua_unlock(OL);
lua_userstateopen(L);
return L;
}
static void close_state (lua_State *L, lua_State *OL) {
LUA_API lua_State *lua_open (int stacksize) {
struct Sopen so;
lua_State *L;
L = luaM_new(NULL, lua_State);
if (L) { /* allocation OK? */
preinit_state(L);
L->_G = NULL;
L->next = L->previous = L;
so.stacksize = stacksize;
so.L = NULL;
if (luaD_runprotected(L, f_luaopen, &so) != 0) {
/* memory allocation error: free partial state */
close_state(L);
L = NULL;
}
}
lua_userstateopen(L);
return L;
}
void luaE_closethread (lua_State *OL, lua_State *L) {
luaF_close(L, L->stack); /* close all upvalues for this thread */
lua_assert(L->openupval == NULL);
if (OL != NULL) { /* are there other threads? */
lua_assert(L->previous != L);
L->previous->next = L->next;
L->next->previous = L->previous;
}
else if (G(L)) { /* last thread; close global state */
if (G(L)->rootudata) /* (avoid problems with incomplete states) */
luaC_callallgcTM(L); /* call GC tag methods for all udata */
lua_assert(G(L)->tmudata == NULL);
L->previous->next = L->next;
L->next->previous = L->previous;
luaM_freearray(OL, L->base_ci, L->size_ci, CallInfo);
luaM_freearray(OL, L->stack, L->stacksize, TObject);
luaM_freelem(OL, L);
}
static void close_state (lua_State *L) {
luaF_close(L, L->stack); /* close all upvalues for this thread */
if (G(L)) { /* close global state */
luaC_collect(L, 1); /* collect all elements */
lua_assert(G(L)->rootproto == NULL);
lua_assert(G(L)->rootudata == NULL);
@@ -131,21 +159,24 @@ static void close_state (lua_State *L, lua_State *OL) {
luaM_freearray(L, G(L)->Mbuffer, G(L)->Mbuffsize, char);
luaM_freelem(NULL, L->_G);
}
luaM_freearray(OL, L->base_ci, L->size_ci, CallInfo);
luaM_freearray(OL, L->stack, L->stacksize, TObject);
luaM_freelem(OL, L);
luaE_closethread(NULL, L);
}
LUA_API void lua_closethread (lua_State *L, lua_State *thread) {
if (L == thread) lua_error(L, "cannot close only thread of a state");
lua_lock(L);
luaE_closethread(L, thread);
lua_unlock(L);
}
LUA_API void lua_close (lua_State *L) {
lua_State *OL;
lua_assert(L != lua_state || lua_gettop(L) == 0);
lua_lock(L);
OL = L->next; /* any surviving thread (if there is one) */
if (OL == L) OL = NULL; /* no surviving threads */
close_state(L, OL);
if (OL) lua_unlock(OL); /* cannot unlock over a freed state */
lua_assert(L != lua_state || memdebug_numblocks == 0);
lua_assert(L != lua_state || memdebug_total == 0);
luaC_callallgcTM(L); /* call GC tag methods for all udata */
lua_assert(G(L)->tmudata == NULL);
while (L->next != L) /* then, close all other threads */
luaE_closethread(L, L->next);
close_state(L);
}