c - 如何访问 lua_pcall 调用的 C 函数的返回值？

Question

我有这个功能:

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

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

Answer 1

我不太明白的第一件事是需要 lua_pop在求和函数中。对 lua_pop 的调用非常不寻常，并且不存在于 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 */
}

然后:

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