1 /* LuaEvent - Copyright (C) 2007 Thomas Harning <harningt@gmail.com>
2 * Licensed as LGPL - See doc/COPYING for details */
4 #include "event_buffer.h"
7 #define EVENT_BUFFER_MT "EVENT_BUFFER_MT"
9 #define BUFFER_ADD_CHECK_INPUT_FIRST 1
11 static le_buffer
* event_buffer_get(lua_State
* L
, int idx
) {
12 return (le_buffer
*)luaL_checkudata(L
, idx
, EVENT_BUFFER_MT
);
14 static int is_event_buffer(lua_State
* L
, int idx
) {
16 lua_getmetatable(L
, idx
);
17 luaL_getmetatable(L
, EVENT_BUFFER_MT
);
18 ret
= lua_rawequal(L
, -2, -1);
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
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
);
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
41 for(i
= 2; i
<= last
; i
++) {
43 if(lua_isstring(L
, i
)) {
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");
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");
54 lua_pushinteger(L
, EVBUFFER_LENGTH(buffer
) - oldLength
);
58 static luaL_Reg buffer_funcs
[] = {
59 {"add",event_buffer_add
},
62 static luaL_Ref funcs
[] = {
66 int event_buffer_register(lua_State
* L
) {
67 luaL_newmetatable(L
, EVENT_BUFFER_MT
);
69 luaL_register(L
, NULL
, buffer_funcs
);
70 lua_setfield(L
, -2, "__index");
73 luaL_register(L
, "luaevent.core.buffer", funcs
);