From d46748ac59c3b8d8895d2671aa4b1eca3914d1d4 Mon Sep 17 00:00:00 2001 From: Aki Date: Wed, 24 Feb 2021 20:43:11 +0100 Subject: Added buffer and garbage collection to stream --- stream.c | 26 ++++++++++++++++++++++++-- 1 file changed, 24 insertions(+), 2 deletions(-) (limited to 'stream.c') diff --git a/stream.c b/stream.c index d6a5254..a322e2b 100644 --- a/stream.c +++ b/stream.c @@ -1,11 +1,13 @@ #include "stream.h" +#include + #include #include /// 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; +} -- cgit v1.1