从C++中访问函数环境中的Lua变量

2012-11-8 0:8:42
收藏：0
阅读：158
评论：2

这可能是一个简单的问题，但是我却遇到了麻烦。这是针对 Lua 5.1 的。

我有一个在其自身环境下运行的脚本。在该环境中，我有一个名为“plugin”的变量，我从 C++ 中这样设置：

    lua_getfield(L, LUA_REGISTRYINDEX, getScriptId());  // 将脚本环境表放入栈中 -- env_table

    lua_pushstring(L, "plugin");  //                       -- env_table, "plugin"
    luaW_push(L, this);   //                                -- env_table, "plugin", *this
    lua_rawset(L, -3);    // env_table["plugin"] = *this    -- env_table

    lua_pop(L, -1);       // 清理                         -- <<empty stack>>

在运行 Lua 脚本之前，我将函数环境设置为：

 lua_getfield(L, LUA_REGISTRYINDEX, getScriptId());    // 将 REGISTRY[scriptId] 推入栈顶           -- function, table
 lua_setfenv(L, -2);                                   // 将该表设置为函数的 env                -- function

当我的脚本运行时，它可以看到并与插件变量进行交互，就像预期的那样。到目前为止，一切顺利。

在某一时刻，Lua 脚本调用了一个 C++ 函数，在该函数中，我想查看插件变量是否已设置。

我尝试了许多许多事情，但似乎看不到插件变量。以下是我尝试的4个事项：

lua_getfield(L, LUA_ENVIRONINDEX, "plugin");
bool isPlugin = !lua_isnil(L, -1);
lua_pop(L, 1);    // 从堆栈中删除我们刚添加的值

lua_getfield(L, LUA_GLOBALSINDEX, "plugin");
bool isPlugin2 = !lua_isnil(L, -1);
lua_pop(L, 1);    // 从堆栈中删除我们刚添加的值

lua_getglobal(L, "plugin");
bool isPlugin3 = !lua_isnil(L, -1);
lua_pop(L, 1);    // 从堆栈中删除我们刚添加的值

lua_pushstring(L, "plugin");
bool isPlugin4 = lua_isuserdata(L, -1);
lua_pop(L, 1);

不幸的是，所有的 isPlugin 变量都返回 false。就好像从 Lua 调用的 C++ 函数无法看到在 Lua 环境中设置的变量。

有什么办法可以从 C++ 中看到插件变量吗？

谢谢！

点赞
用户734069
用户734069

每个 Lua 函数都有它自己的环境。它们不会继承调用它们的环境。因此，如果你的 C++ 函数没有使用具有 plugin 变量的环境，那么它就看不到它。

2012-11-08 03:58:14
用户1801919
用户1801919

你可以将环境作为闭包的一部分传递给 C 函数（参见lua_pushcclosure）。我不知道你的设置是什么，但是我可以看到有三种方式可以解决：

1）你的 C 函数在与函数相同的环境中注册——很好，会起作用。

2）你的 C 函数在全局环境中注册，但调用它的 Lua 函数都驻留在一个特定的环境中——如果在函数注册时该环境存在（因此可以添加到闭包中），仍将起作用。

3）你的 C 函数在全局环境中注册，可以被在不同环境下工作的不同 Lua 函数调用——将不再起作用。

如果是情况 2 或 3，则如果你更改实现以使用变体 1，则可能没有缺点。

编辑：好的，那不起作用。如果你愿意稍微远离 Lua API，有一种获取底层信息的方法。免责声明：我正在使用 5.2，所以我正在尝试为 5.1 调整我的方法。我无法测试此方法，也可能无法奏效。

首先，你需要 #include "lstate.h"。

这是 5.1 中的 lua_State 结构：

struct lua_State {
  CommonHeader;
  lu_byte status;
  StkId top;  /* first free slot in the stack */
  StkId base;  /* base of current function */
  global_State *l_G;
  CallInfo *ci;  /* call info for current function */
  const Instruction *savedpc;  /* `savedpc' of current function */
  StkId stack_last;  /* last free slot in the stack */
  StkId stack;  /* stack base */
  CallInfo *end_ci;  /* points after end of ci array*/
  CallInfo *base_ci;  /* array of CallInfo's */
  int stacksize;
  int size_ci;  /* size of array `base_ci' */
  unsigned short nCcalls;  /* number of nested C calls */
  lu_byte hookmask;
  lu_byte allowhook;
  int basehookcount;
  int hookcount;
  lua_Hook hook;
  TValue l_gt;  /* table of globals */
  TValue env;  /* temporary place for environments */
  GCObject *openupval;  /* list of open upvalues in this stack */
  GCObject *gclist;
  struct lua_longjmp *errorJmp;  /* current error recover point */
  ptrdiff_t errfunc;  /* current error handling function (stack index) */
};

假设 L 是你的 lua_State*。 正如你所看到的，L->ci 保存当前的调用信息，并且调用信息数组包含在 L->base_ci 和 L->end_ci 之间。

因此，调用你的 C 函数的 Lua 函数位于 (L->end_ci-2)（应该与 (L->ci-1) 相同），其堆栈 id（StkId）为 (L->end_ci-2)->func。 我们可以通过这样的方式欺骗 Lua API，使你能够处理低于当前调用函数的堆栈 id：

StkId saved = L->base;
L->base = L->base_ci->base;
int idx = (L->end_ci-2)->func - L->base+1;
lua_getfenv(L, idx);
L->base = saved;

现在环境表应该位于堆栈顶部了。

编辑：Lua API 对有效索引的检查有点棘手。这应该能骗过它们。

2012-11-09 04:05:21

作者:

用户103252

用户103252
推荐文章