diff options
author | Pixel <pixel@nobis-crew.org> | 2011-10-26 15:06:49 -0700 |
---|---|---|
committer | Pixel <pixel@nobis-crew.org> | 2011-10-26 15:06:49 -0700 |
commit | 58327525051d0e282b4bc2d0d7ddd1bbad2e9b65 (patch) | |
tree | e2d524de47749521e1b8c032a5af512e82fb1c49 /src | |
parent | e5804f2cb929083713c0082f39182a858c082587 (diff) |
Adding Lua.
Diffstat (limited to 'src')
-rw-r--r-- | src/BLua.cc | 854 |
1 files changed, 854 insertions, 0 deletions
diff --git a/src/BLua.cc b/src/BLua.cc new file mode 100644 index 0000000..9cfccf1 --- /dev/null +++ b/src/BLua.cc @@ -0,0 +1,854 @@ +#include <stdlib.h> +#include "BLua.h" +#include "Printer.h" +#include "Input.h" + +extern "C" { +#include <lualib.h> +#include <luajit.h> +} + +#ifndef BUFFERSIZE +#define BUFFERSIZE 2048 +#endif + +namespace Balau { + +class LuaStatics { + public: + static const char * getF(lua_State * L, void *, size_t *); + + static int callwrap(lua_State * L, lua_CFunction); + static int collector(lua_State * L); + static int destructor(lua_State * L); + + static int dumpvars(lua_State * L); + + static int iconv(lua_State * L); + static int mkdir(lua_State * L); + static int time(lua_State * L); + static int getenv(lua_State * L); + static int setenv(lua_State * L); + static int unsetenv(lua_State * L); + + static int hex(lua_State * L); + + static int globalindex(lua_State * L); + + static int print(lua_State * L); +}; + +}; + +int Balau::LuaStatics::hex(lua_State * __L) { + Lua L(__L); + int n = L.gettop(); + int x; + String r; + + if (((n != 1) && (n != 2)) || !L.isnumber(1) || ((n == 2) && !L.isstring(2))) + L.error("Incorrect arguments to function `hex'"); + + x = L.tonumber(1); + String fmt = n == 2 ? L.tostring() : "%02x"; + r.set(fmt.to_charp(), x); + + L.push(r); + + return 1; +} + +int Balau::LuaStatics::dumpvars(lua_State * __L) { + Lua L(__L); + int n = L.gettop(); + String prefix; + String varname; + + if ((n > 3) || (n < 2) || !L.isobject(1) || + ((n == 2) && !L.isstring(2)) || + ((n == 3) && !L.isstring(3) && !L.istable(2) && !L.isstring(2))) + L.error("Incorrect arguments to function `dumpvars'"); + + prefix = L.tostring(n); + + if (n == 3) + L.pop(); + + if (L.isstring(2)) + L.getglobal(L.tostring(2).to_charp()); + + IO<Handle> h(L.recast<Balau::Handle>()); + + L.dumpvars(h, prefix); + + return 0; +} + +int Balau::LuaStatics::iconv(lua_State * __L) { + Lua L(__L); + int n = L.gettop(); + String str, from, to; + + if ((n != 3) || !L.isstring(1) || !L.isstring(2) || !L.isstring(3)) + L.error("Incorrect arguments to function `string.iconv'"); + + str = L.tostring(1); + from = L.tostring(2); + to = L.tostring(3); + + str.iconv(from, to); + + L.push(str); + + return 1; +} + +int Balau::LuaStatics::mkdir(lua_State * __L) { + Lua L(__L); + int n = L.gettop(); + String dirname; + + if (n != 1) + L.error("Incorrect arguments to function `mkdir'"); + + dirname = L.tostring(1); + + Balau::FileSystem::mkdir(dirname.to_charp()); + + return 0; +} + +int Balau::LuaStatics::time(lua_State * __L) { + Lua L(__L); + + L.push((lua_Number) ::time(NULL)); + + return 1; +} + +int Balau::LuaStatics::getenv(lua_State * __L) { + Lua L(__L); + int n = L.gettop(); + + if (n != 1) + L.error("Incorrect arguments to function `getenv'"); + +#ifdef _WIN32 + char buffer[BUFSIZ + 1]; + if (GetEnvironmentVariable(L.tostring(1).to_charp(), buffer, BUFSIZ)) { + L.push(buffer); + } else { + L.push(); + } +#else + char * var = ::getenv(L.tostring(1).to_charp()); + if (var) { + L.push(var); + } else { + L.push(); + } +#endif + + return 1; +} + +int Balau::LuaStatics::setenv(lua_State * __L) { + Lua L(__L); + int n = L.gettop(); + + if (n != 2) { + L.error("Incorrect arguments to function `setenv'"); + } + +#ifdef _WIN32 + SetEnvironmentVariable(L.tostring(1).to_charp(), L.tostring(2).to_charp()); +#else + ::setenv(L.tostring(1).to_charp(), L.tostring(2).to_charp(), 1); +#endif + + return 0; +} + +int Balau::LuaStatics::unsetenv(lua_State * __L) { + Lua L(__L); + + int n = L.gettop(); + if (n != 1) + L.error("Incorrect arguments to function `unsetenv'"); + +#ifdef _WIN32 + SetEnvironmentVariable(L.tostring(1).to_charp(), NULL); +#else + ::unsetenv(L.tostring(1).to_charp()); +#endif + + return 0; +} + +int Balau::LuaStatics::print(lua_State * __L) { + Lua L(__L); + + int n = L.gettop(); + int i; + for (i = 1; i <= n; i++) { + const char *s; + s = lua_tostring(__L, i); + if (s == NULL) + L.error("`tostring' must return a string to `print'"); + if (i > 1) + Printer::print("\t"); + Printer::print(s); + L.pop(); + } + Printer::print("\n"); + return 0; +} + +int Balau::LuaStatics::callwrap(lua_State * __L, lua_CFunction func) { + Lua L(__L); + + try { + return func(__L); + } + catch (LuaException e) { + L.error(String("LuaException: ") + e.getMsg()); + } + catch (Balau::GeneralException e) { + L.error(String("GeneralException: ") + e.getMsg()); + } + catch (...) { + L.error("Unknown C++ exception"); + } + + return 0; +} + +struct ObjData { + void * ptr; + bool isObj; +}; + +int Balau::LuaStatics::collector(lua_State * __L) { + Lua L(__L); + ObjData * u = (ObjData *) L.touserdata(); + if (u->isObj) { + LuaExport * obj = (LuaExport *) u->ptr; + delete obj; + } else { + free(u->ptr); + } + u->ptr = NULL; + return 0; +} + +int Balau::LuaStatics::destructor(lua_State * __L) { + Lua L(__L); + L.push("__obj"); + L.gettable(-2, true); + ObjData * u = (ObjData *) L.touserdata(); + if (u->isObj) { + LuaExport * obj = (LuaExport *) u->ptr; + delete obj; + } else { + free(u->ptr); + } + u->ptr = NULL; + L.pop(); + return 0; +} + +void Balau::Lua::setCallWrap(lua_CallWrapper wrapper) { + push((void *) wrapper); + luaJIT_setmode(L, -1, LUAJIT_MODE_WRAPCFUNC | LUAJIT_MODE_ON); + pop(); +} + +Balau::Lua::Lua() : L(lua_open()) { + setCallWrap(LuaStatics::callwrap); + declareFunc("hex", LuaStatics::hex); + declareFunc("dumpvars", LuaStatics::dumpvars); + declareFunc("print", LuaStatics::print); + push("BLUA_THREADS"); + newtable(); + settable(LUA_REGISTRYINDEX); +} + +#define IntPoint(p) ((unsigned int)(lu_mem)(p)) +typedef size_t lu_mem; + +void Balau::Lua::weaken() { + push("BLUA_THREADS"); // -1 = "BLUA_THREADS" + gettable(LUA_REGISTRYINDEX); // -1 = BLUA_THREADS + push((lua_Number) IntPoint(L)); // -2 = BLUA_THREADS, -1 = key-Lt + push(); // -3 = BLUA_THREADS, -2 = key-Lt, -1 = nil + settable(); // -1 = BLUA_THREADS + pop(); +} + +void Balau::Lua::open_base() { + int n = gettop(); + luaopen_base(L); + push("mkdir"); + push(LuaStatics::mkdir); + settable(); + push("time"); + push(LuaStatics::time); + settable(); + push("getenv"); + push(LuaStatics::getenv); + settable(); + push("setenv"); + push(LuaStatics::setenv); + settable(); + push("unsetenv"); + push(LuaStatics::unsetenv); + settable(); + push("print"); + push(LuaStatics::print); + settable(); + while (n < gettop()) remove(n); + push("mkdir"); + push(LuaStatics::mkdir); + settable(LUA_GLOBALSINDEX); + push("time"); + push(LuaStatics::time); + settable(LUA_GLOBALSINDEX); + push("getenv"); + push(LuaStatics::getenv); + settable(LUA_GLOBALSINDEX); + push("setenv"); + push(LuaStatics::setenv); + settable(LUA_GLOBALSINDEX); + push("unsetenv"); + push(LuaStatics::unsetenv); + settable(LUA_GLOBALSINDEX); + push("print"); + push(LuaStatics::print); + settable(LUA_GLOBALSINDEX); +} + +void Balau::Lua::open_table() { + int n = gettop(); + luaopen_table(L); + while (n < gettop()) remove(n); +} + +void Balau::Lua::open_string() { + int n = gettop(); + luaopen_string(L); + push("iconv"); + push(LuaStatics::iconv); + settable(); + while (n < gettop()) remove(n); +} + +void Balau::Lua::open_math() { + int n = gettop(); + luaopen_math(L); + while (n < gettop()) remove(n); +} + +void Balau::Lua::open_debug() { + int n = gettop(); + luaopen_debug(L); + while (n < gettop()) remove(n); +} + +void Balau::Lua::open_jit() { + int n = gettop(); + luaopen_jit(L); + while (n < gettop()) remove(n); +} + +void Balau::Lua::open_bit() { + int n = gettop(); + luaopen_bit(L); + while (n < gettop()) remove(n); +} + +void Balau::Lua::declareFunc(const char * name, lua_CFunction f, int i) { + checkstack(2); + lua_pushstring(L, name); + lua_pushcfunction(L, f); + lua_settable(L, i); +} + +void Balau::Lua::call(const char * f, int i, int nargs) { + checkstack(1); + lua_pushstring(L, f); + lua_gettable(L, i); + lua_insert(L, -1 - nargs); + call(nargs); +} + +void Balau::Lua::settable(int i, bool raw) { + if (raw) { + lua_rawset(L, i); + } else { + lua_settable(L, i); + } +} + +void Balau::Lua::gettable(int i, bool raw) { + if (raw) { + lua_rawget(L, i); + } else { + lua_gettable(L, i); + } +} + +void Balau::Lua::getglobal(const char * name) throw (GeneralException) { + push(name); + gettable(LUA_GLOBALSINDEX); +} + +void Balau::Lua::pushLuaContext() { + String whole_msg; + struct lua_Debug ar; + bool got_error = false; + int level = 0; + + do { + if (lua_getstack(L, level, &ar) == 1) { + if (lua_getinfo(L, "nSl", &ar) != 0) { + push(String("at ") + ar.source + ":" + ar.currentline + " (" + (ar.name ? ar.name : "[top]") + ")"); + } else { + got_error = true; + } + } else { + got_error = true; + } + level++; + } while (!got_error); +} + +void Balau::Lua::error(const char * msg) { + push(msg); + + lua_error(L); +} + +bool Balau::Lua::isobject(int i) { + bool r = false; + if (istable(i)) { + push("__obj"); + gettable(i); + r = isuserdata(); + pop(); + } else { + r = isnil(i); + } + return r; +} + +Balau::String Balau::Lua::tostring(int i) { + const char * r = 0; + size_t l = -1; + switch (type(i)) { + case LUA_TNIL: + r = "(nil)"; + break; + case LUA_TBOOLEAN: + r = toboolean(i) ? "true" : "false"; + break; + case LUA_TNUMBER: + return String(tonumber(i)); + break; + default: + r = lua_tolstring(L, i, &l); + } + return String(r ? r : "<lua-NULL>", l); +} + +struct LoadF { + Balau::IO<Balau::Handle> f; + char buff[BUFFERSIZE]; +}; + +const char * Balau::LuaStatics::getF(lua_State * L, void * ud, size_t * size) { + LoadF * lf = (LoadF *)ud; + + *size = lf->f->read(lf->buff, BUFFERSIZE); + return (*size > 0) ? lf->buff : NULL; +} + +Balau::String Balau::Lua::escapeString(const String & s) { + String r = ""; + int i; + for (i = 0; i < s.strlen(); i++) { + switch(s[i]) { + case '"': case '\\': + r += '\\'; + r += s[i]; + break; + case '\n': + r += "\\n"; + break; + case '\r': + r += "\\r"; + break; + case '\0': + r += "\\000"; + break; + default: + r += s[i]; + } + } + return r; +} + +void Balau::Lua::load(IO<Handle> h, bool docall) throw (GeneralException) { + LoadF lf; + int status; + + lf.f = h; + + checkstack(); + String name = h->getName(); + IO<Input> i = h; + if (!i.isNull()) + name = String("@") + i->getFName(); + status = lua_load(L, LuaStatics::getF, &lf, name.to_charp()); + + if (status) { + pushLuaContext(); + showerror(); + throw LuaException(String("Error loading lua chunk from Handle `") + h->getName() + "'"); + } + + if (docall) + call(); +} + +void Balau::Lua::load(const String & s, bool docall) throw (GeneralException) { + const char * buf = s.to_charp(); + int status; + + status = luaL_loadbuffer(L, buf, s.strlen(), buf); + + if (status) { + pushLuaContext(); + showerror(); + throw LuaException(String("Error loading lua string `") + s + "'"); + } + + if (docall) + call(); +} + +void Balau::Lua::dumpvars(IO<Handle> h, const String & prefix, int i) { + h->writeString(prefix); + h->writeString(" = {\n"); + dumpvars_r(h, i); + h->writeString("}\n"); +} + +void Balau::Lua::dumpvars_r(IO<Handle> h, int i, int depth) throw (GeneralException) { + int j; + String t; + bool dump_value; + + if (lua_type(L, i) != LUA_TTABLE) + throw LuaException("Error dumping variables: variable isn't a table."); + + push(); + + if (i < 0) + i--; + + depth++; + + checkstack(); + while(lua_next(L, i) != 0) { + if (lua_type(L, -2) == LUA_TSTRING) + if (String(lua_tostring(L, -2)) == String("_G")) + continue; + for (j = 0; j < depth; j++) + h->writeString(" "); + + dump_value = true; + + // first, let's dump the key. + switch(lua_type(L, -2)) { + case LUA_TNONE: + throw LuaException("Internal error: got invalid index for key while dumpvars."); + case LUA_TNIL: + throw LuaException("Internal error: got nil index for key while dumpvars."); + case LUA_TNUMBER: + t.set("[%.14g] = ", lua_tonumber(L, -2)); + break; + case LUA_TBOOLEAN: + t = String("[") + (lua_toboolean(L, -2) ? "true" : "false") + "] = "; + break; + case LUA_TSTRING: + t = String("[\"") + escapeString(lua_tostring(L, -2)) + String("\"] = "); + break; + case LUA_TTABLE: + t = "-- [a table]\n"; + dump_value = false; + break; + case LUA_TFUNCTION: + t = "-- [function() ... end]\n"; + dump_value = false; + break; + case LUA_TUSERDATA: + t = "-- [userdata]\n"; + dump_value = false; + break; + case LUA_TTHREAD: + t = "-- [thread]\n"; + dump_value = false; + break; + default: + throw LuaException("Internal error: got unknow index for key while dumpvars."); + } + + // Seems that we can't dump that key. + if (!dump_value) { + pop(); + h->writeString(t); + continue; + } + + // let's look at the value: if it's a function, a userdata or a thread, we can't dump it. + if ((lua_type(L, -1) == LUA_TFUNCTION) || + (lua_type(L, -1) == LUA_TUSERDATA) || + (lua_type(L, -1) == LUA_TTHREAD)) + h->writeString("-- "); + + h->writeString(t); + + // Finally, let's dump the value. + switch(lua_type(L, -1)) { + case LUA_TNONE: + throw LuaException("Internal error: got invalid index for value while dumpvars."); + case LUA_TNIL: + h->writeString("nil,\n"); + case LUA_TNUMBER: + t.set("%.14g,\n", lua_tonumber(L, -1)); + h->writeString(t); + break; + case LUA_TBOOLEAN: + h->writeString(lua_toboolean(L, -1) ? "true" : "false"); + h->writeString(",\n"); + break; + case LUA_TSTRING: + h->writeString("\""); + h->writeString(escapeString(lua_tostring(L, -1))); + h->writeString("\",\n"); + break; + case LUA_TTABLE: + h->writeString("{\n"); + dumpvars_r(h, -1, depth); + for (j = 0; j < depth; j++) + h->writeString(" "); + h->writeString("},\n"); + break; + case LUA_TFUNCTION: + h->writeString("function() ... end\n"); + break; + case LUA_TUSERDATA: + h->writeString("userdata ...\n"); + break; + case LUA_TTHREAD: + h->writeString("thread ...\n"); + break; + default: + throw LuaException("Internal error: got unknow index for value while dumpvars."); + } + + pop(); + } +} + +Balau::Lua Balau::Lua::thread(bool saveit) { + checkstack(); + lua_State * L1 = lua_newthread(L); + if (saveit) { // -1 = thread + push("BLUA_THREADS"); // -2 = thread, -1 = "BLUA_THREADS" + gettable(LUA_REGISTRYINDEX); // -2 = thread, -1 = BLUA_THREADS + push((lua_Number) IntPoint(L1)); // -3 = thread, -2 = BLUA_THREADS, -1 = key-Lt + copy(-3); // -4 = thread, -3 = BLUA_THREADS, -2 = key-Lt, -1 = thread + settable(); // -2 = thread, -1 = BLUA_THREADS + pop(); // -1 = thread + } + return Lua(L1); +} + +Balau::Lua Balau::Lua::thread(const String & code, int nargs, bool saveit) { + Lua L1; + L1 = thread(saveit); + L1.resume(code, nargs); + return L1; +} + +Balau::Lua Balau::Lua::thread(IO<Handle> h, int nargs, bool saveit) { + Lua L1; + L1 = thread(saveit); + L1.resume(h, nargs); + return L1; +} + +bool Balau::Lua::resume(int nargs) throw (GeneralException) { + int r; + + r = lua_resume(L, nargs); + + if ((r == 0) || (r == LUA_YIELD)) + return 0; + + pushLuaContext(); + showerror(); + + switch(r) { + case 0: + return false; + case LUA_YIELD: + return true; + case LUA_ERRRUN: + throw LuaException("Runtime error while running LUA code."); + case LUA_ERRMEM: + throw LuaException("Memory allocation error while running LUA code."); + case LUA_ERRERR: + throw LuaException("Error in Error function."); + case LUA_ERRSYNTAX: + throw LuaException("Syntax error in Lua code."); + default: + throw LuaException(String("Unknow error while running LUA code (err code: ") + String(r) + ")"); + } +} + +bool Balau::Lua::resume(IO<Handle> h, int nargs) { + bool r; + + load(h, false); + r = resume(nargs); + + return r; +} + +bool Balau::Lua::resume(const String & s, int nargs) { + bool r; + + load(s, false); + r = resume(nargs); + + return r; +} + +void Balau::Lua::showstack(int level) { + int n = lua_gettop(L); + int i; + String t; + + if (n == 0) { + Printer::log(level, "Stack empty\n"); + return; + } + + for (i = 1; i <= n; i++) { + switch(lua_type(L, i)) { + case LUA_TNONE: + t = "Invalid"; + break; + case LUA_TNIL: + t = "(Nil)"; + break; + case LUA_TNUMBER: + t.set("(Number) %f", lua_tonumber(L, i)); + break; + case LUA_TBOOLEAN: + t = String("(Bool) ") + (lua_toboolean(L, i) ? "true" : "false"); + break; + case LUA_TSTRING: + t = String("(String) ") + lua_tostring(L, i); + break; + case LUA_TTABLE: + t = "(Table)"; + break; + case LUA_TFUNCTION: + t = "(Function)"; + break; + default: + t = "Unknown"; + } + + Printer::log(level, String(i) + ": " + t + "\n"); + } +} + +void Balau::Lua::showerror() { + Printer::log(M_ERROR, "Lua object: Got an LUA error, inspecting stack.\n"); + + showstack(M_ERROR); +} + +void Balau::LuaObject::push(Lua & L) throw (GeneralException) { + if (m_pushed && m_wantsDestruct) + throw GeneralException("Error: object is owned by the LUA script and can not be pushed."); + L.newtable(); + pushMembers(L); + m_pushed = true; +} + +void Balau::LuaObject::pushMe(Lua & L, void * o, const char * objname, bool obj) { + ObjData * u; + L.push("__obj"); + u = (ObjData *) L.newuser(sizeof(ObjData)); + u->ptr = o; + u->isObj = obj; + L.settable(-3, true); + if (objname && *objname) { + L.push("__objname"); + L.push(objname); + L.settable(-3, true); + } +} + +void * Balau::LuaObject::getMeInternal(Lua & L, int i) { + ObjData * u = NULL; + + if (L.istable(i)) { + L.push("__obj"); + L.gettable(i, true); + if (!(u = (ObjData *) L.touserdata())) + L.error("Table is not an object."); + if (!u->ptr) + L.error("Object already destroyed."); + L.pop(); + } else if (L.isnil(i)) { + u = NULL; + } else { + L.error("Not an object (not even a table)."); + } + + return u ? u->ptr : NULL; +} + +void Balau::LuaObject::pushIt(Lua & L, const char * s, lua_CFunction f) { + L.push(s); + L.push(f); + L.settable(-3, true); +} + +void Balau::LuaObject::pushMeta(Lua & L, const char * s, lua_CFunction f) { + if (!L.getmetatable()) + L.newtable(); + L.push(s); + L.push(f); + L.settable(); + L.setmetatable(); +} + +void Balau::LuaObject::pushDestruct(Lua & L) throw (GeneralException) { + if (m_pushed) + throw GeneralException("Error: can't push destructor, object already pushed"); + push(L); + L.push("__obj"); + L.gettable(-2, true); + pushMeta(L, "__gc", LuaStatics::collector); + L.pop(); + pushIt(L, "destroy", LuaStatics::destructor); + + m_wantsDestruct = true; +} |