lua环境和模块

时间:2016-08-30 13:49:29

标签: lua sandbox

假设我有一个模块:

-- env.lua

local env = {}

function env.resolve(str)
  print("mod", _ENV)

  if _resolve_path ~= nil then
    return _resolve_path(str)
  else
    error("bad env")
  end
end

return env

以及使用它的一些代码:

-- sandbox demo
-- run as: lua env-test.lua

env = require('env')

function _resolve_path(path)
  return "/" .. path
end

print("before main()")
print("", _ENV)
print("", env.resolve("test"))

local sandbox
do
  local _ENV = {
    print = print,
    env = env,
    _resolve_path = function (path)
      return "/chroot/" .. path
    end
  }
  function sandbox()
    print("from sandbox()")
    print("", _ENV)
    print("", env.resolve("test"))
  end
end

sandbox()

print("after main()")
print("", _ENV)
print("", env.resolve("test"))

我想要实现的是来自sandbox()的env.resolve()将使用环境中的自定义_resolve_path函数。它看到环境不适用于从沙盒函数调用的代码。目标是根据调用它们的位置来预测某些模块的行为。例如。沙箱{1,2,3}()具有不同的本地_resolve_path()函数。

1 个答案:

答案 0 :(得分:2)

使用require加载模块时,它将绑定到全局环境。一旦在环境中创建了一个函数,它就会在整个生命周期中拥有该环境。

在Lua 5.2之前,您可以使用set/getfenv来改变环境,但环境现在是词汇。只能通过更改_ENV upvalue

来更改调试库,才能更改环境

那么,如何在不同的环境中运行相同的功能呢?您可以将环境作为参数传递:

function env.resolve(str, _ENV)
  print("mod", _ENV)
  if _resolve_path ~= nil then
    return _resolve_path(str)
  else
    error("bad env")
  end
end

然后在那里拨打resolve,如:

env.resolve('test', _ENV)

或者,如果您希望不必为每个resolve调用指定环境,则可以将resolve函数绑定到每个新环境:

-- env.lua
local print = print
local error = error
local env = {}

-- this is the actual resolve function that takes the environment as a parameter
local function resolve_env(str, _ENV)
  print("mod", _ENV)
  if _resolve_path ~= nil then
    return _resolve_path(str)
  else
    error("bad env")
  end
end

-- this is the module (ie. global) resolve
function env.resolve(str)
  return resolve_env(str, _ENV)
end

-- this function binds a resolve function to a sandbox environment
function env.bind(_ENV)
  _ENV.env = {
    resolve = function(str)
      return resolve_env(str, _ENV)
    end
  }
  return _ENV
end

return env

沙箱现在可以设置绑定解析:

-- sandbox.lua
env = require 'env'

function _resolve_path(path)
  return "/" .. path
end

print("before main()")
print("", _ENV)
print("", env.resolve("test"))

local sandbox; do
  local _ENV = env.bind{
    print = print,
    _resolve_path = function (path)
      return "/chroot/" .. path
    end
  }

  function sandbox()
    print("from sandbox()")
    print("", _ENV)
    print("", env.resolve("test"))
  end
end

sandbox()

print("after main()")
print("", _ENV)
print("", env.resolve("test"))

哪会产生结果:

$ lua sandbox.lua
before main()
        table: 00612f40
mod     table: 00612f40
        /test
from sandbox()
        table: 0061c7a8
mod     table: 0061c7a8
        /chroot/test
after main()
        table: 00612f40
mod     table: 00612f40
        /test