misc: fix warnings
[vlc.git] / modules / misc / lua / httpd.c
1 /*****************************************************************************
2  * httpd.c: HTTPd wrapper
3  *****************************************************************************
4  * Copyright (C) 2007 the VideoLAN team
5  * $Id$
6  *
7  * Authors: Antoine Cellerier <dionoea at videolan tod org>
8  *
9  * This program is free software; you can redistribute it and/or modify
10  * it under the terms of the GNU General Public License as published by
11  * the Free Software Foundation; either version 2 of the License, or
12  * (at your option) any later version.
13  *
14  * This program is distributed in the hope that it will be useful,
15  * but WITHOUT ANY WARRANTY; without even the implied warranty of
16  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
17  * GNU General Public License for more details.
18  *
19  * You should have received a copy of the GNU General Public License
20  * along with this program; if not, write to the Free Software
21  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
22  *****************************************************************************/
23
24 /*****************************************************************************
25  * Preamble
26  *****************************************************************************/
27 #ifndef  _GNU_SOURCE
28 #   define  _GNU_SOURCE
29 #endif
30
31 #ifdef HAVE_CONFIG_H
32 # include "config.h"
33 #endif
34
35 #include <vlc/vlc.h>
36 #include <vlc_httpd.h>
37
38 #include <lua.h>        /* Low level lua C API */
39 #include <lauxlib.h>    /* Higher level C API */
40 #include <lualib.h>     /* Lua libs */
41
42 #include "vlc.h"
43
44 /*****************************************************************************
45  * Local prototypes
46  *****************************************************************************/
47 static uint8_t *vlclua_todata( lua_State *L, int narg, int *i_data );
48
49 /*****************************************************************************
50  * HTTPD Host
51  *****************************************************************************/
52 #if 0
53 /* This is kind of a useless function since TLS with the 4 last args
54  * unset does the same thing as far as I know. */
55 int vlclua_httpd_host_new( lua_State *L )
56 {
57     vlc_object_t *p_this = vlclua_get_this( L );
58     const char *psz_host = luaL_checkstring( L, 1 );
59     int i_port = luaL_checkint( L, 2 );
60     httpd_host_t *p_httpd_host = httpd_HostNew( p_this, psz_host, i_port );
61     if( !p_httpd_host )
62         return luaL_error( L, "Failed to create HTTP host \"%s:%d\".",
63                            psz_host, i_port );
64     vlclua_push_vlc_object( L, p_httpd_host, vlclua_httpd_host_delete );
65     return 1;
66 }
67 #endif
68
69 int vlclua_httpd_tls_host_new( lua_State *L )
70 {
71     vlc_object_t *p_this = vlclua_get_this( L );
72     const char *psz_host = luaL_checkstring( L, 1 );
73     int i_port = luaL_checkint( L, 2 );
74     const char *psz_cert = luaL_optstring( L, 3, NULL );
75     const char *psz_key = luaL_optstring( L, 4, NULL );
76     const char *psz_ca = luaL_optstring( L, 5, NULL );
77     const char *psz_crl = luaL_optstring( L, 6, NULL );
78     httpd_host_t *p_httpd_host = httpd_TLSHostNew( p_this, psz_host, i_port,
79                                                    psz_cert, psz_key,
80                                                    psz_ca, psz_crl );
81     if( !p_httpd_host )
82         return luaL_error( L, "Failed to create HTTP TLS host \"%s:%d\" "
83                            "(cert: \"%s\", key: \"%s\", ca: \"%s\", "
84                            "crl: \"%s\").", psz_host, i_port,
85                            psz_cert, psz_key, psz_ca, psz_crl );
86     vlclua_push_vlc_object( L, (vlc_object_t*)p_httpd_host, vlclua_httpd_host_delete );
87     return 1;
88 }
89
90 #define ARG_1_IS_HTTPD_HOST httpd_host_t *p_httpd_host = \
91     (httpd_host_t*)vlclua_checkobject( L, 1, VLC_OBJECT_HTTPD_HOST );
92 int vlclua_httpd_host_delete( lua_State *L )
93 {
94     ARG_1_IS_HTTPD_HOST
95     httpd_HostDelete( p_httpd_host );
96     return 0;
97 }
98
99 /*****************************************************************************
100  * HTTPd Handler
101  *****************************************************************************/
102 struct httpd_handler_sys_t
103 {
104     lua_State *L;
105     int ref;
106 };
107
108 static int vlclua_httpd_handler_callback(
109      httpd_handler_sys_t *p_sys, httpd_handler_t *p_handler, char *psz_url,
110      uint8_t *psz_request, int i_type, uint8_t *p_in, int i_in,
111      char *psz_remote_addr, char *psz_remote_host,
112      uint8_t **pp_data, int *pi_data )
113 {
114     VLC_UNUSED(p_handler);
115     lua_State *L = p_sys->L;
116
117     /* function data */
118     lua_pushvalue( L, 1 );
119     lua_pushvalue( L, 2 );
120     /* function data function data */
121     lua_pushstring( L, psz_url );
122     /* function data function data url */
123     lua_pushstring( L, (const char *)psz_request );
124     /* function data function data url request */
125     lua_pushinteger( L, i_type ); /* Q: what does i_type stand for? */
126     /* function data function data url request type */
127     lua_pushlstring( L, (const char *)p_in, i_in ); /* Q: what do p_in contain? */
128     /* function data function data url request type in */
129     lua_pushstring( L, psz_remote_addr );
130     /* function data function data url request type in addr */
131     lua_pushstring( L, psz_remote_host );
132     /* function data function data url request type in addr host */
133     if( lua_pcall( L, 7, 1, 0 ) )
134     {
135         /* function data err */
136         vlc_object_t *p_this = vlclua_get_this( L );
137         const char *psz_err = lua_tostring( L, -1 );
138         msg_Err( p_this, "Error while runing the lua HTTPd handler "
139                  "callback: %s", psz_err );
140         lua_settop( L, 2 );
141         /* function data */
142         return VLC_EGENERIC;
143     }
144     /* function data outdata */
145     *pp_data = vlclua_todata( L, -1, pi_data );
146     lua_pop( L, 1 );
147     /* function data */
148     return VLC_SUCCESS;
149 }
150
151 int vlclua_httpd_handler_new( lua_State * L )
152 {
153     ARG_1_IS_HTTPD_HOST
154     const char *psz_url = luaL_checkstring( L, 2 );
155     const char *psz_user = luaL_nilorcheckstring( L, 3 );
156     const char *psz_password = luaL_nilorcheckstring( L, 4 );
157     const vlc_acl_t *p_acl = NULL; /* FIXME 5 */
158     /* Stack item 6 is the callback function */
159     luaL_argcheck( L, lua_isfunction( L, 6 ), 6, "Should be a function" );
160     /* Stack item 7 is the callback data */
161     lua_settop( L, 7 );
162     httpd_handler_sys_t *p_sys = (httpd_handler_sys_t*)
163                                  malloc( sizeof( httpd_handler_sys_t ) );
164     if( !p_sys )
165         return luaL_error( L, "Failed to allocate private buffer." );
166     p_sys->L = lua_newthread( L );
167     p_sys->ref = luaL_ref( L, LUA_REGISTRYINDEX ); /* pops the object too */
168     /* use lua_xmove to move the lua callback function and data to
169      * the callback's stack. */
170     lua_xmove( L, p_sys->L, 2 );
171     httpd_handler_t *p_handler = httpd_HandlerNew(
172                             p_httpd_host, psz_url, psz_user, psz_password,
173                             p_acl, vlclua_httpd_handler_callback, p_sys );
174     if( !p_handler )
175         return luaL_error( L, "Failed to create HTTPd handler." );
176     lua_pushlightuserdata( L, p_handler ); /* FIXME */
177     return 1;
178 }
179
180 int vlclua_httpd_handler_delete( lua_State *L )
181 {
182     httpd_handler_t *p_handler = (httpd_handler_t*)luaL_checklightuserdata( L, 1 ); /* FIXME */
183     httpd_handler_sys_t *p_sys = httpd_HandlerDelete( p_handler );
184     luaL_unref( p_sys->L, LUA_REGISTRYINDEX, p_sys->ref );
185     free( p_sys );
186     return 0;
187 }
188
189 /*****************************************************************************
190  * HTTPd File
191  *****************************************************************************/
192 struct httpd_file_sys_t
193 {
194     lua_State *L;
195     int ref;
196 };
197
198 static int vlclua_httpd_file_callback(
199     httpd_file_sys_t *p_sys, httpd_file_t *p_file, uint8_t *psz_request,
200     uint8_t **pp_data, int *pi_data )
201 {
202     VLC_UNUSED(p_file);
203     lua_State *L = p_sys->L;
204
205     /* function data */
206     lua_pushvalue( L, 1 );
207     lua_pushvalue( L, 2 );
208     /* function data function data */
209     lua_pushstring( L, (const char *)psz_request );
210     /* function data function data request */
211     if( lua_pcall( L, 2, 1, 0 ) )
212     {
213         /* function data err */
214         vlc_object_t *p_this = vlclua_get_this( L );
215         const char *psz_err = lua_tostring( L, -1 );
216         msg_Err( p_this, "Error while runing the lua HTTPd file callback: %s",
217                  psz_err );
218         lua_settop( L, 2 );
219         /* function data */
220         return VLC_EGENERIC;
221     }
222     /* function data outdata */
223     *pp_data = vlclua_todata( L, -1, pi_data );
224     lua_pop( L, 1 );
225     /* function data */
226     return VLC_SUCCESS;
227 }
228
229 int vlclua_httpd_file_new( lua_State *L )
230 {
231     ARG_1_IS_HTTPD_HOST
232     const char *psz_url = luaL_checkstring( L, 2 );
233     const char *psz_mime = luaL_nilorcheckstring( L, 3 );
234     const char *psz_user = luaL_nilorcheckstring( L, 4 );
235     const char *psz_password = luaL_nilorcheckstring( L, 5 );
236     const vlc_acl_t *p_acl = lua_isnil( L, 6 ) ? NULL : luaL_checklightuserdata( L, 6 );
237     /* Stack item 7 is the callback function */
238     luaL_argcheck( L, lua_isfunction( L, 7 ), 7, "Should be a function" );
239     /* Stack item 8 is the callback data */
240     httpd_file_sys_t *p_sys = (httpd_file_sys_t *)
241                               malloc( sizeof( httpd_file_sys_t ) );
242     if( !p_sys )
243         return luaL_error( L, "Failed to allocate private buffer." );
244     p_sys->L = lua_newthread( L );
245     p_sys->ref = luaL_ref( L, LUA_REGISTRYINDEX ); /* pops the object too */
246     lua_xmove( L, p_sys->L, 2 );
247     httpd_file_t *p_file = httpd_FileNew( p_httpd_host, psz_url, psz_mime,
248                                           psz_user, psz_password, p_acl,
249                                           vlclua_httpd_file_callback, p_sys );
250     if( !p_file )
251         return luaL_error( L, "Failed to create HTTPd file." );
252     lua_pushlightuserdata( L, p_file ); /* FIXME */
253     return 1;
254 }
255
256 int vlclua_httpd_file_delete( lua_State *L )
257 {
258     httpd_file_t *p_file = (httpd_file_t*)luaL_checklightuserdata( L, 1 ); /* FIXME */
259     /* FIXME: How do we delete p_sys ? the struct is hidden in the VLC core */
260     httpd_file_sys_t *p_sys = httpd_FileDelete( p_file );
261     luaL_unref( p_sys->L, LUA_REGISTRYINDEX, p_sys->ref );
262     free( p_sys );
263     return 0;
264 }
265
266 /*****************************************************************************
267  * HTTPd Redirect
268  *****************************************************************************/
269 int vlclua_httpd_redirect_new( lua_State *L )
270 {
271     ARG_1_IS_HTTPD_HOST
272     const char *psz_url_dst = luaL_checkstring( L, 2 );
273     const char *psz_url_src = luaL_checkstring( L, 3 );
274     httpd_redirect_t *p_redirect = httpd_RedirectNew( p_httpd_host,
275                                                       psz_url_dst,
276                                                       psz_url_src );
277     if( !p_redirect )
278         return luaL_error( L, "Failed to create HTTPd redirect." );
279     lua_pushlightuserdata( L, p_redirect ); /* FIXME */
280     return 1;
281 }
282
283 int vlclua_httpd_redirect_delete( lua_State *L )
284 {
285     httpd_redirect_t *p_redirect = (httpd_redirect_t*)luaL_checklightuserdata( L, 1 ); /* FIXME */
286     httpd_RedirectDelete( p_redirect );
287     return 0;
288 }
289
290 /*****************************************************************************
291  * Utils
292  *****************************************************************************/
293 static uint8_t *vlclua_todata( lua_State *L, int narg, int *pi_data )
294 {
295     size_t i_data;
296     const char *psz_data = lua_tolstring( L, narg, &i_data );
297     uint8_t *p_data = (uint8_t*)malloc( i_data * sizeof(uint8_t) );
298     *pi_data = (int)i_data;
299     if( !p_data )
300     {
301         luaL_error( L, "Error while allocating buffer." );
302         return NULL; /* To please gcc even though luaL_error longjmp-ed out of here */
303     }
304     memcpy( p_data, psz_data, i_data );
305     return p_data;
306 }