Disable pointer auto-vivification unless new is specified.

Since it is essentially allocating non-gc managed objects,
it can lead to memory leaks and shouldn't happen invisibly.
Also support using the 'assign' key to request assign()
from another object before processing the current map.
develop
Alexander Gavrilov 2012-03-28 15:28:42 +08:00 committed by Petr Mrázek
parent fe091de0b2
commit 929657bed4
2 changed files with 58 additions and 13 deletions

@ -168,10 +168,16 @@ static void autovivify_ptr(lua_State *state, int fname_idx, void **pptr,
{ {
lua_getfield(state, val_index, "new"); lua_getfield(state, val_index, "new");
if (!lua_isnil(state, -1)) // false or nil => bail out
if (!lua_toboolean(state, -1))
field_error(state, fname_idx, "null and autovivify not requested", "write");
// not 'true' => call df.new()
if (!lua_isboolean(state, -1))
{ {
int top = lua_gettop(state); int top = lua_gettop(state);
// Verify new points to a reasonable type of object
type_identity *suggested = get_object_identity(state, top, "autovivify", true, true); type_identity *suggested = get_object_identity(state, top, "autovivify", true, true);
if (!is_type_compatible(state, target, 0, suggested, top+1, false)) if (!is_type_compatible(state, target, 0, suggested, top+1, false))
@ -179,17 +185,22 @@ static void autovivify_ptr(lua_State *state, int fname_idx, void **pptr,
lua_pop(state, 1); lua_pop(state, 1);
// Invoke df.new()
lua_getfield(state, LUA_REGISTRYINDEX, DFHACK_NEW_NAME); lua_getfield(state, LUA_REGISTRYINDEX, DFHACK_NEW_NAME);
lua_swap(state); lua_swap(state);
lua_call(state, 1, 1); lua_call(state, 1, 1);
// Retrieve the pointer
void *nval = get_object_internal(state, target, top, false); void *nval = get_object_internal(state, target, top, false);
// shouldn't happen: this means suggested type is compatible,
// but its new() result isn't for some reason.
if (!nval) if (!nval)
field_error(state, fname_idx, "inconsistent autovivify type", "write"); field_error(state, fname_idx, "inconsistent autovivify type", "write");
*pptr = nval; *pptr = nval;
} }
// otherwise use the target type
else else
{ {
if (!target) if (!target)

@ -616,20 +616,32 @@ static void invoke_resize(lua_State *state, int table, lua_Integer size)
lua_call(state, 2, 0); lua_call(state, 2, 0);
} }
static void copy_table(lua_State *state, int dest, int src, int skipkey) static void copy_table(lua_State *state, int dest, int src, int skipbase)
{ {
// stack: (skipbase) skipkey skipkey |
int top = lua_gettop(state);
lua_pushnil(state); lua_pushnil(state);
while (lua_next(state, src)) while (lua_next(state, src))
{ {
if (lua_equal(state, -2, skipkey)) for (int i = skipbase+1; i <= top; i++)
{
if (lua_rawequal(state, -2, i))
{
lua_pop(state, 1); lua_pop(state, 1);
else goto next_outer;
}
}
{ {
lua_pushvalue(state, -2); lua_pushvalue(state, -2);
lua_swap(state); lua_swap(state);
lua_settable(state, dest); lua_settable(state, dest);
} }
next_outer:;
} }
} }
@ -655,18 +667,40 @@ static int meta_assign(lua_State *state)
{ {
type_identity *id = get_object_identity(state, 1, "df.assign()", false); type_identity *id = get_object_identity(state, 1, "df.assign()", false);
int base = lua_gettop(state);
// x:assign{ assign = foo } => x:assign(foo)
bool has_assign = false;
lua_pushstring(state, "assign");
lua_dup(state);
lua_rawget(state, 2);
if (!lua_isnil(state,-1))
{
has_assign = true;
lua_getfield(state, LUA_REGISTRYINDEX, DFHACK_ASSIGN_NAME);
lua_pushvalue(state, 1);
lua_pushvalue(state, base+2);
lua_call(state, 2, 0);
}
lua_pop(state, 1);
// new is used by autovivification and should be skipped
lua_pushstring(state, "new");
if (id->isContainer()) if (id->isContainer())
{ {
// check resize field
lua_pushstring(state, "resize"); lua_pushstring(state, "resize");
int resize_str = lua_gettop(state);
lua_dup(state); lua_dup(state);
lua_rawget(state, 2); lua_rawget(state, 2);
if (lua_isnil(state,-1)) if (lua_isnil(state,-1) && !has_assign)
{ {
/* /*
* nil or missing resize field => 1-based lua array * no assign && nil or missing resize field => 1-based lua array
*/ */
int size = lua_objlen(state, 2); int size = lua_objlen(state, 2);
@ -682,7 +716,7 @@ static int meta_assign(lua_State *state)
} }
else else
{ {
if (lua_isboolean(state, -1)) if (lua_isboolean(state, -1) || lua_isnil(state, -1))
{ {
// resize=false => just assign // resize=false => just assign
// resize=true => find the largest index // resize=true => find the largest index
@ -711,13 +745,13 @@ static int meta_assign(lua_State *state)
} }
lua_pop(state, 1); lua_pop(state, 1);
copy_table(state, 1, 2, resize_str); copy_table(state, 1, 2, base);
} }
} }
else else
{ {
lua_pushstring(state, "new");
copy_table(state, 1, 2, lua_gettop(state)); copy_table(state, 1, 2, base);
} }
} }