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

errors in 'resume' should be all protected

Browse Source
This commit is contained in:
Roberto Ierusalimschy
2009-07-08 13:06:07 -03:00
parent 0c258c8492
commit aabe3ddbf4
1 changed files with 9 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
lbaselib.c
View File

@@ -1,5 +1,5 @@
/* /*
** $Id: lbaselib.c,v 1.214 2009/03/23 14:26:12 roberto Exp roberto $ ** $Id: lbaselib.c,v 1.215 2009/04/08 18:04:33 roberto Exp roberto $
** Basic library ** Basic library
** See Copyright Notice in lua.h ** See Copyright Notice in lua.h
*/ */
@@ -512,8 +512,10 @@ static const luaL_Reg base_funcs[] = {
static int auxresume (lua_State *L, lua_State *co, int narg) { static int auxresume (lua_State *L, lua_State *co, int narg) {
int status; int status;
if (!lua_checkstack(co, narg)) if (!lua_checkstack(co, narg)) {
return luaL_error(L, "too many arguments to resume"); lua_pushliteral(L, "too many arguments to resume");
return -1; /* error flag */
}
if (lua_status(co) == LUA_OK && lua_gettop(co) == 0) { if (lua_status(co) == LUA_OK && lua_gettop(co) == 0) {
lua_pushliteral(L, "cannot resume dead coroutine"); lua_pushliteral(L, "cannot resume dead coroutine");
return -1; /* error flag */ return -1; /* error flag */
@@ -522,8 +524,10 @@ static int auxresume (lua_State *L, lua_State *co, int narg) {
status = lua_resume(co, narg); status = lua_resume(co, narg);
if (status == LUA_OK || status == LUA_YIELD) { if (status == LUA_OK || status == LUA_YIELD) {
int nres = lua_gettop(co); int nres = lua_gettop(co);
if (!lua_checkstack(L, nres + 1)) if (!lua_checkstack(L, nres + 1)) {
return luaL_error(L, "too many results to resume"); lua_pushliteral(L, "too many results to resume");
return -1; /* error flag */
}
lua_xmove(co, L, nres); /* move yielded values */ lua_xmove(co, L, nres); /* move yielded values */
return nres; return nres;
} }