diff options
Diffstat (limited to 'lua/lsircbot.c')
-rw-r--r-- | lua/lsircbot.c | 113 |
1 files changed, 113 insertions, 0 deletions
diff --git a/lua/lsircbot.c b/lua/lsircbot.c new file mode 100644 index 0000000..0b97d5a --- /dev/null +++ b/lua/lsircbot.c @@ -0,0 +1,113 @@ +#include <sys/stat.h> +#include <sys/socket.h> +#include <sys/un.h> + +#include <fcntl.h> +#include <stdio.h> +#include <unistd.h> + +#include <lua.h> +#include <lualib.h> +#include <lauxlib.h> + +#define SBCONN_META "sircbot-connection" +#ifndef VERSION +#define VERSION "unknown" +#endif + +#if LUA_VERSION_NUM < 502 +# define luaL_newlib(L,l) (lua_newtable(L), luaL_register(L,NULL,l)) +# define luaL_setfuncs(L,l,n) (assert(n==0), luaL_register(L,NULL,l)) +#endif + +static int *Lsbconn_checkarg(lua_State *L, int index) +{ + int *fdp; + luaL_checktype(L, index, LUA_TUSERDATA); + fdp = (int *) luaL_checkudata(L, index, SBCONN_META); + if (fdp == NULL) { + const char *msg = lua_pushfstring(L, "%s expected, got %s", + SBCONN_META, + luaL_typename(L, index)); + luaL_argerror(L, index, msg); + } + return fdp; +} + +static int Lsbconn_new(lua_State *L) +{ + int *fdp; + const char *channel_name = luaL_checklstring(L, 1, NULL); + struct sockaddr_un sun; + fdp = (int *) lua_newuserdata(L, sizeof(int *)); + luaL_getmetatable(L, SBCONN_META); + lua_setmetatable(L, -2); + + *fdp = socket(PF_UNIX, SOCK_STREAM, 0); + if (*fdp < 0) + luaL_error(L, "Failed to create socket for '%s'", channel_name); + + sun.sun_family = AF_UNIX; + snprintf(sun.sun_path, sizeof(sun.sun_path), "/var/run/sircbot/%s", + channel_name); + if (connect(*fdp, (struct sockaddr *) &sun, sizeof(sun)) < 0) + luaL_error(L, "%s", sun.sun_path); + + return 1; +} + +static int Lsbconn_destroy(lua_State *L) +{ + int *fdp = Lsbconn_checkarg(L, 1); + close(*fdp); + return 1; +} + +static int Lsbconn_send(lua_State *L) +{ + int *fdp; + const char *msg; + int r; + + fdp = Lsbconn_checkarg(L, 1); + msg = luaL_checklstring(L, 2, NULL); + r = write(*fdp, msg, strlen(msg)); + lua_pushinteger(L, r); + return 1; +} + +static const luaL_Reg sbconn_meta_methods[] = { + { "__gc", Lsbconn_destroy }, + { NULL, NULL } +}; + +static const luaL_Reg sircbot_methods[] = { + { "connect", Lsbconn_new }, + { "send", Lsbconn_send }, + { NULL, NULL } +}; + +LUALIB_API int luaopen_sircbot(lua_State *L) +{ + /* register sircbot library */ + luaL_newlib(L, sircbot_methods); + + /* version */ + lua_pushliteral(L, "version"); + lua_pushliteral(L, VERSION); + lua_settable(L, -3); + + /* register metatable for it */ + luaL_newmetatable(L, SBCONN_META); + luaL_setfuncs(L, sbconn_meta_methods, 0); + lua_pushliteral(L, "__index"); + lua_pushvalue(L, -3); + lua_rawset(L, -3); + lua_pushliteral(L, "__metatable"); + lua_pushvalue(L, -3); + lua_rawset(L, -3); + lua_pop(L, 1); + + return 1; +} + |