1 /*****************************************************************************
2 * stream.c: stream functions
3 *****************************************************************************
4 * Copyright (C) 2007-2008 the VideoLAN team
7 * Authors: Antoine Cellerier <dionoea at videolan tod org>
8 * Pierre d'Herbemont <pdherbemont # videolan.org>
10 * This program is free software; you can redistribute it and/or modify
11 * it under the terms of the GNU General Public License as published by
12 * the Free Software Foundation; either version 2 of the License, or
13 * (at your option) any later version.
15 * This program is distributed in the hope that it will be useful,
16 * but WITHOUT ANY WARRANTY; without even the implied warranty of
17 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
18 * GNU General Public License for more details.
20 * You should have received a copy of the GNU General Public License
21 * along with this program; if not, write to the Free Software
22 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
23 *****************************************************************************/
25 /*****************************************************************************
27 *****************************************************************************/
36 #include <vlc_common.h>
37 #include <vlc_plugin.h>
41 #include <lua.h> /* Low level lua C API */
42 #include <lauxlib.h> /* Higher level C API */
47 /*****************************************************************************
49 *****************************************************************************/
50 static int vlclua_stream_read( lua_State * );
51 static int vlclua_stream_readline( lua_State * );
52 static int vlclua_stream_delete( lua_State * );
53 static int vlclua_stream_add_filter( lua_State *L );
55 static const luaL_Reg vlclua_stream_reg[] = {
56 { "read", vlclua_stream_read },
57 { "readline", vlclua_stream_readline },
58 { "addfilter", vlclua_stream_add_filter },
62 static int vlclua_stream_new_inner( lua_State *L, stream_t *p_stream )
65 return luaL_error( L, "Error when opening stream" );
67 stream_t **pp_stream = lua_newuserdata( L, sizeof( stream_t * ) );
68 *pp_stream = p_stream;
70 if( luaL_newmetatable( L, "stream" ) )
73 luaL_register( L, NULL, vlclua_stream_reg );
74 lua_setfield( L, -2, "__index" );
75 lua_pushcfunction( L, vlclua_stream_delete );
76 lua_setfield( L, -2, "__gc" );
79 lua_setmetatable( L, -2 );
83 static int vlclua_stream_new( lua_State *L )
85 vlc_object_t * p_this = vlclua_get_this( L );
86 const char * psz_url = luaL_checkstring( L, 1 );
87 stream_t *p_stream = stream_UrlNew( p_this, psz_url );
88 return vlclua_stream_new_inner( L, p_stream );
91 static int vlclua_memory_stream_new( lua_State *L )
93 vlc_object_t * p_this = vlclua_get_this( L );
94 /* FIXME: duplicating the whole buffer is suboptimal. Keeping a reference to the string so that it doesn't get garbage collected would be better */
95 char * psz_content = strdup( luaL_checkstring( L, 1 ) );
96 stream_t *p_stream = stream_MemoryNew( p_this, (uint8_t *)psz_content, strlen( psz_content ), false );
97 return vlclua_stream_new_inner( L, p_stream );
100 static int vlclua_stream_read( lua_State *L )
103 stream_t **pp_stream = (stream_t **)luaL_checkudata( L, 1, "stream" );
104 int n = luaL_checkint( L, 2 );
105 uint8_t *p_read = malloc( n );
106 if( !p_read ) return vlclua_error( L );
107 i_read = stream_Read( *pp_stream, p_read, n );
108 lua_pushlstring( L, (const char *)p_read, i_read );
113 static int vlclua_stream_readline( lua_State *L )
115 stream_t **pp_stream = (stream_t **)luaL_checkudata( L, 1, "stream" );
116 char *psz_line = stream_ReadLine( *pp_stream );
119 lua_pushstring( L, psz_line );
127 static int vlclua_stream_add_filter( lua_State *L )
129 vlc_object_t *p_this = vlclua_get_this( L );
131 /* Make sure that we have 1 argument (+ 1 object) */
134 stream_t **pp_stream = (stream_t **)luaL_checkudata( L, 1, "stream" );
135 if( !*pp_stream ) return vlclua_error( L );
136 const char *psz_filter = NULL;
138 if( lua_isstring( L, 2 ) )
139 psz_filter = lua_tostring( L, 2 );
141 if( !psz_filter || !*psz_filter )
143 msg_Dbg( p_this, "adding all automatic stream filters" );
146 /* Add next automatic stream */
147 stream_t *p_filtered = stream_FilterNew( *pp_stream, NULL );
152 msg_Dbg( p_this, "inserted an automatic stream filter" );
153 *pp_stream = p_filtered;
156 luaL_getmetatable( L, "stream" );
157 lua_setmetatable( L, 1 );
161 /* Add a named filter */
162 stream_t *p_filter = stream_FilterNew( *pp_stream, psz_filter );
164 msg_Dbg( p_this, "Unable to open requested stream filter '%s'",
168 *pp_stream = p_filter;
169 luaL_getmetatable( L, "stream" );
170 lua_setmetatable( L, 1 );
177 static int vlclua_stream_delete( lua_State *L )
179 stream_t **pp_stream = (stream_t **)luaL_checkudata( L, 1, "stream" );
180 stream_Delete( *pp_stream );
184 /*****************************************************************************
186 *****************************************************************************/
187 void luaopen_stream( lua_State *L )
189 lua_pushcfunction( L, vlclua_stream_new );
190 lua_setfield( L, -2, "stream" );
191 lua_pushcfunction( L, vlclua_memory_stream_new );
192 lua_setfield( L, -2, "memory_stream" );