123 lines
3.3 KiB
C
123 lines
3.3 KiB
C
/*=========================================================================*\
|
|
* Simple exception support
|
|
* LuaSocket toolkit
|
|
\*=========================================================================*/
|
|
#include <stdio.h>
|
|
|
|
#include "lua.h"
|
|
#include "lauxlib.h"
|
|
#include "compat.h"
|
|
|
|
#include "except.h"
|
|
|
|
#if LUA_VERSION_NUM < 502
|
|
#define lua_pcallk(L, na, nr, err, ctx, cont) \
|
|
((void)ctx,(void)cont,lua_pcall(L, na, nr, err))
|
|
#endif
|
|
|
|
#if LUA_VERSION_NUM < 503
|
|
typedef int lua_KContext;
|
|
#endif
|
|
|
|
/*=========================================================================*\
|
|
* Internal function prototypes.
|
|
\*=========================================================================*/
|
|
static int global_protect(lua_State *L);
|
|
static int global_newtry(lua_State *L);
|
|
static int protected_(lua_State *L);
|
|
static int finalize(lua_State *L);
|
|
static int do_nothing(lua_State *L);
|
|
|
|
/* except functions */
|
|
static luaL_Reg func[] = {
|
|
{"newtry", global_newtry},
|
|
{"protect", global_protect},
|
|
{NULL, NULL}
|
|
};
|
|
|
|
/*-------------------------------------------------------------------------*\
|
|
* Try factory
|
|
\*-------------------------------------------------------------------------*/
|
|
static void wrap(lua_State *L) {
|
|
lua_newtable(L);
|
|
lua_pushnumber(L, 1);
|
|
lua_pushvalue(L, -3);
|
|
lua_settable(L, -3);
|
|
lua_insert(L, -2);
|
|
lua_pop(L, 1);
|
|
}
|
|
|
|
static int finalize(lua_State *L) {
|
|
if (!lua_toboolean(L, 1)) {
|
|
lua_pushvalue(L, lua_upvalueindex(1));
|
|
lua_pcall(L, 0, 0, 0);
|
|
lua_settop(L, 2);
|
|
wrap(L);
|
|
lua_error(L);
|
|
return 0;
|
|
} else return lua_gettop(L);
|
|
}
|
|
|
|
static int do_nothing(lua_State *L) {
|
|
(void) L;
|
|
return 0;
|
|
}
|
|
|
|
static int global_newtry(lua_State *L) {
|
|
lua_settop(L, 1);
|
|
if (lua_isnil(L, 1)) lua_pushcfunction(L, do_nothing);
|
|
lua_pushcclosure(L, finalize, 1);
|
|
return 1;
|
|
}
|
|
|
|
/*-------------------------------------------------------------------------*\
|
|
* Protect factory
|
|
\*-------------------------------------------------------------------------*/
|
|
static int unwrap(lua_State *L) {
|
|
if (lua_istable(L, -1)) {
|
|
lua_pushnumber(L, 1);
|
|
lua_gettable(L, -2);
|
|
lua_pushnil(L);
|
|
lua_insert(L, -2);
|
|
return 1;
|
|
} else return 0;
|
|
}
|
|
|
|
static int protected_finish(lua_State *L, int status, lua_KContext ctx) {
|
|
(void)ctx;
|
|
if (status != 0 && status != LUA_YIELD) {
|
|
if (unwrap(L)) return 2;
|
|
else return lua_error(L);
|
|
} else return lua_gettop(L);
|
|
}
|
|
|
|
#if LUA_VERSION_NUM == 502
|
|
static int protected_cont(lua_State *L) {
|
|
int ctx = 0;
|
|
int status = lua_getctx(L, &ctx);
|
|
return protected_finish(L, status, ctx);
|
|
}
|
|
#else
|
|
#define protected_cont protected_finish
|
|
#endif
|
|
|
|
static int protected_(lua_State *L) {
|
|
int status;
|
|
lua_pushvalue(L, lua_upvalueindex(1));
|
|
lua_insert(L, 1);
|
|
status = lua_pcallk(L, lua_gettop(L) - 1, LUA_MULTRET, 0, 0, protected_cont);
|
|
return protected_finish(L, status, 0);
|
|
}
|
|
|
|
static int global_protect(lua_State *L) {
|
|
lua_pushcclosure(L, protected_, 1);
|
|
return 1;
|
|
}
|
|
|
|
/*-------------------------------------------------------------------------*\
|
|
* Init module
|
|
\*-------------------------------------------------------------------------*/
|
|
int except_open(lua_State *L) {
|
|
luaL_setfuncs(L, func, 0);
|
|
return 0;
|
|
}
|