检查 Lua C 中的保护模式

Question

在 Lua C 处理程序中，我在抛出错误之前在注册表中设置了一个标志。如果使用 pcall 从 Lua 调用处理程序，我想确保设置（或清除）此标志。以下是更多详细信息：

我有一个基本的 Lua 函数 handle 使用 lua_pcall 从 C 调用：

function handle()
  c_handle()
end

C 中的句柄具有下面的基本结构，我们希望在抛出错误之前设置一个标志以将错误与其他错误区分开来。

int c_handle(lua_State *L)
{
  if (condition) {
    set a flag in registry
    throw error with lua_error
}

如果 Lua 使用 pcall 调用处理程序，我想要么不设置注册表标志，要么清除它：

function handle()
  pcall(c_handle)
end

Answer 1

您可以对脚本进行沙箱处理并将pcall 替换为您自己的版本，该版本会回调c_handle()，如下所示：

local newenv = {
    pcall = function(f, arg1, arg2, ...)
        if pcall(f, arg1, arg2, ...) then
            c_handle(true) -- no error, pass true to c_handle
        else
            c_handle(false) -- error, pass false to c_handle
        end if
    end
}
setmetatable(newenv, {__index = _G})
setfenv(1, newenv)

所以现在当您进行受保护的呼叫时，它将通过c_handle。

编辑

不太确定将参数传递给 C 函数的语法，但我想这将是它的基本要点。