aboutsummaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
authorThomas Harning Jr <harningt@gmail.com>2007-09-06 23:22:55 -0400
committerThomas Harning Jr <harningt@gmail.com>2007-09-06 23:22:55 -0400
commitc8d56ebf5f7deb3891fb2c60b91b74f71a1d3f2c (patch)
tree6b21330c5e113476fbed203e4239302bc477e7d3
parent64a4223203928443ee5a49dbb7579e1db53a4056 (diff)
downloadluaevent-prosody-c8d56ebf5f7deb3891fb2c60b91b74f71a1d3f2c.tar.gz
luaevent-prosody-c8d56ebf5f7deb3891fb2c60b91b74f71a1d3f2c.zip
Added event_buffer object + 'add' functionality
-rw-r--r--include/event_buffer.h18
-rw-r--r--src/event_buffer.c76
-rw-r--r--src/luaevent.c2
3 files changed, 96 insertions, 0 deletions
diff --git a/include/event_buffer.h b/include/event_buffer.h
new file mode 100644
index 0000000..b34ea2d
--- /dev/null
+++ b/include/event_buffer.h
@@ -0,0 +1,18 @@
+/* LuaEvent - Copyright (C) 2007 Thomas Harning <harningt@gmail.com>
+ * Licensed as LGPL - See doc/COPYING for details */
+#ifndef EVENT_BUFFER_H
+#define EVENT_BUFFER_H
+
+#include "luaevent.h"
+#include <lua.h>
+#include <sys/types.h>
+#include <sys/time.h>
+#include <event.h>
+
+typedef struct {
+ struct evbuffer* buffer;
+} le_buffer;
+
+int event_buffer_register(lua_State* L);
+
+#endif
diff --git a/src/event_buffer.c b/src/event_buffer.c
new file mode 100644
index 0000000..15c11a9
--- /dev/null
+++ b/src/event_buffer.c
@@ -0,0 +1,76 @@
+/* LuaEvent - Copyright (C) 2007 Thomas Harning <harningt@gmail.com>
+ * Licensed as LGPL - See doc/COPYING for details */
+
+#include "event_buffer.h"
+#include <lauxlib.h>
+
+#define EVENT_BUFFER_MT "EVENT_BUFFER_MT"
+
+#define BUFFER_ADD_CHECK_INPUT_FIRST 1
+
+static le_buffer* event_buffer_get(lua_State* L, int idx) {
+ return (le_buffer*)luaL_checkudata(L, idx, EVENT_BUFFER_MT);
+}
+static int is_event_buffer(lua_State* L, int idx) {
+ int ret;
+ lua_getmetatable(L, idx);
+ luaL_getmetatable(L, EVENT_BUFFER_MT);
+ ret = lua_rawequal(L, -2, -1);
+ lua_pop(L, 2);
+ return ret;
+}
+
+/* LUA: buffer:add(...)
+ progressively adds items to the buffer
+ if arg[*] is string, treat as a string:format call
+ if arg[*] is a buffer, perform event_add_buffer
+ returns number of bytes added
+*/
+int event_buffer_add(lua_State* L) {
+ le_buffer* buf = event_buffer_get(L, 1);
+ struct evbuffer* buffer = buf->buffer;
+ int oldLength = EVBUFFER_LENGTH(buffer);
+ int last = lua_top(L);
+ int i;
+ for(i = 2; i <= last; i++) {
+ if(!lua_isstring(L, i) && !is_event_buffer(L, i))
+ luaL_argerror(L, i, "Argument is not a string or buffer object");
+/* Optionally perform checks and data loading separately to avoid overfilling the buffer */
+#if BUFFER_ADD_CHECK_INPUT_FIRST
+ }
+ for(i = 2; i <= last; i++) {
+#endif
+ if(lua_isstring(L, i)) {
+ size_t len;
+ const char* data = lua_tolstring(L, i, &len);
+ if(0 != evbuffer_add(buffer, data, len))
+ luaL_error(L, "Failed to add data to the buffer");
+ } else {
+ le_buffer* buf2 = event_buffer_get(L, i);
+ if(0 != evbuffer_add_buffer(buffer, buf2->buffer))
+ luaL_error(L, "Failed to move buffer-data to the buffer");
+ }
+ }
+ lua_pushinteger(L, EVBUFFER_LENGTH(buffer) - oldLength);
+ return 1;
+}
+
+static luaL_Reg buffer_funcs[] = {
+ {"add",event_buffer_add},
+ {NULL, NULL}
+};
+static luaL_Ref funcs[] = {
+ {NULL, NULL}
+};
+
+int event_buffer_register(lua_State* L) {
+ luaL_newmetatable(L, EVENT_BUFFER_MT);
+ lua_newtable(L);
+ luaL_register(L, NULL, buffer_funcs);
+ lua_setfield(L, -2, "__index");
+ lua_pop(L, 1);
+
+ luaL_register(L, "luaevent.core.buffer", funcs);
+ return 0;
+}
+ \ No newline at end of file
diff --git a/src/luaevent.c b/src/luaevent.c
index 8903af1..c1428c1 100644
--- a/src/luaevent.c
+++ b/src/luaevent.c
@@ -3,6 +3,7 @@
#include "luaevent.h"
#include "event_callback.h"
+#include "event_buffer.h"
#include <lua.h>
#include <lauxlib.h>
@@ -119,6 +120,7 @@ void setNamedIntegers(lua_State* L, namedInteger* p) {
int luaopen_luaevent_core(lua_State* L) {
/* Register external items */
event_callback_register(L);
+ event_buffer_register(L);
/* Setup metatable */
luaL_newmetatable(L, EVENT_BASE_MT);
lua_newtable(L);