diff options
author | Natanael Copa <natanael.copa@gmail.com> | 2010-09-10 10:30:53 +0200 |
---|---|---|
committer | Natanael Copa <natanael.copa@gmail.com> | 2010-09-10 10:31:29 +0200 |
commit | 855ccd6c97cc29efd494abe56608f44a7974d3fe (patch) | |
tree | f01a33193bdc63dcca499c7da719bec253c2456e | |
parent | 62522d63f12d180b86d3ecc8e97532aa4ddb77d9 (diff) | |
download | sircbot-855ccd6c97cc29efd494abe56608f44a7974d3fe.tar.bz2 sircbot-855ccd6c97cc29efd494abe56608f44a7974d3fe.tar.xz |
send clients
-rw-r--r-- | Makefile | 21 | ||||
-rw-r--r-- | lsircbot.c | 104 | ||||
-rw-r--r-- | sircbot-send.c | 40 |
3 files changed, 160 insertions, 5 deletions
@@ -4,15 +4,26 @@ GIT_VERSION := $(shell [ -d .git ] && echo "-$$(git describe --always)") CFLAGS ?= -g CFLAGS += -DVERSION=\"$(VERSION)$(GIT_VERSION)\" -all: sircbot +TARGETS = sircbot sircbot-send -sircbot_OBJ = sircbot.o irc.o +ifneq (NOLUA,) +TARGETS += sircbot.so +LUA_LIBS=-llua +endif -sircbot: $(sircbot_OBJ) +all: $(TARGETS) + +sircbot: sircbot.o irc.o + $(CC) $(LDFLAGS) -o $@ $^ + +sircbot-send: sircbot-send.o $(CC) $(LDFLAGS) -o $@ $^ +sircbot.so: lsircbot.o + $(CC) -shared $(LDFLAGS) -o $@ $^ $(LUA_LIBS) + .c.o: - $(CC) $(CFLAGS) -c $^ + $(CC) -fPIC $(CFLAGS) -c $^ clean: - rm -rf sircbot $(sircbot_OBJ) + rm -f *.o $(TARGETS) diff --git a/lsircbot.c b/lsircbot.c new file mode 100644 index 0000000..c283f0e --- /dev/null +++ b/lsircbot.c @@ -0,0 +1,104 @@ +#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 + +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) + luaL_typerror(L, index, SBCONN_META); + 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_register(L, "sircbot", 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_register(L, NULL, sbconn_meta_methods); + 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; +} + diff --git a/sircbot-send.c b/sircbot-send.c new file mode 100644 index 0000000..6f109d3 --- /dev/null +++ b/sircbot-send.c @@ -0,0 +1,40 @@ +#include <sys/stat.h> +#include <sys/socket.h> +#include <sys/un.h> + +#include <fcntl.h> +#include <stdio.h> +#include <unistd.h> + +int connect_channel(const char *channel_name) +{ + struct sockaddr_un sun; + int fd = socket(PF_UNIX, SOCK_STREAM, 0); + if (fd < 0) + return fd; + + sun.sun_family = AF_UNIX; + snprintf(sun.sun_path, sizeof(sun.sun_path), "/var/run/sircbot/%s", + channel_name); + if (connect(fd, (struct sockaddr *) &sun, sizeof(sun)) == 0) + return fd; + perror(sun.sun_path); + close(fd); + return -1; +} + +int main(int argc, char *argv[]) +{ + int i; + int fd = connect_channel(argv[1]); + char buf[2048]; + + if (fd < 0) + return 1; + + while (fgets(buf, sizeof(buf), stdin)) + if (write(fd, buf, strlen(buf)) < 0) + err(1, "write"); + return 0; +} + |