将shared_ptr传递给Lua

时间:2018-08-27 11:50:23

标签: c++ lua luajit

前段时间,我决定采用一种引用计数方法来管理资源的生存期,这是一个技巧。后来我面临一个新的挑战,就是要保持对象的生命周期,直到Lua在单独的线程中执行长时间运行的任务(这是通过turbo库实现的)。

我想到的想法是将shared_ptr的副本传递给Lua,并让垃圾回收器决定何时不再需要该资源。 但是问题是Lua两次清除数据,我不知道为什么。

在这里,我将描述我执行的一些步骤。首先,我要创建一个表,为其设置元表并将其设置为全局

static const struct luaL_reg FileTableDefinition[] = {
  { "__gc",     destroy },
  { NULL,       NULL } };

const char* id = "PointerManager";
lua_newtable( state );
luaL_newmetatable( state, id );
luaL_register( state, NULL, FileTableDefinition );
lua_pushliteral( state, "__index" );
lua_pushvalue( state, -2 );
lua_rawset( state, -3 );
lua_setglobal( state, id );

在第二步中,我正在用户数据中分配一个带对象的shared_ptr,并从步骤2中定义的表中设置析构函数。 File是一个导出到LUA的类,因此用户实际上可以执行对该类的调用。

void *userData = lua_newuserdata( state, sizeof( std::shared_ptr<File> ) );

// Handle allocation failure
if ( ! userData) { return; }

// Use the placement parameters of the new operator to allocate
// shared_ptr within the userdata provided by Lua. Copy our other
// shared_ptr into it, increasing the reference count
new(userData) std::shared_ptr<File>( aFile );

// Now just set the metatable on this new object
luaL_getmetatable( state, "PointerManager" );
lua_setmetatable( state, -2 );

对象的析构函数定义如下:

int destroy( lua_State* state )
{
    void* resourcePtr = luaL_checkudata( state, 1, "PointerManager" );  
    if( resourcePtr )
    {
        auto resource = static_cast<std::shared_ptr<File>*>( resourcePtr );
        resource->reset();
        return 1;
    }
    return 0;
}

我正在对该事件进行脚本调用(lua_pcall)。脚本定义如下:

local turbo = require("turbo")

function sleep(n)
  os.execute("sleep " .. tonumber(n))
end

function on_file_event ( file )
    print("on_file_event enter.")
    turbo.ioloop.instance():add_callback(function()
        local thread = turbo.thread.Thread(function(th)
            print("executing")
            sleep(1)
            print( file:filepath() )
            file = nil
            collectgarbage("collect")
            print("stopped executing")
            th:stop()
        end)

        turbo.ioloop.instance():close()
    end):start()
    print("on_file_event exit.")
end

我在类析构函数中放了一个cout,以某种方式调用了它两次。两次都调用了destroy函数(一次是在我调用垃圾收集器时,一次是在程序退出时),而无论我在进行一次底层对象的共享指针析构函数的重置都达到了两次。共享指针的内存和指针本身是相同的,线程ID也相同。我也做了Valgrind潜水,但没有发现腐败。

Calling destroy file 
        >> ~File object deleted (/opt/nids/artifacts/generic/32). Address: 0x1bec850

Calling destroy file 
        >> ~File object deleted (/opt/nids/artifacts/generic/32). Address: 0x1bec850

我做错了什么吗?

1 个答案:

答案 0 :(得分:0)

我找到了原因。它是turbo库内部的基础分支。当我在lua中创建新线程时-发生了C派生,因此地址相同,但实际上它们位于不同的内存空间中。后来释放了内存,出现了两套打印件。

我能够通过strace确认这一点。