Ограничение времени выполнения для скрипта Lua, вызываемого из C API - PullRequest
Новая
       48

Ограничение времени выполнения для скрипта Lua, вызываемого из C API

9 голосов
/ 04 августа 2010 
luaL_loadfile(mState, path.c_str());
lua_pcall(mState, 0, 0, 0);

Есть ли способ установить ограничение времени выполнения (скажем, 10-20 секунд) для этих двух операторов C ++, которые загружают, а затем исполняют файл lua?

Поскольку файл Lua ненадежен, я не хочу, чтобы злонамеренный пользователь бесконечно зависал в программе с бесконечным циклом в коде Lua.

Пометка C, потому что Lua API это C, тэг C ++, потому что я использую C ++

1 Ответ

21 голосов
/ 04 августа 2010

Существует lua_sethook, который может использоваться для указания интерпретатору вызывать ловушку после выполнения каждой команды count.Таким образом, вы можете отслеживать пользовательский скрипт и завершать его, если он съел свою квоту: 

 int lua_sethook (lua_State *L, lua_Hook f, int mask, int count);

Это также можно использовать из Lua: 

debug.sethook(function() print("That's enough for today"); os.exit(0); end, "", 10000)
for i=1,10000 do end

Если вы используете методы изhttp://lua -users.org / wiki / SandBoxes затем вы можете настроить безопасную среду исполнения с sethook() и друзьями полностью из Lua, а затем переключиться в режим песочницы при выполнении сценария пользователя.Я попробовал это здесь, просто для начала: 

-- set an execution quota 
local function set_quota(secs)
 local st=os.clock()
 function check() 
  if os.clock()-st > secs then 
    debug.sethook() -- disable hooks
    error("quota exceeded")
  end
 end
 debug.sethook(check,"",100000);
end

-- these are the global objects, the user can use:
local env = {print=print}

-- The user code is allowed to run for 5 seconds.
set_quota(5)

-- run code under environment:
local function run(untrusted_code)
  local untrusted_function, message = loadstring(untrusted_code)
  if not untrusted_function then return nil, message end
  setfenv(untrusted_function, env)
  return pcall(untrusted_function)
end

-- here is the user code:
local userscript=[[
function fib(n) 
 if n<2 then return n
 else return fib(n-2)+fib(n-1)
 end
end
for n=1,42 do print(n,fib(n)) end
]]
-- call it:
local r,m=run(userscript)
print(r,m)

Это должно вывести значения fib () в течение 5 секунд, а затем показать ошибку.

...