src/event_buffer.c

   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 void event_buffer_check(lua_State* L, int idx) {
  15         le_buffer* buf = (le_buffer*)luaL_checkudata(L, idx, EVENT_BUFFER_MT);
  16         if(!buf->buffer)
  17                 luaL_argerror(L, idx, "Attempt to use closed event_buffer object");
  18         return buf;
  19 }
  20 static int is_event_buffer(lua_State* L, int idx) {
  21         int ret;
  22         lua_getmetatable(L, idx);
  23         luaL_getmetatable(L, EVENT_BUFFER_MT);
  24         ret = lua_rawequal(L, -2, -1);
  25         lua_pop(L, 2);
  26         return ret;
  27 }
  28
  29 /* TODO: Use lightuserdata mapping to locate hanging object instances */
  30 static int event_buffer_push(lua_State* L, struct evbuffer* buffer) {
  31         le_buffer *buf = (le_buffer*)lua_newuserdata(L, sizeof(le_buffer));
  32         buf->buffer = buffer;
  33         luaL_getmetatable(L, EVENT_BUFFER_MT);
  34         lua_setmetatable(L, -2);
  35         return 1;
  36 }
  37
  38 static int event_buffer_push_new(lua_State* L) {
  39         return event_buffer_push(L, evbuffer_new());
  40 }
  41
  42 static int event_buffer_gc(lua_State* L) {
  43         le_buffer* buf = event_buffer_get(L, 1);
  44         if(buf->buffer) {
  45                 evbuffer_free(buf->buffer);
  46                 buf->buffer = NULL;
  47         }
  48         return 0;
  49 }
  50
  51 /* LUA: buffer:add(...)
  52         progressively adds items to the buffer
  53                 if arg[*] is string, treat as a string:format call
  54                 if arg[*] is a buffer, perform event_add_buffer
  55         returns number of bytes added
  56 */
  57 static int event_buffer_add(lua_State* L) {
  58         le_buffer* buf = event_buffer_check(L, 1);
  59         struct evbuffer* buffer = buf->buffer;
  60         int oldLength = EVBUFFER_LENGTH(buffer);
  61         int last = lua_top(L);
  62         int i;
  63         for(i = 2; i <= last; 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");
  66 /* Optionally perform checks and data loading separately to avoid overfilling the buffer */
  67 #if BUFFER_ADD_CHECK_INPUT_FIRST
  68         }
  69         for(i = 2; i <= last; i++) {
  70 #endif
  71                 if(lua_isstring(L, i)) {
  72                         size_t len;
  73                         const char* data = lua_tolstring(L, i, &len);
  74                         if(0 != evbuffer_add(buffer, data, len))
  75                                 luaL_error(L, "Failed to add data to the buffer");
  76                 } else {
  77                         le_buffer* buf2 = event_buffer_check(L, i);
  78                         if(0 != evbuffer_add_buffer(buffer, buf2->buffer))
  79                                 luaL_error(L, "Failed to move buffer-data to the buffer");
  80                 }
  81         }
  82         lua_pushinteger(L, EVBUFFER_LENGTH(buffer) - oldLength);
  83         return 1;
  84 }
  85
  86 static int event_buffer_get_length(lua_State* L) {
  87         le_buffer* buf = event_buffer_check(L, 1);
  88         lua_pushinteger(L, EVBUFFER_LENGTH(buf->buffer));
  89         return 1;
  90 }
  91
  92 /* MAYBE: Could add caching */
  93 static int event_buffer_get_data(lua_State* L) {
  94         le_buffer* buf = event_buffer_check(L, 1);
  95         lua_pushlstring(L, EVBUFFER_DATA(buf->buffer), EVBUFFER_LENGTH(buf->buffer));
  96         return 1;
  97 }
  98
  99 static int event_buffer_drain(lua_State* L) {
 100         le_buffer* buf = event_buffer_check(L, 1);
 101         size_t len = luaL_checkinteger(L, 2);
 102         evbuffer_drain(buf->buffer, len);
 103         return 0;
 104 }
 105
 106 static luaL_Reg buffer_funcs[] = {
 107         {"add",event_buffer_add},
 108         {"length",event_buffer_get_length},
 109         {"get_data",event_buffer_get_data},
 110         {"drain",event_buffer_drain},
 111         {"close",event_buffer_gc},
 112         {NULL, NULL}
 113 };
 114 static luaL_Ref funcs[] = {
 115         {"new",event_buffer_push_new},
 116         {NULL, NULL}
 117 };
 118
 119 int event_buffer_register(lua_State* L) {
 120         luaL_newmetatable(L, EVENT_BUFFER_MT);
 121         lua_pushcfunction(L, event_buffer_gc);
 122         lua_setfield(L, -2, "__gc");
 123         lua_newtable(L);
 124         luaL_register(L, NULL, buffer_funcs);
 125         lua_setfield(L, -2, "__index");
 126         lua_pop(L, 1);
 127
 128         luaL_register(L, "luaevent.core.buffer", funcs);
 129         return 0;
 130 }
 131