如何在 Lua C API 中使用 `debug.getinfo(f).nparams`？

Lua手册介绍如何获取有关函数的信息：

要获取有关函数的信息，您需要将其推入堆栈并从“>”字符开始使用what字符串。(在这种情况下，lua_getinfo会从堆栈的顶部弹出函数。)例如，要知道函数f在哪一行定义的，可以编写以下代码：

lua_Debug ar;
lua_getglobal(L, "f");  /* get global 'f' */
lua_getinfo(L, ">S", &ar);
printf("%d\n", ar.linedefined);

因此，您唯一需要的就是将要检查的函数值推到 Lua 堆栈的顶部。

这里有实现一个 C 函数的示例，该函数调用作为第一个参数传递的函数以其余参数为参数--作为数组或单独参数，具体取决于期望参数的数量：

/* 在 Ubuntu Linux 上构建 `args.c` 的指令：
 * `gcc -Wall -I/usr/include/lua5.2 -fpic -shared -o args.so args.c`
 */
#include <lua.h>
#include <lauxlib.h>

/* 将 Lua 堆栈区间复制到数组 */
static void push_pack(lua_State* L, int first, int last) {
  int i;
  first = lua_absindex(L, first);
  last = lua_absindex(L, last);
  lua_createtable(L, last > first ? last - first : 0, 0);
  for (i = first; i <= last; ++i) {
    lua_pushvalue(L, i);
    lua_rawseti(L, -2, i-first+1);
  }
}

static int args(lua_State* L) {
  lua_Debug ar;
  /* 第一个参数必须是要调用的函数，剩余的是该函数的附加参数 */
  luaL_checktype(L, 1, LUA_TFUNCTION);
  /* 查询期望参数的数量 */
  lua_pushvalue(L, 1);
  lua_getinfo(L, ">u", &ar);
  if (ar.nparams > 1 || ar.isvararg) { /* 分别作为多个参数传递 */
    lua_call(L, lua_gettop(L)-1, LUA_MULTRET);
    return lua_gettop(L);
  } else { /* 创建数组并作为单个参数传递 */
    int top = lua_gettop(L);
    lua_pushvalue(L, 1); /* 首先推入函数 */
    push_pack(L, 2, top); /* 然后推入参数数组 */
    lua_call(L, 1, LUA_MULTRET);
    return lua_gettop(L) - top;
  }
}

int luaopen_args(lua_State* L) {
  lua_pushcfunction(L, args);
  return 1;
}

在 Lua 5.2 上测试：

#!/usr/bin/lua5.2

local args = require("args")

local function f(x, ...)
  print(">", x, ...)
  return 5, 6
end

local function g(x)
  print(">", x)
  for i,v in ipairs(x) do
    print("", i, v)
  end
  return 7, 8
end

local function h(x, y)
  print(">", x, y)
  return 9, 10
end

print(args(f, 1, 2, 3))
print(args(g, 1, 2, 3))
print(args(h, 1, 2, 3))
print(args(print, 1, 2, 3))