C++ 如何访问 lua_pcall 调用的 C 函数的返回值

发布于08月10日

我有这个功能:

static int generic_pcall_wrapper(lua_State* L, int (*funct)(lua_State*)){
    int narg = lua_gettop(L);
    lua_pushcfunction(L, funct);
    lua_rotate(L, 1, 1);

    int res = lua_pcall(L, narg, LUA_MULTRET, 0);
    if(res != 0){
        //handle error
    }
    return ??; // I would like to return 1 as in sum()
}

static int sum(lua_State* L){
    int a = lua_tointeger(L, -1);
    int b = lua_tointeger(L, -2);    
    lua_pop(L, 1);   
    lua_pushnumber(L, a + b);
    return 1;
}

static int foo(lua_State* L){
    return generic_pcall_wrapper(L, sum);
}

我想让Generic_pcall_wrapper()返回sum()返回的值.

请注意，此函数应该接受许多不同的函数，并且它们如何处理返回值未指定，唯一可靠的值是每个函数返回的数字.

推荐答案

我不太明白的第一件事是，为什么求和函数需要lua_pop.这个对lua_op的调用非常不寻常，在the official examples中不存在.

static int sum(lua_State* L)
{
  int a = lua_tointeger(L, -1);
  int b = lua_tointeger(L, -2);    
  lua_pop(L, 1); /* unusual, probably not needed */
  lua_pushnumber(L, a + b);
  return 1;
}

关于原始问题，在回调sum中，返回值表示已将多少结果压入堆栈.该值由Lua运行时使用，Lua API的用户不可用.因此，如果API用户稍后需要该值，则需要使用Lua函数保存该值.

使用Lua API，一种自然的方法是将其推送到堆栈上.

static int sum(lua_State* L)
{
  int a = lua_tointeger(L, -1);
  int b = lua_tointeger(L, -2);
  lua_pushnumber(L, a + b); /* push the result of sum          */
  lua_pushinteger(L, 1);    /* sum return 1 value on the stack */
  return 2; /* return all the values (sum) and then #values */
}

然后:

泛型包装获取堆栈上的返回值的数量
包装器从堆栈中删除该特定值
返回给调用方的值的个数

static int generic_pcall_wrapper(lua_State* L, int (*funct)(lua_State*))
{
  int narg = lua_gettop(L);
  lua_pushcfunction(L, funct);

  //push argument again, but above the function
  int stack_size = lua_gettop(L);
  
  for (int i = 1; i < stack_size; i++)
    lua_pushvalue(L, i);
    
  int res = lua_pcall(L, narg, LUA_MULTRET, 0);

  /* Get the number of values which have been pushed */
  int res_count = lua_tointeger(L, -1);

  /* Remove the number of values */
  lua_pop(L);

  return res_count;
}