khanat-code-old/code/ryzom/tools/leveldesign/variable_parser/lua_helper.cpp

563 lines
15 KiB
C++

// Ryzom - MMORPG Framework <http://dev.ryzom.com/projects/ryzom/>
// Copyright (C) 2010 Winch Gate Property Limited
//
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU Affero General Public License as
// published by the Free Software Foundation, either version 3 of the
// License, or (at your option) any later version.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU Affero General Public License for more details.
//
// You should have received a copy of the GNU Affero General Public License
// along with this program. If not, see <http://www.gnu.org/licenses/>.
#include "stdafx.h"
#include "lua_helper.h"
#include "nel/misc/file.h"
#include "nel/misc/path.h"
//#include "interface_manager.h"
extern "C"
{
#ifdef max
# undef max
#endif
#include "lualib.h"
}
#ifdef NL_OS_WINDOWS
# ifndef NL_EXTENDED_FOR_SCOPE
# undef for
# endif
#endif
#undef new
#include <luabind/luabind.hpp>
#define new NL_NEW
using namespace std;
using namespace NLMISC;
// ***************************************************************************
const char *CLuaState::_NELSmallScriptTableName= "NELSmallScriptTable";
uint CLuaStackChecker::_ExceptionContextCounter = 0;
// ***************************************************************************
void CLuaStackChecker::incrementExceptionContextCounter()
{
++ _ExceptionContextCounter;
}
// ***************************************************************************
void CLuaStackChecker::decrementExceptionContextCounter()
{
nlassert(_ExceptionContextCounter > 0);
-- _ExceptionContextCounter;
}
// ***************************************************************************
CLuaState::CLuaState()
{
#ifdef LUA_NEVRAX_VERSION
_State = lua_open(NULL, NULL);
#else
_State = lua_open();
#endif
nlassert(_State);
// *** Load base libs
{
CLuaStackChecker lsc(this);
luaopen_base (_State);
luaopen_table (_State);
luaopen_io (_State);
luaopen_string (_State);
luaopen_math (_State);
luaopen_debug (_State);
// open are buggy????
clear();
}
// *** Register basics
CLuaStackChecker lsc(this);
// do: LUA_REGISTRYINDEX.(lightuserdata*)this.classes= {}
pushLightUserData((void *) this);
newTable();
push("classes");
newTable(); // registry class
setTable(-3);
setTable(LUA_REGISTRYINDEX);
// add pointer from lua state to this CLuaState object
// do: LUA_REGISTRYINDEX.(lightuserdata*)_State= this
pushLightUserData((void *) _State); // NB : as creator of the state, we make the assumption that
// no one will be using this pointer in the registry (cf. ref manual about registry)
pushLightUserData((void *) this);
setTable(LUA_REGISTRYINDEX);
// Create the Table that contains Function cache for small script execution
push(_NELSmallScriptTableName); // 1:TableName
newTable(); // 1:TableName 2:table
setTable(LUA_REGISTRYINDEX); // ...
_SmallScriptPool= 0;
// *** luabind init
luabind::open(_State);
}
// ***************************************************************************
CLuaStackRestorer::CLuaStackRestorer(CLuaState *state, int finalSize) : _State(state), _FinalSize(finalSize)
{
}
// ***************************************************************************
CLuaStackRestorer::~CLuaStackRestorer()
{
nlassert(_State);
_State->setTop(_FinalSize);
}
// ***************************************************************************
CLuaState::~CLuaState()
{
nlassert(_State);
lua_close(_State);
// Clear Small Script Cache
_SmallScriptPool= 0;
_SmallScriptCache.clear();
}
// ***************************************************************************
CLuaState *CLuaState::fromStatePointer(lua_State *state)
{
nlassert(state);
int initialStackSize = lua_gettop(state);
lua_checkstack(state, initialStackSize + 2);
lua_pushlightuserdata(state, (void *) state);
lua_gettable(state, LUA_REGISTRYINDEX);
if (!lua_islightuserdata(state, -1))
{
lua_pop(state, 1);
return NULL;
}
CLuaState *ls = (CLuaState *) lua_touserdata(state, -1);
lua_pop(state, 1);
nlassert(initialStackSize == lua_gettop(state));
return ls;
}
// ***************************************************************************
struct CLuaReader
{
const std::string *Str;
bool Done;
};
void CLuaState::loadScript(const std::string &code, const std::string &dbgSrc)
{
struct CHelper
{
static const char *luaChunkReaderFromString(lua_State *L, void *ud, size_t *sz)
{
CLuaReader *rd = (CLuaReader *) ud;
if (!rd->Done)
{
rd->Done = true;
*sz = rd->Str->size();
return rd->Str->c_str();
}
else
{
*sz = 0;
return NULL;
}
}
};
CLuaReader rd;
rd.Str = &code;
rd.Done = false;
int result = lua_load(_State, CHelper::luaChunkReaderFromString, (void *) &rd, dbgSrc.c_str());
if (result !=0)
{
// pop the error code
string err= toString();
pop();
// throw error
throw ELuaParseError(err);
}
}
// ***************************************************************************
void CLuaState::executeScriptInternal(const std::string &code, const std::string &dbgSrc, int numRet)
{
CLuaStackChecker lsc(this, numRet);
// load the script
loadScript(code, dbgSrc);
// execute
if (pcall(0, numRet) != 0)
{
// pop the error code
string err= toString();
pop();
// throw error
throw ELuaExecuteError(err);
}
}
// ***************************************************************************
void CLuaState::executeScript(const std::string &code, int numRet)
{
// run the script, with dbgSrc==script
executeScriptInternal(code, code, numRet);
}
// ***************************************************************************
bool CLuaState::executeScriptNoThrow(const std::string &code, int numRet)
{
try
{
executeScript(code, numRet);
}
catch (ELuaError &e)
{
nlwarning(e.what());
return false;
}
return true;
}
// ***************************************************************************
bool CLuaState::executeFile(const std::string &pathName)
{
CIFile inputFile;
if(!inputFile.open(pathName))
return false;
// load the script text
string script;
/*
while(!inputFile.eof())
{
char tmpBuff[5000];
inputFile.getline(tmpBuff, 5000);
script+= tmpBuff;
script+= "\n";
}
*/
script.resize(NLMISC::CFile::getFileSize(pathName));
inputFile.serialBuffer((uint8 *) &script[0], script.size());
// execute the script text, with dbgSrc==filename (use @ for lua internal purpose)
executeScriptInternal(script, string("@") + NLMISC::CFile::getFilename(pathName));
return true;
}
// ***************************************************************************
void CLuaState::executeSmallScript(const std::string &script)
{
// *** if the small script has not already been called before, parse it now
TSmallScriptCache::iterator it= _SmallScriptCache.find(script);
if(it==_SmallScriptCache.end())
{
CLuaStackChecker lsc(this);
// add it to a function
loadScript(script, script);
// Assign the method to the NEL table: NELSmallScriptTable[_SmallScriptPool]= function
push(_NELSmallScriptTableName); // 1:function 2:NelTableName
getTable(LUA_REGISTRYINDEX); // 1:function 2:NelTable
insert(-2); // 1:NelTable 2:function
rawSetI(-2, _SmallScriptPool); // 1:NelTable
pop();
// bkup in cache map
it= _SmallScriptCache.insert(make_pair(script, _SmallScriptPool)).first;
// next allocated
_SmallScriptPool++;
}
// *** Execute the function associated to the script
CLuaStackChecker lsc(this);
push(_NELSmallScriptTableName); // 1:NelTableName
getTable(LUA_REGISTRYINDEX); // 1:NelTable
// get the function at the given index in the "NELSmallScriptTable" table
rawGetI(-1, it->second); // 1:NelTable 2:function
// execute
if (pcall(0, 0) != 0)
{
// Stack: 1: NelTable 2:errorcode
// pop the error code, and clear stack
string err= toString();
pop(); // 1:NelTable
pop(); // ....
// throw error
throw ELuaExecuteError(err);
}
else
{
// Stack: 1:NelTable
pop(); // ....
}
}
// ***************************************************************************
void CLuaState::registerFunc(const char *name, lua_CFunction function)
{
lua_register(_State, name, function);
}
// ***************************************************************************
void CLuaState::pushCClosure(lua_CFunction function, int n)
{
nlassert(function);
nlassert(getTop() >= n);
lua_pushcclosure(_State, function, n);
}
// ***************************************************************************
void CLuaState::push(TLuaWrappedFunction function)
{
struct CForwarder
{
static int callFunc(lua_State *ls)
{
nlassert(ls);
TLuaWrappedFunction func = (TLuaWrappedFunction) lua_touserdata(ls, lua_upvalueindex(1));
CLuaState *state = (CLuaState *) lua_touserdata(ls, lua_upvalueindex(2));
nlassert(func);
nlassert(state);
// get real function pointer from the values in the closure
int numResults;
int initialStackSize = state->getTop();
try
{
// call the actual function
numResults = func(*state);
}
catch(const std::exception &e)
{
// restore stack to its initial size
state->setTop(initialStackSize);
lua_pushstring(ls, e.what());
// TODO : see if this is safe to call lua error there" ... (it does a long jump)
lua_error(ls);
}
return numResults;
}
};
pushLightUserData((void *) function);
pushLightUserData((void *) this);
pushCClosure(CForwarder::callFunc, 2);
}
// ***************************************************************************
// Wrapped function
void CLuaState::registerFunc(const char *name, TLuaWrappedFunction function)
{
nlassert(function);
CLuaStackChecker lsc(this);
push(name);
push(function);
setTable(LUA_GLOBALSINDEX);
}
// ***************************************************************************
bool CLuaState::getTableBooleanValue(const char *name, bool defaultValue)
{
nlassert(name);
push(name);
getTable(-2);
if (isNil())
{
pop();
return defaultValue;
}
bool result = toBoolean(-1);
pop();
return result;
}
// ***************************************************************************
double CLuaState::getTableNumberValue(const char *name, double defaultValue)
{
nlassert(name);
push(name);
getTable(-2);
if (isNil())
{
pop();
return defaultValue;
}
double result = toNumber(-1);
pop();
return result;
}
// ***************************************************************************
const char *CLuaState::getTableStringValue(const char *name, const char *defaultValue)
{
nlassert(name);
push(name);
getTable(-2);
if (isNil())
{
pop();
return defaultValue;
}
const char *result = toString(-1);
pop();
return result;
}
// ***************************************************************************
void CLuaState::getStackContext(string &ret, uint stackLevel)
{
nlassert(_State);
ret.clear();
lua_Debug dbg;
if(lua_getstack (_State, stackLevel, &dbg))
{
if(lua_getinfo(_State, "lS", &dbg))
{
ret= NLMISC::toString("%s:%d:", dbg.short_src, dbg.currentline);
}
}
}
// ***************************************************************************
int CLuaState::pcallByName(const char *functionName, int nargs, int nresults, int funcTableIndex /*=LUA_GLOBALSINDEX*/, int errfunc /*= 0*/)
{
int initialStackSize = getTop();
nlassert(functionName);
nlassert(isTable(funcTableIndex));
pushValue(funcTableIndex);
push(functionName);
getTable(-2);
remove(-2); // get rid of the table
nlassert(getTop() >= nargs); // not enough arguments on the stack
// insert function before its arguments
insert(- 1 - nargs);
int result = pcall(nargs, nresults, errfunc);
int currSize = getTop();
if (result == 0)
{
nlassert(currSize == initialStackSize - nargs + nresults);
}
else
{
// errors, the stack contains a single string
if (errfunc == 0)
{
nlassert(currSize == initialStackSize - nargs + 1);
}
// else if there's an error handler, can't know the size of stack
}
return result;
}
// ***************************************************************************
void CLuaState::dumpStack()
{
nlinfo("LUA STACK CONTENT (size = %d)", getTop());
nlinfo("=================");
CLuaStackChecker lsc(this);
for(int k = 1; k <= getTop(); ++k)
{
pushValue(k);
std::string value = toString(-1) ? toString(-1) : "?";
nlinfo("Stack entry %d : type = %s, value = %s", k, getTypename(type(-1)), value.c_str());
pop();
}
}
// ***************************************************************************
void CLuaState::getStackAsString(std::string &dest)
{
dest = NLMISC::toString("Stack size = %d\n", getTop());
CLuaStackChecker lsc(this);
for(int k = 1; k <= getTop(); ++k)
{
pushValue(k);
std::string value = toString(-1) ? toString(-1) : "?";
dest += NLMISC::toString("Stack entry %d : type = %s, value = %s\n", k, getTypename(type(-1)), value.c_str());
pop();
}
}
//================================================================================
CLuaStackChecker::~CLuaStackChecker()
{
nlassert(_State);
if (!_ExceptionContextCounter)
{
int currSize = _State->getTop();
if (currSize != _FinalWantedSize)
{
static volatile bool assertWanted = true;
if (assertWanted)
{
nlwarning("Lua stack size error : expected size is %d, current size is %d", _FinalWantedSize, currSize);
_State->dumpStack();
nlassert(0);
}
}
}
else
{
// this object dtor was called because an exception was thrown, so let the exception
// propagate (the stack must be broken, but because of the exception, not because of code error)
_State->setTop(_FinalWantedSize);
}
}
// ***************************************************************************
void ELuaWrappedFunctionException::init(CLuaState *ls, const std::string &reason)
{
/* // Print first Lua Stack Context
CInterfaceManager *pIM= CInterfaceManager::getInstance();
if(ls)
{
ls->getStackContext(_Reason, 1); // 1 because 0 is the current C function => return 1 for script called
// enclose with cool colors
pIM->formatLuaStackContext(_Reason);
}
// Append the reason
_Reason+= reason;*/
}
// ***************************************************************************
ELuaWrappedFunctionException::ELuaWrappedFunctionException(CLuaState *luaState)
{
init(luaState, "");
}
// ***************************************************************************
ELuaWrappedFunctionException::ELuaWrappedFunctionException(CLuaState *luaState, const std::string &reason)
{
init(luaState, reason);
}
// ***************************************************************************
ELuaWrappedFunctionException::ELuaWrappedFunctionException(CLuaState *luaState, const char *format, ...)
{
std::string reason;
NLMISC_CONVERT_VARGS (reason, format, NLMISC::MaxCStringSize);
init(luaState, reason);
}