diff options
-rw-r--r-- | stream.c | 26 | ||||
-rw-r--r-- | stream.h | 9 |
2 files changed, 33 insertions, 2 deletions
@@ -1,11 +1,13 @@ #include "stream.h" +#include <stdlib.h> + #include <lauxlib.h> #include <lua.h> /// Creates and pushes new Stream into the Lua stack. /// \param L Lua state to push to -/// \param fd File descriptor used by stream +/// \param fd File descriptor used by the Stream /// \return TODO int stream_push_new(lua_State * L, const int fd) { @@ -14,10 +16,30 @@ int stream_push_new(lua_State * L, const int fd) if (1 == luaL_newmetatable(L, "stream")) { - // TODO: initialize metatable for stream + lua_pushstring(L, "__gc"); + lua_pushcfunction(L, stream_gc); + lua_rawset(L, -3); } lua_setmetatable(L, -2); return LUA_OK; } + +/// Metamethod to handle garbage collection of Stream userdata. +/// \param L Lua state in which Stream resides +/// \return Always zero, which is the number of the results pushed to the stack +int stream_gc(lua_State * L) +{ + struct stream * s = lua_touserdata(L, -1); + + if (NULL != s) + { + if (NULL != s->in.data) + { + free(s->in.data); + } + } + + return 0; +} @@ -2,9 +2,18 @@ #include <lua.h> +struct buffer +{ + char * data; + int length; + int offset; +}; + struct stream { int fd; + struct buffer in; }; int stream_push_new(lua_State *, const int); +int stream_gc(lua_State *); |