c - Lua 将句柄传递给使用 newlib 创建的函数

Question

我正在使用 luaL_newlib(luaState, afbFunction)，其中 afbFunction 是静态 luaL_Reg 数组。不幸的是，luaL_Reg 只支持两个字段:name 和 func。因此，无法将句柄传递给函数并实现如下内容:

static const luaL_Reg afbFunction[] = {
    {"notice" , LuaPrintMsg, MSG_NOTICE},
    {"info"   , LuaPrintMsg, MSG_INFO},
    {"warning", LuaPrintWar, MSG_WARN},
    {NULL, NULL}  /* sentinel */
};

我有两个动机，没有像 Lua 函数那样多的 C 入口点:

1. 我有一个通用的包装器，用于处理输入/输出的所有 C 函数从 Lua(表)到 C(jsonc)的参数值。
2. 要连接的 Lua 函数的数量来自配置文件，无需重新构建代码即可更改。长话短说，配置文件有一个 sharelib 路径，后跟要公开的函数列表。

有人有办法为多个 Lua 函数设置唯一的 C 入口点吗？当使用 luaL_newlib() 时，C 函数接收 lib 表作为第二个参数，但它似乎接收到完整的表而不是相应的条目，甚至测试标签也是不可能的。

Answer 1

使用上值!

如果你不使用 luaL_newlib (这是一个定义为......的宏

#define luaL_newlib(L,l)  \
  (luaL_checkversion(L), luaL_newlibtable(L,l), luaL_setfuncs(L,l,0))

…) 但显式调用 luaL_setfuncs ，你可以让每个函数都有一些上值。可以用 nil 初始化它们，稍后你可以修补它们。虽然您不能使用额外的字段扩展 luaL_Reg，但您可以定义一个额外的 struct 并为其他参数设置一个单独的数组。

最小示例实现(可能需要更多错误检查):

(应该可以原样复制和粘贴，文本 block 应该变成注释)

---

#include <lua.h>
#include <lauxlib.h>
/*

首先，定义一个struct，它首先是函数名，然后是您需要的任何内容。此示例仅使用一个字符串，但您可以使用 int、void* ......甚至有多个值。

*/
typedef struct xlua_uvinfo {
    const char *name;  /* this is how we identify the functions */
    const char *uv;  /* this can change arbitrarily */
} xlua_uvinfo;
/*

然后我们将有一个函数来修补堆栈上的 libtable 以放入真正的上值数据。您需要调整您推送的字段的类型 (lua_pushX)、顺序和/或名称。

请注意，由于我们通过名称识别函数，因此并非所有函数都需要获取上值。 (因此您可以在 luaL_Reg 中包含普通函数，这将简单地以不会在此处填充的虚拟上值槽结束。)

*/
static void setupvalues( lua_State *L, const xlua_uvinfo *l ) {
    for (; l->name != NULL; l++) {
        /* get the function by name */
        if (lua_getfield( L, -1, l->name ) == LUA_TFUNCTION) {
            /* this needs to change according to what you have in the struct */
            lua_pushstring( L, l->uv );
            lua_setupvalue( L, -2, 1 );
        }
        /* pop the function */
        lua_pop( L, 1 );
    }
}
/*

这是您的函数的虚拟实现。我使用字符串前缀——这很容易调试/检查。

*/
#define MSG_NOTICE "NOTE"
#define MSG_INFO "INFO"
#define MSG_WARN "WARN"

static int LuaPrintMsg( lua_State *L ) {
   const char *prefix = lua_tostring( L, lua_upvalueindex(1) );
   printf( "%s: %s\n", prefix, lua_tostring( L, 1 ) );
   return 0;
}

/* plain luaL_Reg array */
static const luaL_Reg afbFunction[] = {
    {"notice" , LuaPrintMsg},
    {"info"   , LuaPrintMsg},
    {"warning", LuaPrintMsg},
    {NULL, NULL}
};
/*

现在是变化的部分:为上值添加一个额外的数组，然后调整 luaopen_X 函数。

*/
/* extra upvalue array (must also be terminated by { NULL, NULL } !) */
static const xlua_uvinfo afbUpvalues[] = {
    {"notice" , MSG_NOTICE},
    {"info"   , MSG_INFO},
    {"warning", MSG_WARN},
    {NULL, NULL}
};

int luaopen_foo( lua_State *L ) {
    /* as mentioned above, we can't use `luaL_newlib`, expand manually */
    luaL_checkversion( L );
    /* create the empty table, pre-allocated to the right size */
    luaL_newlibtable( L, afbFunction );
    /* now push a dummy upvalue (more if you add extra upvalues!) */
    lua_pushnil( L );
    /* register the functions, giving 1 upvalue to each */
    luaL_setfuncs( L, afbFunction, 1 );
    /* now set the actual upvalues */
    setupvalues( L, afbUpvalues );
    return 1;
}

---

如果将其保存为 foo.c，编译为(在 Linux 上)gcc -shared -fPIC -o foo.so foo.c 然后运行 ​​lua -l foo，您将获得所需的行为:

foo.info "foo"
--> INFO: foo
foo.notice "bar"
--> NOTE: bar
foo.warning "baz"
--> WARN: baz