Mercurial > luaevent-prosody / file comparison
comparison: src/event_buffer.c
src/event_buffer.c
- changeset 26
- a117895e867c
- child 27
- 0d82e4f49a56
equal
deleted
inserted
replaced
| 25:5778073d2903 | 26:a117895e867c |
|---|---|
| 1 /* LuaEvent - Copyright (C) 2007 Thomas Harning <harningt@gmail.com> | |
| 2 * Licensed as LGPL - See doc/COPYING for details */ | |
| 3 | |
| 4 #include "event_buffer.h" | |
| 5 #include <lauxlib.h> | |
| 6 | |
| 7 #define EVENT_BUFFER_MT "EVENT_BUFFER_MT" | |
| 8 | |
| 9 #define BUFFER_ADD_CHECK_INPUT_FIRST 1 | |
| 10 | |
| 11 static le_buffer* event_buffer_get(lua_State* L, int idx) { | |
| 12 return (le_buffer*)luaL_checkudata(L, idx, EVENT_BUFFER_MT); | |
| 13 } | |
| 14 static int is_event_buffer(lua_State* L, int idx) { | |
| 15 int ret; | |
| 16 lua_getmetatable(L, idx); | |
| 17 luaL_getmetatable(L, EVENT_BUFFER_MT); | |
| 18 ret = lua_rawequal(L, -2, -1); | |
| 19 lua_pop(L, 2); | |
| 20 return ret; | |
| 21 } | |
| 22 | |
| 23 /* LUA: buffer:add(...) | |
| 24 progressively adds items to the buffer | |
| 25 if arg[*] is string, treat as a string:format call | |
| 26 if arg[*] is a buffer, perform event_add_buffer | |
| 27 returns number of bytes added | |
| 28 */ | |
| 29 int event_buffer_add(lua_State* L) { | |
| 30 le_buffer* buf = event_buffer_get(L, 1); | |
| 31 struct evbuffer* buffer = buf->buffer; | |
| 32 int oldLength = EVBUFFER_LENGTH(buffer); | |
| 33 int last = lua_top(L); | |
| 34 int i; | |
| 35 for(i = 2; i <= last; i++) { | |
| 36 if(!lua_isstring(L, i) && !is_event_buffer(L, i)) | |
| 37 luaL_argerror(L, i, "Argument is not a string or buffer object"); | |
| 38 /* Optionally perform checks and data loading separately to avoid overfilling the buffer */ | |
| 39 #if BUFFER_ADD_CHECK_INPUT_FIRST | |
| 40 } | |
| 41 for(i = 2; i <= last; i++) { | |
| 42 #endif | |
| 43 if(lua_isstring(L, i)) { | |
| 44 size_t len; | |
| 45 const char* data = lua_tolstring(L, i, &len); | |
| 46 if(0 != evbuffer_add(buffer, data, len)) | |
| 47 luaL_error(L, "Failed to add data to the buffer"); | |
| 48 } else { | |
| 49 le_buffer* buf2 = event_buffer_get(L, i); | |
| 50 if(0 != evbuffer_add_buffer(buffer, buf2->buffer)) | |
| 51 luaL_error(L, "Failed to move buffer-data to the buffer"); | |
| 52 } | |
| 53 } | |
| 54 lua_pushinteger(L, EVBUFFER_LENGTH(buffer) - oldLength); | |
| 55 return 1; | |
| 56 } | |
| 57 | |
| 58 static luaL_Reg buffer_funcs[] = { | |
| 59 {"add",event_buffer_add}, | |
| 60 {NULL, NULL} | |
| 61 }; | |
| 62 static luaL_Ref funcs[] = { | |
| 63 {NULL, NULL} | |
| 64 }; | |
| 65 | |
| 66 int event_buffer_register(lua_State* L) { | |
| 67 luaL_newmetatable(L, EVENT_BUFFER_MT); | |
| 68 lua_newtable(L); | |
| 69 luaL_register(L, NULL, buffer_funcs); | |
| 70 lua_setfield(L, -2, "__index"); | |
| 71 lua_pop(L, 1); | |
| 72 | |
| 73 luaL_register(L, "luaevent.core.buffer", funcs); | |
| 74 return 0; | |
| 75 } | |
| 76 |
