如何访问 lua_pcall 调用的 C function 的返回值？

Question

我有这个 function：

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() 返回的值。

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

Answer 1

我不太明白的第一件事是总和function中需要lua_pop 。 这个对 lua_pop 的调用非常不寻常，并且在官方示例中没有出现。

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 */
}

然后：

1. 通用包装器获取堆栈上返回值的数量
2. 包装器从堆栈中移除这个特殊值
3. 返回给调用者的值的数量

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;
}