bbac0838dd
* IWYU Core * IWYU Noise * IWYU Utility * IWYU Audio * IWYU Platform * IWYU Lua * IWYU Network * IWYU Physics2D * IWYU Physics3D * IWYU Renderer * IWYU Graphics * IWYU NDKServer * IWYU Fix * Try to fix compilation * Other fixes
96 lines
2.2 KiB
C++
96 lines
2.2 KiB
C++
// Copyright (C) 2017 Jérôme Leclercq
|
|
// This file is part of the "Nazara Engine - Lua scripting module"
|
|
// For conditions of distribution and use, see copyright notice in Config.hpp
|
|
|
|
#include <Nazara/Lua/LuaInstance.hpp>
|
|
#include <Lua/lauxlib.h>
|
|
#include <Lua/lua.h>
|
|
#include <Lua/lualib.h>
|
|
#include <Nazara/Core/Clock.hpp>
|
|
#include <Nazara/Core/Error.hpp>
|
|
#include <cassert>
|
|
#include <cstdlib>
|
|
#include <stdexcept>
|
|
#include <Nazara/Lua/Debug.hpp>
|
|
|
|
namespace Nz
|
|
{
|
|
namespace
|
|
{
|
|
int AtPanic(lua_State* internalState)
|
|
{
|
|
String lastError(lua_tostring(internalState, -1));
|
|
|
|
throw std::runtime_error("Lua panic: " + lastError);
|
|
}
|
|
}
|
|
|
|
LuaInstance::LuaInstance() :
|
|
LuaState(nullptr),
|
|
m_memoryLimit(0),
|
|
m_memoryUsage(0),
|
|
m_timeLimit(1000),
|
|
m_level(0)
|
|
{
|
|
m_state = lua_newstate(MemoryAllocator, this);
|
|
lua_atpanic(m_state, AtPanic);
|
|
lua_sethook(m_state, TimeLimiter, LUA_MASKCOUNT, 1000);
|
|
luaL_openlibs(m_state);
|
|
}
|
|
|
|
LuaInstance::~LuaInstance()
|
|
{
|
|
if (m_state)
|
|
lua_close(m_state);
|
|
}
|
|
|
|
inline void LuaInstance::SetMemoryUsage(std::size_t memoryUsage)
|
|
{
|
|
m_memoryUsage = memoryUsage;
|
|
}
|
|
|
|
void* LuaInstance::MemoryAllocator(void* ud, void* ptr, std::size_t osize, std::size_t nsize)
|
|
{
|
|
LuaInstance* instance = static_cast<LuaInstance*>(ud);
|
|
std::size_t memoryLimit = instance->GetMemoryLimit();
|
|
std::size_t memoryUsage = instance->GetMemoryUsage();
|
|
|
|
if (nsize == 0)
|
|
{
|
|
assert(memoryUsage >= osize);
|
|
|
|
instance->SetMemoryUsage(memoryUsage - osize);
|
|
std::free(ptr);
|
|
|
|
return nullptr;
|
|
}
|
|
else
|
|
{
|
|
std::size_t usage = memoryUsage + nsize;
|
|
if (ptr)
|
|
usage -= osize;
|
|
|
|
if (memoryLimit != 0 && usage > memoryLimit)
|
|
{
|
|
NazaraError("Lua memory usage is over memory limit (" + String::Number(usage) + " > " + String::Number(memoryLimit) + ')');
|
|
return nullptr;
|
|
}
|
|
|
|
instance->SetMemoryUsage(usage);
|
|
|
|
return std::realloc(ptr, nsize);
|
|
}
|
|
}
|
|
|
|
void LuaInstance::TimeLimiter(lua_State* internalState, lua_Debug* debug)
|
|
{
|
|
NazaraUnused(debug);
|
|
|
|
LuaInstance* instance;
|
|
lua_getallocf(internalState, reinterpret_cast<void**>(&instance));
|
|
|
|
if (instance->m_clock.GetMilliseconds() > instance->GetTimeLimit())
|
|
luaL_error(internalState, "maximum execution time exceeded");
|
|
}
|
|
}
|