Mercurial > luaevent-prosody / file comparison

comparison: src/event_buffer.c

src/event_buffer.c

changeset 28
f7b864b661f4
parent 27
0d82e4f49a56
child 29
f6d0eb94ca6c
equal deleted inserted replaced
27:0d82e4f49a56 28:f7b864b661f4
9 #define BUFFER_ADD_CHECK_INPUT_FIRST 1 9 #define BUFFER_ADD_CHECK_INPUT_FIRST 1
10 10
11 static le_buffer* event_buffer_get(lua_State* L, int idx) { 11 static le_buffer* event_buffer_get(lua_State* L, int idx) {
12 return (le_buffer*)luaL_checkudata(L, idx, EVENT_BUFFER_MT); 12 return (le_buffer*)luaL_checkudata(L, idx, EVENT_BUFFER_MT);
13 } 13 }
14 static void event_buffer_check(lua_State* L, int idx) { 14 static le_buffer* event_buffer_check(lua_State* L, int idx) {
15 le_buffer* buf = (le_buffer*)luaL_checkudata(L, idx, EVENT_BUFFER_MT); 15 le_buffer* buf = (le_buffer*)luaL_checkudata(L, idx, EVENT_BUFFER_MT);
16 if(!buf->buffer) 16 if(!buf->buffer)
17 luaL_argerror(L, idx, "Attempt to use closed event_buffer object"); 17 luaL_argerror(L, idx, "Attempt to use closed event_buffer object");
18 return buf; 18 return buf;
19 } 19 }
56 */ 56 */
57 static int event_buffer_add(lua_State* L) { 57 static int event_buffer_add(lua_State* L) {
58 le_buffer* buf = event_buffer_check(L, 1); 58 le_buffer* buf = event_buffer_check(L, 1);
59 struct evbuffer* buffer = buf->buffer; 59 struct evbuffer* buffer = buf->buffer;
60 int oldLength = EVBUFFER_LENGTH(buffer); 60 int oldLength = EVBUFFER_LENGTH(buffer);
61 int last = lua_top(L); 61 int last = lua_gettop(L);
62 int i; 62 int i;
63 for(i = 2; i <= last; i++) { 63 for(i = 2; i <= last; i++) {
64 if(!lua_isstring(L, i) && !is_event_buffer(L, i)) 64 if(!lua_isstring(L, i) && !is_event_buffer(L, i))
65 luaL_argerror(L, i, "Argument is not a string or buffer object"); 65 luaL_argerror(L, i, "Argument is not a string or buffer object");
66 /* Optionally perform checks and data loading separately to avoid overfilling the buffer */ 66 /* Optionally perform checks and data loading separately to avoid overfilling the buffer */
90 } 90 }
91 91
92 /* MAYBE: Could add caching */ 92 /* MAYBE: Could add caching */
93 static int event_buffer_get_data(lua_State* L) { 93 static int event_buffer_get_data(lua_State* L) {
94 le_buffer* buf = event_buffer_check(L, 1); 94 le_buffer* buf = event_buffer_check(L, 1);
95 lua_pushlstring(L, EVBUFFER_DATA(buf->buffer), EVBUFFER_LENGTH(buf->buffer)); 95 lua_pushlstring(L, (const char*)EVBUFFER_DATA(buf->buffer), EVBUFFER_LENGTH(buf->buffer));
96 return 1; 96 return 1;
97 } 97 }
98 98
99 static int event_buffer_drain(lua_State* L) { 99 static int event_buffer_drain(lua_State* L) {
100 le_buffer* buf = event_buffer_check(L, 1); 100 le_buffer* buf = event_buffer_check(L, 1);
109 {"get_data",event_buffer_get_data}, 109 {"get_data",event_buffer_get_data},
110 {"drain",event_buffer_drain}, 110 {"drain",event_buffer_drain},
111 {"close",event_buffer_gc}, 111 {"close",event_buffer_gc},
112 {NULL, NULL} 112 {NULL, NULL}
113 }; 113 };
114 static luaL_Ref funcs[] = { 114 static luaL_Reg funcs[] = {
115 {"new",event_buffer_push_new}, 115 {"new",event_buffer_push_new},
116 {NULL, NULL} 116 {NULL, NULL}
117 }; 117 };
118 118
119 int event_buffer_register(lua_State* L) { 119 int event_buffer_register(lua_State* L) {
126 lua_pop(L, 1); 126 lua_pop(L, 1);
127 127
128 luaL_register(L, "luaevent.core.buffer", funcs); 128 luaL_register(L, "luaevent.core.buffer", funcs);
129 return 0; 129 return 0;
130 } 130 }
131

Mercurial Repository: luaevent-prosody

mercurial