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

时间:2012-11-08 00:08:42

标签: c++ lua

这可能是一个简单的问题,但我很难过。这是Lua 5.1。

我有一个在自己的环境中运行的脚本。在那个环境中,我有一个名为“plugin”的变量,我是用C ++设置的,如下所示:

    lua_getfield(L, LUA_REGISTRYINDEX, getScriptId());  // Put script's env table onto the stack  -- 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);           // Cleanup                       -- <<empty stack>>

在运行我的Lua脚本之前,我设置了这样的函数环境:

 lua_getfield(L, LUA_REGISTRYINDEX, getScriptId());    // Push REGISTRY[scriptId] onto stack           -- function, table
 lua_setfenv(L, -2);                                   // Set that table to be the env for function    -- function

当我的脚本运行时,它可以按预期看到插件变量并与之交互。到现在为止还挺好。

有一次,Lua脚本调用C ++函数,在该函数中,我想查看是否设置了插件变量。

我尝试了很多东西,而且我似乎无法看到插件变量。以下是我尝试过的4件事:

lua_getfield(L, LUA_ENVIRONINDEX, "plugin");
bool isPlugin = !lua_isnil(L, -1);
lua_pop(L, 1);    // Remove the value we just added from the stack

lua_getfield(L, LUA_GLOBALSINDEX, "plugin");
bool isPlugin2 = !lua_isnil(L, -1);
lua_pop(L, 1);    // Remove the value we just added from the stack

lua_getglobal(L, "plugin");
bool isPlugin3 = !lua_isnil(L, -1);
lua_pop(L, 1);    // Remove the value we just added from the stack

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

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

知道如何从C ++中看到插件变量吗?

谢谢!

2 个答案:

答案 0 :(得分:2)

Lua中的每个功能都有自己的环境。它们不会继承任何人调用它们的环境。因此,如果您的C ++函数不使用具有此plugin变量的环境,那么它将无法看到它。

答案 1 :(得分:0)

您可以将环境作为闭包的一部分传递给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保存当前的callinfo,callinfo数组包含在L->base_ciL->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检查有效索引有点棘手。这应该欺骗他们。