stjr/SRB2
0
0
Fork 0
mirror of https://git.do.srb2.org/STJr/SRB2.git synced 2024-11-15 01:01:33 +00:00
Code Issues Projects Releases Packages Wiki Activity

Better check for tag list userdata

Browse source
This commit is contained in:
James R 2020-12-14 08:08:01 -08:00
parent 00dff6d283
commit d0f3a6d737
1 changed files with 31 additions and 21 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

52
src/lua_taglib.c
View file

@ -156,13 +156,9 @@ static int lib_numTaggroupElements(lua_State *L)
return 1; return 1;
} }
static void push_taglist(lua_State *L, int idx) #ifdef MUTABLE_TAGS
{ static int meta_ref[2];
lua_getmetatable(L, idx); #endif
lua_pushliteral(L, "taglist");
lua_rawget(L, -2);
lua_remove(L, -2);
}
static int has_valid_field(lua_State *L) static int has_valid_field(lua_State *L)
{ {
@ -191,10 +187,19 @@ static taglist_t * valid_taglist(lua_State *L, int idx, boolean getting)
static taglist_t * check_taglist(lua_State *L, int idx) static taglist_t * check_taglist(lua_State *L, int idx)
{ {
luaL_checktype(L, idx, LUA_TUSERDATA); if (lua_isuserdata(L, idx) && lua_getmetatable(L, idx))
push_taglist(L, idx); {
luaL_argcheck(L, lua_toboolean(L, -1), idx, "must be a tag list"); lua_getref(L, meta_ref[0]);
return valid_taglist(L, idx, false); lua_getref(L, meta_ref[1]);
if (lua_rawequal(L, -3, -2) || lua_rawequal(L, -3, -1))
{
lua_pop(L, 3);
return valid_taglist(L, idx, false);
}
}
return luaL_argerror(L, idx, "must be a tag list"), NULL;
} }
static int taglist_get(lua_State *L) static int taglist_get(lua_State *L)
@ -223,7 +228,7 @@ static int taglist_get(lua_State *L)
} }
else else
{ {
push_taglist(L, 1); lua_getmetatable(L, 1);
lua_replace(L, 1); lua_replace(L, 1);
lua_rawget(L, 1); lua_rawget(L, 1);
return 1; return 1;
@ -395,12 +400,14 @@ static void open_taglist(lua_State *L)
#define new_literal(L, s) \ #define new_literal(L, s) \
(lua_pushliteral(L, s), luaL_ref(L, -2)) (lua_pushliteral(L, s), luaL_ref(L, -2))
static void set_taglist_metatable(lua_State *L, const char *meta) #ifdef MUTABLE_TAGS
static int
#else
static void
#endif
set_taglist_metatable(lua_State *L, const char *meta)
{ {
lua_createtable(L, 0, 4); luaL_newmetatable(L, meta);
lua_getglobal(L, "taglist");
lua_setfield(L, -2, "taglist");
lua_pushcfunction(L, taglist_get); lua_pushcfunction(L, taglist_get);
lua_createtable(L, 0, 1); lua_createtable(L, 0, 1);
new_literal(L, "valid"); new_literal(L, "valid");
@ -412,7 +419,9 @@ static void set_taglist_metatable(lua_State *L, const char *meta)
lua_pushcfunction(L, taglist_equal); lua_pushcfunction(L, taglist_equal);
lua_setfield(L, -2, "__eq"); lua_setfield(L, -2, "__eq");
lua_setfield(L, LUA_REGISTRYINDEX, meta); #ifdef MUTABLE_TAGS
return luaL_ref(L, LUA_REGISTRYINDEX);
#endif
} }
int LUA_TagLib(lua_State *L) int LUA_TagLib(lua_State *L)
@ -431,10 +440,11 @@ int LUA_TagLib(lua_State *L)
open_taglist(L); open_taglist(L);
set_taglist_metatable(L, META_TAGLIST);
#ifdef MUTABLE_TAGS #ifdef MUTABLE_TAGS
set_taglist_metatable(L, META_SECTORTAGLIST); meta_ref[0] = set_taglist_metatable(L, META_TAGLIST);
meta_ref[1] = set_taglist_metatable(L, META_SECTORTAGLIST);
#else
set_taglist_metatable(L, META_TAGLIST);
#endif #endif
return 0; return 0;