1
0
mirror of https://github.com/Llewellynvdm/conky.git synced 2024-11-18 11:05:18 +00:00

C++ binding for lua, with some fancy features

such as being able to throw exceptions across c++-lua boundary. For more details, see comments
in the source code.
This commit is contained in:
Pavel Labath 2010-02-10 17:34:09 +01:00
parent 9fefa79551
commit e4461782ef
3 changed files with 726 additions and 1 deletions

View File

@ -38,7 +38,7 @@ set(conky_sources colours.cc combine.cc common.cc conky.cc core.cc
diskio.cc entropy.cc exec.cc fs.cc mail.cc mixer.cc net_stat.cc template.cc diskio.cc entropy.cc exec.cc fs.cc mail.cc mixer.cc net_stat.cc template.cc
mboxscan.cc read_tcp.cc scroll.cc specials.cc tailhead.cc mboxscan.cc read_tcp.cc scroll.cc specials.cc tailhead.cc
temphelper.cc text_object.cc timeinfo.cc top.cc algebra.cc prioqueue.c proc.cc temphelper.cc text_object.cc timeinfo.cc top.cc algebra.cc prioqueue.c proc.cc
user.cc) user.cc luamm.cc)
# add timed thread library # add timed thread library
add_library(timed-thread timed-thread.cc) add_library(timed-thread timed-thread.cc)

458
src/luamm.cc Normal file
View File

@ -0,0 +1,458 @@
/* -*- mode: c; c-basic-offset: 4; tab-width: 4; indent-tabs-mode: t -*-
* vim: ts=4 sw=4 noet ai cindent syntax=cpp
*
* luamm: C++ binding for lua
*
* Copyright (C) 2010 Pavel Labath et al.
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU 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 General Public License for more details.
* You should have received a copy of the GNU General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>.
*
*/
#include "luamm.hh"
namespace lua {
namespace {
// keys for storing values in lua registry
const char cpp_exception_metatable[] = "lua::cpp_exception_metatable";
const char cpp_function_metatable [] = "lua::cpp_function_metatable";
const char lua_exception_namespace[] = "lua::lua_exception_namespace";
const char this_cpp_object [] = "lua::this_cpp_object";
// destructor for C++ objects stored as lua userdata
template<typename T>
int destroy_cpp_object(lua_State *l)
{
T *ptr = static_cast<T *>(lua_touserdata(l, -1));
assert(ptr);
ptr->~T();
return 0;
}
// converts C++ exceptions to strings, so lua can do something with them
int exception_to_string(lua_State *l)
{
std::exception_ptr *ptr = static_cast<std::exception_ptr *>(lua_touserdata(l, -1));
assert(ptr);
try {
std::rethrow_exception(*ptr);
}
catch(std::exception &e) {
lua_pushstring(l, e.what());
}
catch(...) {
lua_pushstring(l, ptr->__cxa_exception_type()->name());
}
return 1;
}
int absindex(lua_State *l, int index) throw()
{ return index<0 && -index<=lua_gettop(l) ? lua_gettop(l)+1+index : index; }
// Just like getfield(), only without calling metamethods (or throwing random exceptions)
inline void rawgetfield(lua_State *l, int index, const char *k) throw(std::bad_alloc)
{
index = absindex(l, index);
if(not lua_checkstack(l, 1))
throw std::bad_alloc();
lua_pushstring(l, k);
lua_rawget(l, index);
}
// Just like setfield(), only without calling metamethods (or throwing random exceptions)
inline void rawsetfield(lua_State *l, int index, const char *k) throw(std::bad_alloc)
{
index = absindex(l, index);
if(not lua_checkstack(l, 2))
throw std::bad_alloc();
lua_pushstring(l, k);
lua_insert(l, -2);
lua_rawset(l, index);
}
int closure_trampoline(lua_State *l)
{
lua_checkstack(l, 2);
rawgetfield(l, REGISTRYINDEX, this_cpp_object);
assert(lua_islightuserdata(l, -1));
state *L = static_cast<state *>( lua_touserdata(l, -1) );
lua_pop(l, 1);
try {
cpp_function *fn = static_cast<cpp_function *>( L->touserdata(lua_upvalueindex(1)) );
assert(fn);
return (*fn)(L);
}
catch(lua::exception &e) {
// rethrow lua errors as such
e.push_lua_error(L);
}
catch(...) {
// C++ exceptions (pointers to them, actually) are stored as lua userdata and
// then thrown
void *ptr = L->newuserdata(sizeof(std::exception_ptr));
L->rawgetfield(REGISTRYINDEX, cpp_exception_metatable);
L->setmetatable(-2);
new(ptr) std::exception_ptr(std::current_exception());
}
// lua_error does longjmp(), so destructors for objects in this function will not be
// called
return lua_error(l);
}
/*
* This function is called when lua encounters an error outside of any protected
* environment
* Throwing the exception through lua code appears to work, even if it was compiled
* without -fexceptions. If it turns out, it fails in some conditions, it could be
* replaced with some longjmp() magic. But that shouldn't be necessary, as this function
* will not be called under normal conditions (we execute everything in protected mode).
*/
int panic_throw(lua_State *l)
{
if(not lua_checkstack(l, 1))
throw std::bad_alloc();
rawgetfield(l, REGISTRYINDEX, this_cpp_object);
assert(lua_islightuserdata(l, -1));
state *L = static_cast<state *>( lua_touserdata(l, -1) );
lua_pop(l, 1);
throw lua::exception(L);
}
// protected mode wrappers for various lua functions
int safe_concat_trampoline(lua_State *l)
{
lua_concat(l, lua_gettop(l));
return 1;
}
template<int (*compare)(lua_State *, int, int)>
int safe_compare_trampoline(lua_State *l)
{
int r = compare(l, 1, 2);
lua_pop(l, 2);
lua_pushinteger(l, r);
return 1;
}
int safe_gc_trampoline(lua_State *l)
{
int what = lua_tointeger(l, -2);
int data = lua_tointeger(l, -1);
lua_pop(l, 2);
lua_pushinteger(l, lua_gc(l, what, data));
return 1;
}
template<void (*misc)(lua_State *, int), int nresults>
int safe_misc_trampoline(lua_State *l)
{
misc(l, 1);
return nresults;
}
int safe_next_trampoline(lua_State *l)
{
int r = lua_next(l, 1);
lua_checkstack(l, 1);
lua_pushinteger(l, r);
return r ? 3 : 1;
}
}
std::string exception::get_error_msg(state *L)
{
static const std::string default_msg("Unknown lua exception");
try {
return L->tostring(-1);
}
catch(not_string_error &e) {
return default_msg;
}
}
exception::exception(state *l)
: std::runtime_error(get_error_msg(l)), L(l)
{
L->checkstack(1);
L->rawgetfield(REGISTRYINDEX, lua_exception_namespace);
L->insert(-2);
key = L->ref(-2);
L->pop(1);
}
exception::~exception() throw()
{
if(not L)
return;
L->checkstack(1);
L->rawgetfield(REGISTRYINDEX, lua_exception_namespace);
L->unref(-1, key);
L->pop();
}
void exception::push_lua_error(state *l)
{
if(l != L)
throw std::runtime_error("Cannot transfer exceptions between different lua contexts");
l->checkstack(2);
l->rawgetfield(REGISTRYINDEX, lua_exception_namespace);
l->rawgeti(-1, key);
l->replace(-2);
}
state::state()
{
if(lua_State *l = luaL_newstate())
cobj.reset(l, &lua_close);
else {
// docs say this can happen only in case of a memory allocation error
throw std::bad_alloc();
}
// set our panic function
lua_atpanic(cobj.get(), panic_throw);
checkstack(2);
// store a pointer to ourselves
pushlightuserdata(this);
rawsetfield(REGISTRYINDEX, this_cpp_object);
// a metatable for C++ exceptions travelling through lua code
newmetatable(cpp_exception_metatable);
lua_pushcfunction(cobj.get(), &exception_to_string);
rawsetfield(-2, "__tostring");
pushboolean(false);
rawsetfield(-2, "__metatable");
lua_pushcfunction(cobj.get(), &destroy_cpp_object<std::exception_ptr>);
rawsetfield(-2, "__gc");
pop();
// a metatable for C++ functions callable from lua code
newmetatable(cpp_function_metatable);
lua_pushcfunction(cobj.get(), &destroy_cpp_object<cpp_function>);
rawsetfield(-2, "__gc");
pop();
// while they're travelling through C++ code, lua exceptions will reside here
newtable();
rawsetfield(REGISTRYINDEX, lua_exception_namespace);
luaL_openlibs(cobj.get());
}
void state::call(int nargs, int nresults, int errfunc)
{
int r = lua_pcall(cobj.get(), nargs, nresults, errfunc);
if(r == 0)
return;
if(r == LUA_ERRMEM) {
// memory allocation error, cross your fingers
throw std::bad_alloc();
}
checkstack(3);
rawgetfield(REGISTRYINDEX, cpp_exception_metatable);
if(getmetatable(-2)) {
if(rawequal(-1, -2)) {
// it's a C++ exception, rethrow it
std::exception_ptr *ptr = static_cast<std::exception_ptr *>(touserdata(-3));
assert(ptr);
/*
* we create a copy, so we can pop the object without fearing the exception will
* be collected by lua's GC
*/
std::exception_ptr t(*ptr); ptr = NULL;
pop(3);
std::rethrow_exception(t);
}
pop(2);
}
// it's a lua exception, wrap it
if(r == LUA_ERRERR)
throw lua::errfunc_error(this);
else
throw lua::exception(this);
}
void state::checkstack(int extra) throw(std::bad_alloc)
{
if(not lua_checkstack(cobj.get(), extra))
throw std::bad_alloc();
}
void state::concat(int n)
{
assert(n>=0);
checkstack(1);
lua_pushcfunction(cobj.get(), safe_concat_trampoline);
insert(-n-1);
call(n, 1, 0);
}
bool state::equal(int index1, int index2)
{
// avoid pcall overhead in trivial cases
if( rawequal(index1, index2) )
return true;
return safe_compare(&safe_compare_trampoline<lua_equal>, index1, index2);
}
int state::gc(int what, int data)
{
checkstack(3);
lua_pushcfunction(cobj.get(), safe_gc_trampoline);
pushinteger(what);
pushinteger(data);
call(2, 1, 0);
assert(isnumber(-1));
int r = tointeger(-1);
pop();
return r;
}
void state::getfield(int index, const char *k)
{
checkstack(1);
index = absindex(index);
pushstring(k);
gettable(index);
}
void state::gettable(int index)
{
checkstack(2);
pushvalue(index);
insert(-2);
lua_pushcfunction(cobj.get(), (&safe_misc_trampoline<&lua_gettable, 1>));
insert(-3);
call(2, 1, 0);
}
bool state::lessthan(int index1, int index2)
{
return safe_compare(&safe_compare_trampoline<&lua_lessthan>, index1, index2);
}
void state::loadstring(const char *s) throw(lua::syntax_error, std::bad_alloc)
{
switch(luaL_loadstring(cobj.get(), s)) {
case 0:
return;
case LUA_ERRSYNTAX:
throw lua::syntax_error(this);
case LUA_ERRMEM:
throw std::bad_alloc();
default:
assert(0);
}
}
bool state::next(int index)
{
checkstack(2);
pushvalue(index);
insert(-2);
lua_pushcfunction(cobj.get(), &safe_next_trampoline);
insert(-3);
call(2, MULTRET, 0);
assert(isnumber(-1));
int r = tointeger(-1);
pop();
return r;
}
void state::pushclosure(const cpp_function &fn, int n)
{
checkstack(2);
void *ptr = newuserdata(sizeof(cpp_function));
rawgetfield(REGISTRYINDEX, cpp_function_metatable);
setmetatable(-2);
new(ptr) cpp_function(fn);
insert(-n-1);
lua_pushcclosure(cobj.get(), &closure_trampoline, n+1);
}
void state::rawgetfield(int index, const char *k) throw(std::bad_alloc)
{ lua::rawgetfield(cobj.get(), index, k); }
void state::rawsetfield(int index, const char *k) throw(std::bad_alloc)
{ lua::rawsetfield(cobj.get(), index, k); }
bool state::safe_compare(lua_CFunction trampoline, int index1, int index2)
{
// if one of the indexes is invalid, return false
if(isnone(index1) || isnone(index2))
return false;
// convert relative indexes into absolute
index1 = absindex(index1);
index2 = absindex(index2);
checkstack(3);
lua_pushcfunction(cobj.get(), trampoline);
pushvalue(index1);
pushvalue(index2);
call(2, 1, 0);
assert(isnumber(-1));
int r = tointeger(-1);
pop();
return r;
}
void state::setfield(int index, const char *k)
{
checkstack(1);
index = absindex(index);
pushstring(k);
insert(-2);
settable(index);
}
void state::settable(int index)
{
checkstack(2);
pushvalue(index);
insert(-3);
lua_pushcfunction(cobj.get(), (&safe_misc_trampoline<&lua_settable, 0>));
insert(-4);
call(2, 0, 0);
}
std::string state::tostring(int index) throw(lua::not_string_error)
{
size_t len;
const char *str = lua_tolstring(cobj.get(), index, &len);
if(not str)
throw not_string_error();
return std::string(str, len);
}
}

267
src/luamm.hh Normal file
View File

@ -0,0 +1,267 @@
/* -*- mode: c; c-basic-offset: 4; tab-width: 4; indent-tabs-mode: t -*-
* vim: ts=4 sw=4 noet ai cindent syntax=cpp
*
* luamm: C++ binding for lua
*
* Copyright (C) 2010 Pavel Labath et al.
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU 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 General Public License for more details.
* You should have received a copy of the GNU General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>.
*
*/
#include <assert.h>
#include <memory>
#include <stdexcept>
#include <lua.hpp>
namespace lua {
class state;
typedef lua_Integer integer;
typedef lua_Number number;
typedef std::function<int(state *)> cpp_function;
enum {
ENVIRONINDEX = LUA_ENVIRONINDEX,
GLOBALSINDEX = LUA_GLOBALSINDEX,
REGISTRYINDEX = LUA_REGISTRYINDEX
};
enum {
GCSTOP = LUA_GCSTOP,
GCRESTART = LUA_GCRESTART,
GCCOLLECT = LUA_GCCOLLECT,
GCCOUNT = LUA_GCCOUNT,
GCCOUNTB = LUA_GCCOUNTB,
GCSTEP = LUA_GCSTEP,
GCSETPAUSE = LUA_GCSETPAUSE,
GCSETSTEPMUL = LUA_GCSETSTEPMUL
};
enum {
MULTRET = LUA_MULTRET
};
enum {
TBOOLEAN = LUA_TBOOLEAN,
TFUNCTION = LUA_TFUNCTION,
TLIGHTUSERDATA = LUA_TLIGHTUSERDATA,
TNIL = LUA_TNIL,
TNONE = LUA_TNONE,
TNUMBER = LUA_TNUMBER,
TSTRING = LUA_TSTRING,
TTABLE = LUA_TTABLE,
TTHREAD = LUA_TTHREAD,
TUSERDATA = LUA_TUSERDATA
};
// we reserve one upvalue for the function pointer
int upvalueindex(int n)
{ return lua_upvalueindex(n+1); }
/*
* Lua error()s are wrapped in this class when rethrown into C++ code. what() returns the
* error message. push_lua_error() pushes the error onto lua stack. The error can only be
* pushed into the same state it was generated in.
*/
class exception: public std::runtime_error {
/*
* We only allow moving, to avoid complications with multiple references. It shouldn't be
* difficult to modify this to work with copying, if that proves unavoidable.
*/
state *L;
int key;
static std::string get_error_msg(state *L);
exception(const exception &) = delete;
const exception& operator=(const exception &) = delete;
public:
exception(exception &&other)
: std::runtime_error(std::move(other)), L(other.L), key(other.key)
{ other.L = NULL; }
explicit exception(state *l);
virtual ~exception() throw();
void push_lua_error(state *l);
};
class not_string_error: public std::runtime_error {
public:
not_string_error()
: std::runtime_error("Cannot convert value to a string")
{}
};
// the name says it all
class syntax_error: public lua::exception {
syntax_error(const syntax_error &) = delete;
const syntax_error& operator=(const syntax_error &) = delete;
public:
syntax_error(state *L)
: lua::exception(L)
{}
syntax_error(syntax_error &&other)
: lua::exception(std::move(other))
{}
};
// double fault, lua encountered an error while running the error handler function
class errfunc_error: public lua::exception {
errfunc_error(const errfunc_error &) = delete;
const errfunc_error& operator=(const errfunc_error &) = delete;
public:
errfunc_error(state *L)
: lua::exception(L)
{}
errfunc_error(errfunc_error &&other)
: lua::exception(std::move(other))
{}
};
// a fancy wrapper around lua_State
class state {
std::shared_ptr<lua_State> cobj;
bool safe_compare(lua_CFunction trampoline, int index1, int index2);
public:
state();
/*
* Lua functions come in three flavours
* a) functions that never throw an exception
* b) functions that throw only in case of a memory allocation error
* c) functions that throw other kinds of errors
*
* Calls to type a functions are simply forwarded to the C api.
* Type c functions are executed in protected mode, to make sure they don't longjmp()
* over us (and our destructors). This add a certain amount overhead. If you care about
* performance, try using the raw versions (if possible).
* Type b functions are not executed in protected mode atm. as memory allocation errors
* don't happen that often (as opposed to the type c, where the user get deliberately set
* a metamethod that throws an error). That means those errors will do something
* undefined, but hopefully that won't be a problem.
*
* Semantics are mostly identical to those of the underlying C api. Any deviation is
* noted in the respective functions comment. The most important difference is that
* instead of return values, we use exceptions to indicate errors. The lua and C++
* exception mechanisms are integrated. That means one can throw a C++ exception and
* catch it in lua (with pcall). Lua error()s can be caught in C++ as exceptions of type
* lua::exception.
*/
// type a, never throw
int absindex(int index) throw() { return index<0 && -index<=gettop() ? gettop()+1+index : index; }
bool getmetatable(int index) throw() { return lua_getmetatable(cobj.get(), index); }
int gettop() throw() { return lua_gettop(cobj.get()); }
void insert(int index) throw() { lua_insert(cobj.get(), index); }
bool isfunction(int index) throw() { return lua_isfunction(cobj.get(), index); }
bool islightuserdata(int index) throw() { return lua_islightuserdata(cobj.get(), index); }
bool isnone(int index) throw() { return lua_isnone(cobj.get(), index); }
bool isnumber(int index) throw() { return lua_isnumber(cobj.get(), index); }
bool isstring(int index) throw() { return lua_isstring(cobj.get(), index); }
void pop(int n = 1) throw() { lua_pop(cobj.get(), n); }
void pushboolean(bool b) throw() { lua_pushboolean(cobj.get(), b); }
void pushinteger(integer n) throw() { lua_pushinteger(cobj.get(), n); }
void pushlightuserdata(void *p) throw() { lua_pushlightuserdata(cobj.get(), p); }
void pushnil() throw() { lua_pushnil(cobj.get()); }
void pushvalue(int index) throw() { lua_pushvalue(cobj.get(), index); }
void rawget(int index) throw() { lua_rawget(cobj.get(), index); }
void rawgeti(int index, int n) throw() { lua_rawgeti(cobj.get(), index, n); }
bool rawequal(int index1, int index2) throw() { return lua_rawequal(cobj.get(), index1, index2); }
void rawset(int index) throw() { lua_rawset(cobj.get(), index); }
void replace(int index) throw() { lua_replace(cobj.get(), index); }
// lua_setmetatable returns int, but docs don't specify it's meaning :/
int setmetatable(int index) throw() { return lua_setmetatable(cobj.get(), index); }
integer tointeger(int index) throw() { return lua_tointeger(cobj.get(), index); }
number tonumber(int index) throw() { return lua_tonumber(cobj.get(), index); }
void* touserdata(int index) throw() { return lua_touserdata(cobj.get(), index); }
int type(int index) throw() { return lua_type(cobj.get(), index); }
// typename is a reserved word :/
const char* type_name(int tp) throw() { return lua_typename(cobj.get(), tp); }
void unref(int t, int ref) throw() { return luaL_unref(cobj.get(), t, ref); }
// type b, throw only on memory allocation errors
// checkstack correctly throws bad_alloc, because lua_checkstack kindly informs us of
// that sitution
void checkstack(int extra) throw(std::bad_alloc);
bool newmetatable(const char *tname) { return luaL_newmetatable(cobj.get(), tname); }
void newtable() { lua_newtable(cobj.get()); }
void *newuserdata(size_t size) { return lua_newuserdata(cobj.get(), size); }
// cpp_function can be anything that std::function can handle, everything else remains
// identical
void pushclosure(const cpp_function &fn, int n);
void pushfunction(const cpp_function &fn) { pushclosure(fn, 0); }
void pushstring(const char *s) { lua_pushstring(cobj.get(), s); }
void rawgetfield(int index, const char *k) throw(std::bad_alloc);
void rawsetfield(int index, const char *k) throw(std::bad_alloc);
int ref(int t) { return luaL_ref(cobj.get(), t); }
// len recieves length, if not null. Returned value may contain '\0'
const char* tocstring(int index, size_t *len = NULL) { return lua_tolstring(cobj.get(), index, len); }
// type c, throw everything but the kitchen sink
// call() is a protected mode call, we don't allow unprotected calls
void call(int nargs, int nresults, int errfunc = 0);
void concat(int n);
bool equal(int index1, int index2);
int gc(int what, int data);
void getfield(int index, const char *k);
void gettable(int index);
void getglobal(const char *name) { getfield(GLOBALSINDEX, name); }
bool lessthan(int index1, int index2);
void loadstring(const char *s) throw(lua::syntax_error, std::bad_alloc);
bool next(int index);
// register is a reserved word :/
void register_fn(const char *name, const cpp_function &f) { pushfunction(f); setglobal(name); }
void setfield(int index, const char *k);
void setglobal(const char *name) { setfield(GLOBALSINDEX, name); }
void settable(int index);
// lua_tostring uses NULL to indicate conversion error, since there is no such thing as a
// NULL std::string, we throw an exception. Returned value may contain '\0'
std::string tostring(int index) throw(lua::not_string_error);
};
/*
* Can be used to automatically pop temporary values off the lua stack on exit from the
* function/block (e.g. via an exception). The constructor parameter indicates the number of
* values to pop(). That can be later changed with the overloaded operators. The idiom is:
* stack_sentry s(L);
* L.an_operation_that_pushes_something(); ++s;
* ...
*/
class stack_sentry {
state *L;
int n;
stack_sentry(const stack_sentry &) = delete;
const stack_sentry& operator=(const stack_sentry &) = delete;
public:
explicit stack_sentry(state &l, int n_ = 0) throw()
: L(&l), n(n_)
{}
~stack_sentry() throw() { L->pop(n); }
void operator++() throw() { ++n; }
void operator--() throw() { --n; }
void operator+=(int n_) throw() { n+=n_; }
void operator-=(int n_) throw() { n-=n_; }
};
}