mirror of https://github.com/lua/lua
new function xpcall
This commit is contained in:
parent
390fc99a5c
commit
2dcc31574f
49
lbaselib.c
49
lbaselib.c
|
@ -1,5 +1,5 @@
|
|||
/*
|
||||
** $Id: lbaselib.c,v 1.90 2002/07/04 17:58:02 roberto Exp roberto $
|
||||
** $Id: lbaselib.c,v 1.91 2002/07/17 16:25:13 roberto Exp roberto $
|
||||
** Basic library
|
||||
** See Copyright Notice in lua.h
|
||||
*/
|
||||
|
@ -214,10 +214,9 @@ static int luaB_ipairs (lua_State *L) {
|
|||
static int passresults (lua_State *L, int status) {
|
||||
if (status == 0) return 1;
|
||||
else {
|
||||
int numres = (status == LUA_ERRRUN) ? 3 : 2;
|
||||
lua_pushnil(L);
|
||||
lua_insert(L, -numres);
|
||||
return numres;
|
||||
lua_insert(L, -2);
|
||||
return 2;
|
||||
}
|
||||
}
|
||||
|
||||
|
@ -278,13 +277,33 @@ static int luaB_pcall (lua_State *L) {
|
|||
int status;
|
||||
luaL_check_any(L, 1);
|
||||
status = lua_pcall(L, lua_gettop(L) - 1, LUA_MULTRET);
|
||||
if (status != 0)
|
||||
return passresults(L, status);
|
||||
else {
|
||||
lua_pushboolean(L, 1);
|
||||
lua_insert(L, 1);
|
||||
return lua_gettop(L); /* return `true' + all results */
|
||||
if (status) /* error? */
|
||||
lua_pcallreset(L); /* reset error handler */
|
||||
lua_pushboolean(L, (status == 0));
|
||||
lua_insert(L, 1);
|
||||
return lua_gettop(L); /* return status + all results */
|
||||
}
|
||||
|
||||
|
||||
static int luaB_xpcall (lua_State *L) {
|
||||
int status;
|
||||
int ref;
|
||||
luaL_check_any(L, 2);
|
||||
lua_settop(L, 2);
|
||||
ref = lua_ref(L, 1); /* save error function */
|
||||
status = lua_pcall(L, 0, LUA_MULTRET);
|
||||
if (status) { /* error? */
|
||||
if (status == LUA_ERRRUN) { /* run-time error? */
|
||||
lua_getref(L, ref); /* get error function */
|
||||
lua_pushvalue(L, -2); /* error message */
|
||||
lua_call(L, 1, 1); /* call error function */
|
||||
}
|
||||
lua_pcallreset(L); /* reset error handler */
|
||||
}
|
||||
lua_unref(L, ref); /* free reference */
|
||||
lua_pushboolean(L, (status == 0));
|
||||
lua_insert(L, 1);
|
||||
return lua_gettop(L); /* return status + all results */
|
||||
}
|
||||
|
||||
|
||||
|
@ -466,6 +485,7 @@ static const luaL_reg base_funcs[] = {
|
|||
{"rawget", luaB_rawget},
|
||||
{"rawset", luaB_rawset},
|
||||
{"pcall", luaB_pcall},
|
||||
{"xpcall", luaB_xpcall},
|
||||
{"collectgarbage", luaB_collectgarbage},
|
||||
{"gcinfo", luaB_gcinfo},
|
||||
{"loadfile", luaB_loadfile},
|
||||
|
@ -488,15 +508,8 @@ static int luaB_resume (lua_State *L) {
|
|||
int status;
|
||||
lua_settop(L, 0);
|
||||
status = lua_resume(L, co);
|
||||
if (status != 0) {
|
||||
if (status == LUA_ERRRUN) {
|
||||
if (lua_isstring(L, -1) && lua_isstring(L, -2))
|
||||
lua_concat(L, 2);
|
||||
else
|
||||
lua_pop(L, 1);
|
||||
}
|
||||
if (status != 0)
|
||||
return lua_error(L);
|
||||
}
|
||||
return lua_gettop(L);
|
||||
}
|
||||
|
||||
|
|
Loading…
Reference in New Issue