Code/Engine/Core/Scripting/LuaWrapper/Initialize.cpp - dev - redgpu/ezEngine

Code/Engine/Core/Scripting/LuaWrapper/Initialize.cpp
100 строк2 KB
C-Core
Re-enabled security relevant warnings (#1339)
25 июн 2024, 12:03
25 июн 2024, 12:037083663
#include <Core/CorePCH.h>

#include <Core/Scripting/LuaWrapper.h>

#ifdef BUILDSYSTEM_ENABLE_LUA_SUPPORT

ezLuaWrapper::ezLuaWrapper()
{
  m_bReleaseOnExit = true;
  m_pState = nullptr;

  Clear();
}

ezLuaWrapper::ezLuaWrapper(lua_State* s)
{
  m_pState = s;
  m_bReleaseOnExit = false;
}

ezLuaWrapper::~ezLuaWrapper()
{
  if (m_bReleaseOnExit)
    lua_close(m_pState);
}

void ezLuaWrapper::Clear()
{
  EZ_ASSERT_DEV(m_bReleaseOnExit, "Cannot clear a script that did not create the Lua state itself.");

  if (m_pState)
    lua_close(m_pState);

  m_pState = lua_newstate(lua_allocator, nullptr);

  luaL_openlibs(m_pState);
}

ezResult ezLuaWrapper::ExecuteString(const char* szString, const char* szDebugChunkName, ezLogInterface* pLogInterface) const
{
  EZ_ASSERT_DEV(m_States.m_iLuaReturnValues == 0,
    "ezLuaWrapper::ExecuteString: You didn't discard the return-values of the previous script call. {0} Return-values were expected.",
    m_States.m_iLuaReturnValues);

  if (!pLogInterface)
    pLogInterface = ezLog::GetThreadLocalLogSystem();

  int error = luaL_loadbuffer(m_pState, szString, ezStringUtils::GetStringElementCount(szString), szDebugChunkName);

  if (error != LUA_OK)
  {
    EZ_LOG_BLOCK("ezLuaWrapper::ExecuteString");

    ezLog::Error(pLogInterface, "[lua]Lua compile error: {0}", lua_tostring(m_pState, -1));
    ezLog::Info(pLogInterface, "[luascript]Script: {0}", szString);

    return EZ_FAILURE;
  }

  error = lua_pcall(m_pState, 0, 0, 0);

  if (error != LUA_OK)
  {
    EZ_LOG_BLOCK("ezLuaWrapper::ExecuteString");

    ezLog::Error(pLogInterface, "[lua]Lua error: {0}", lua_tostring(m_pState, -1));
    ezLog::Info(pLogInterface, "[luascript]Script: {0}", szString);

    return EZ_FAILURE;
  }

  return EZ_SUCCESS;
}

void* ezLuaWrapper::lua_allocator(void* ud, void* ptr, size_t osize, size_t nsize)
{
  EZ_IGNORE_UNUSED(ud);

  /// \todo Create optimized allocator.

  if (nsize == 0)
  {
    delete[] (ezUInt8*)ptr;
    return (nullptr);
  }

  ezUInt8* ucPtr = new ezUInt8[nsize];

  if (ptr != nullptr)
  {
    ezMemoryUtils::Copy(ucPtr, (ezUInt8*)ptr, ezUInt32(osize < nsize ? osize : nsize));

    delete[] (ezUInt8*)ptr;
  }

  return ((void*)ucPtr);
}


#endif // BUILDSYSTEM_ENABLE_LUA_SUPPORT