?  search:
summary | log | graphiclog1 | graphiclog2 | commit | commitdiff | tree | refs | edit | fork
blame | history | raw | HEAD
freetype: font fallback for Windows
[vlc.git] / modules / lua / vlc.c
blob4340038bd048b8d0012fe19448b0165037b5b0c2
1 /*****************************************************************************
2 * vlc.c: Generic lua interface functions
3 *****************************************************************************
4 * Copyright (C) 2007-2008 the VideoLAN team
5 * $Id$
6 *
7 * Authors: Antoine Cellerier <dionoea at videolan tod org>
8 * Pierre d'Herbemont <pdherbemont # videolan.org>
9 *
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.
14 *
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.
19 *
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 *****************************************************************************/
24
25 /*****************************************************************************
26 * Preamble
27 *****************************************************************************/
28 #ifdef HAVE_CONFIG_H
29 # include "config.h"
30 #endif
31
32 #include <assert.h>
33 #include <sys/stat.h>
34
35 #define VLC_MODULE_LICENSE VLC_LICENSE_GPL_2_PLUS
36
37 #include "vlc.h"
38
39 #include <vlc_plugin.h>
40 #include <vlc_charset.h>
41 #include <vlc_fs.h>
42 #include <vlc_services_discovery.h>
43 #include <vlc_stream.h>
44
45 /*****************************************************************************
46 * Module descriptor
47 *****************************************************************************/
48 #define INTF_TEXT N_("Lua interface")
49 #define INTF_LONGTEXT N_("Lua interface module to load")
50
51 #define CONFIG_TEXT N_("Lua interface configuration")
52 #define CONFIG_LONGTEXT N_("Lua interface configuration string. Format is: '[\"<interface module name>\"] = { <option> = <value>, ...}, ...'.")
53 #define PASS_TEXT N_( "Password" )
54 #define PASS_LONGTEXT N_( "A single password restricts access " \
55 "to this interface." )
56 #define SRC_TEXT N_( "Source directory" )
57 #define SRC_LONGTEXT N_( "Source directory" )
58 #define INDEX_TEXT N_( "Directory index" )
59 #define INDEX_LONGTEXT N_( "Allow to build directory index" )
60
61 #define TELNETHOST_TEXT N_( "Host" )
62 #define TELNETHOST_LONGTEXT N_( "This is the host on which the " \
63 "interface will listen. It defaults to all network interfaces (0.0.0.0)." \
64 " If you want this interface to be available only on the local " \
65 "machine, enter \"127.0.0.1\"." )
66 #define TELNETPORT_TEXT N_( "Port" )
67 #define TELNETPORT_LONGTEXT N_( "This is the TCP port on which this " \
68 "interface will listen. It defaults to 4212." )
69 #define TELNETPWD_TEXT N_( "Password" )
70 #define TELNETPWD_LONGTEXT N_( "A single password restricts access " \
71 "to this interface." )
72 #define RCHOST_TEXT N_("TCP command input")
73 #define RCHOST_LONGTEXT N_("Accept commands over a socket rather than stdin. " \
74 "You can set the address and port the interface will bind to." )
75 #define CLIHOST_TEXT N_("CLI input")
76 #define CLIHOST_LONGTEXT N_( "Accept commands from this source. " \
77 "The CLI defaults to stdin (\"*console\"), but can also bind to a " \
78 "plain TCP socket (\"localhost:4212\") or use the telnet protocol " \
79 "(\"telnet://0.0.0.0:4212\")" )
80
81 static int vlc_sd_probe_Open( vlc_object_t * );
82
83 vlc_module_begin ()
84 set_shortname( N_("Lua") )
85 set_description( N_("Lua interpreter") )
86 set_category( CAT_INTERFACE )
87 set_subcategory( SUBCAT_INTERFACE_MAIN )
88
89 add_string( "lua-intf", "dummy", INTF_TEXT, INTF_LONGTEXT, false )
90 add_string( "lua-config", "", CONFIG_TEXT, CONFIG_LONGTEXT, false )
91 set_capability( "interface", 0 )
92 set_callbacks( Open_LuaIntf, Close_LuaIntf )
93 add_shortcut( "luaintf" )
94
95 add_submodule ()
96 set_section( N_("Lua HTTP"), 0 )
97 add_password ( "http-password", NULL, PASS_TEXT, PASS_LONGTEXT, false )
98 add_string ( "http-src", NULL, SRC_TEXT, SRC_LONGTEXT, true )
99 add_bool ( "http-index", false, INDEX_TEXT, INDEX_LONGTEXT, true )
100 set_capability( "interface", 0 )
101 set_callbacks( Open_LuaHTTP, Close_LuaIntf )
102 add_shortcut( "luahttp", "http" )
103 set_description( N_("Lua HTTP") )
104
105 add_submodule ()
106 set_section( N_("Lua CLI"), 0 )
107 add_string( "rc-host", NULL, RCHOST_TEXT, RCHOST_LONGTEXT, true )
108 add_string( "cli-host", NULL, CLIHOST_TEXT, CLIHOST_LONGTEXT, true )
109 set_capability( "interface", 25 )
110 set_description( N_("Command-line interface") )
111 set_callbacks( Open_LuaCLI, Close_LuaIntf )
112 #ifndef _WIN32
113 add_shortcut( "luacli", "luarc", "cli", "rc" )
114 #else
115 add_shortcut( "luacli", "luarc" )
116 #endif
117
118 add_submodule ()
119 set_section( N_("Lua Telnet"), 0 )
120 add_string( "telnet-host", "localhost", TELNETHOST_TEXT,
121 TELNETHOST_LONGTEXT, true )
122 add_integer( "telnet-port", TELNETPORT_DEFAULT, TELNETPORT_TEXT,
123 TELNETPORT_LONGTEXT, true )
124 change_integer_range( 1, 65535 )
125 add_password( "telnet-password", NULL, TELNETPWD_TEXT,
126
127 TELNETPWD_LONGTEXT, true )
128 set_capability( "interface", 0 )
129 set_callbacks( Open_LuaTelnet, Close_LuaIntf )
130 set_description( N_("Lua Telnet") )
131 add_shortcut( "luatelnet", "telnet" )
132
133 add_submodule ()
134 set_shortname( N_( "Lua Meta Fetcher" ) )
135 set_description( N_("Fetch meta data using lua scripts") )
136 set_capability( "meta fetcher", 10 )
137 set_callbacks( FetchMeta, NULL )
138
139 add_submodule ()
140 set_shortname( N_( "Lua Meta Reader" ) )
141 set_description( N_("Read meta data using lua scripts") )
142 set_capability( "meta reader", 10 )
143 set_callbacks( ReadMeta, NULL )
144
145 add_submodule ()
146 add_shortcut( "luaplaylist" )
147 set_shortname( N_("Lua Playlist") )
148 set_description( N_("Lua Playlist Parser Interface") )
149 set_capability( "demux", 2 )
150 set_callbacks( Import_LuaPlaylist, Close_LuaPlaylist )
151
152 add_submodule ()
153 set_shortname( N_( "Lua Art" ) )
154 set_description( N_("Fetch artwork using lua scripts") )
155 set_capability( "art finder", 10 )
156 set_callbacks( FindArt, NULL )
157
158 add_submodule ()
159 set_shortname( N_("Lua Extension") )
160 set_description( N_("Lua Extension") )
161 add_shortcut( "luaextension" )
162 set_capability( "extension", 1 )
163 set_callbacks( Open_Extension, Close_Extension )
164
165 add_submodule ()
166 set_description( N_("Lua SD Module") )
167 add_shortcut( "luasd" )
168 set_capability( "services_discovery", 0 )
169 add_string( "lua-sd", "", NULL, NULL, false )
170 change_volatile()
171 add_string( "lua-longname", "", NULL, NULL, false )
172 change_volatile()
173 set_callbacks( Open_LuaSD, Close_LuaSD )
174
175 VLC_SD_PROBE_SUBMODULE
176
177 vlc_module_end ()
178
179 /*****************************************************************************
180 *
181 *****************************************************************************/
182 static const char *ppsz_lua_exts[] = { ".luac", ".lua", ".vle", NULL };
183 static int file_select( const char *file )
184 {
185 int i = strlen( file );
186 int j;
187 for( j = 0; ppsz_lua_exts[j]; j++ )
188 {
189 int l = strlen( ppsz_lua_exts[j] );
190 if( i >= l && !strcmp( file+i-l, ppsz_lua_exts[j] ) )
191 return 1;
192 }
193 return 0;
194 }
195
196 static int file_compare( const char **a, const char **b )
197 {
198 return strcmp( *a, *b );
199 }
200
201 int vlclua_dir_list( const char *luadirname, char ***pppsz_dir_list )
202 {
203 #define MAX_DIR_LIST_SIZE 5
204 *pppsz_dir_list = malloc(MAX_DIR_LIST_SIZE*sizeof(char *));
205 if (!*pppsz_dir_list)
206 return VLC_EGENERIC;
207 char **ppsz_dir_list = *pppsz_dir_list;
208
209 int i = 0;
210 char *datadir = config_GetUserDir( VLC_DATA_DIR );
211
212 if( likely(datadir != NULL)
213 && likely(asprintf( &ppsz_dir_list[i], "%s"DIR_SEP"lua"DIR_SEP"%s",
214 datadir, luadirname ) != -1) )
215 i++;
216 free( datadir );
217
218 #if !(defined(__APPLE__) || defined(_WIN32))
219 char *psz_libpath = config_GetLibDir();
220 if( likely(psz_libpath != NULL) )
221 {
222 if( likely(asprintf( &ppsz_dir_list[i], "%s"DIR_SEP"lua"DIR_SEP"%s",
223 psz_libpath, luadirname ) != -1) )
224 i++;
225 free( psz_libpath );
226 }
227 #endif
228
229 char *psz_datapath = config_GetDataDir();
230 if( likely(psz_datapath != NULL) )
231 {
232 if( likely(asprintf( &ppsz_dir_list[i], "%s"DIR_SEP"lua"DIR_SEP"%s",
233 psz_datapath, luadirname ) != -1) )
234 i++;
235
236 #if defined(__APPLE__)
237 if( likely(asprintf( &ppsz_dir_list[i],
238 "%s"DIR_SEP"share"DIR_SEP"lua"DIR_SEP"%s",
239 psz_datapath, luadirname ) != -1) )
240 i++;
241 #endif
242 free( psz_datapath );
243 }
244
245 ppsz_dir_list[i] = NULL;
246
247 assert( i < MAX_DIR_LIST_SIZE);
248
249 return VLC_SUCCESS;
250 }
251
252 void vlclua_dir_list_free( char **ppsz_dir_list )
253 {
254 for( char **ppsz_dir = ppsz_dir_list; *ppsz_dir; ppsz_dir++ )
255 free( *ppsz_dir );
256 free( ppsz_dir_list );
257 }
258
259 /*****************************************************************************
260 * Will execute func on all scripts in luadirname, and stop if func returns
261 * success.
262 *****************************************************************************/
263 int vlclua_scripts_batch_execute( vlc_object_t *p_this,
264 const char * luadirname,
265 int (*func)(vlc_object_t *, const char *, const luabatch_context_t *),
266 void * user_data)
267 {
268 char **ppsz_dir_list = NULL;
269 int i_ret;
270
271 if( (i_ret = vlclua_dir_list( luadirname, &ppsz_dir_list )) != VLC_SUCCESS)
272 return i_ret;
273
274 i_ret = VLC_EGENERIC;
275 for( char **ppsz_dir = ppsz_dir_list; *ppsz_dir; ppsz_dir++ )
276 {
277 char **ppsz_filelist;
278
279 msg_Dbg( p_this, "Trying Lua scripts in %s", *ppsz_dir );
280 int i_files = vlc_scandir( *ppsz_dir, &ppsz_filelist, file_select,
281 file_compare );
282 if( i_files < 0 )
283 continue;
284
285 char **ppsz_file = ppsz_filelist;
286 char **ppsz_fileend = ppsz_filelist + i_files;
287
288 while( ppsz_file < ppsz_fileend )
289 {
290 char *psz_filename;
291
292 if( asprintf( &psz_filename,
293 "%s" DIR_SEP "%s", *ppsz_dir, *ppsz_file ) == -1 )
294 psz_filename = NULL;
295 free( *(ppsz_file++) );
296
297 if( likely(psz_filename != NULL) )
298 {
299 msg_Dbg( p_this, "Trying Lua playlist script %s", psz_filename );
300 i_ret = func( p_this, psz_filename, user_data );
301 free( psz_filename );
302 if( i_ret == VLC_SUCCESS )
303 break;
304 }
305 }
306
307 while( ppsz_file < ppsz_fileend )
308 free( *(ppsz_file++) );
309 free( ppsz_filelist );
310
311 if( i_ret == VLC_SUCCESS )
312 break;
313 }
314 vlclua_dir_list_free( ppsz_dir_list );
315 return i_ret;
316 }
317
318 char *vlclua_find_file( const char *psz_luadirname, const char *psz_name )
319 {
320 char **ppsz_dir_list = NULL;
321 vlclua_dir_list( psz_luadirname, &ppsz_dir_list );
322
323 for( char **ppsz_dir = ppsz_dir_list; *ppsz_dir; ppsz_dir++ )
324 {
325 for( const char **ppsz_ext = ppsz_lua_exts; *ppsz_ext; ppsz_ext++ )
326 {
327 char *psz_filename;
328 struct stat st;
329
330 if( asprintf( &psz_filename, "%s"DIR_SEP"%s%s", *ppsz_dir,
331 psz_name, *ppsz_ext ) < 0 )
332 {
333 vlclua_dir_list_free( ppsz_dir_list );
334 return NULL;
335 }
336
337 if( vlc_stat( psz_filename, &st ) == 0
338 && S_ISREG( st.st_mode ) )
339 {
340 vlclua_dir_list_free( ppsz_dir_list );
341 return psz_filename;
342 }
343 free( psz_filename );
344 }
345 }
346 vlclua_dir_list_free( ppsz_dir_list );
347 return NULL;
348 }
349
350 /*****************************************************************************
351 * Meta data setters utility.
352 * Playlist item table should be on top of the stack when these are called
353 *****************************************************************************/
354 #undef vlclua_read_meta_data
355 void vlclua_read_meta_data( vlc_object_t *p_this, lua_State *L,
356 input_item_t *p_input )
357 {
358 #define TRY_META( a, b ) \
359 lua_getfield( L, -1, a ); \
360 if( lua_isstring( L, -1 ) && \
361 strcmp( lua_tostring( L, -1 ), "" ) ) \
362 { \
363 char *psz_value = strdup( lua_tostring( L, -1 ) ); \
364 EnsureUTF8( psz_value ); \
365 msg_Dbg( p_this, #b ": %s", psz_value ); \
366 input_item_Set ## b ( p_input, psz_value ); \
367 free( psz_value ); \
368 } \
369 lua_pop( L, 1 ); /* pop a */
370 TRY_META( "title", Title );
371 TRY_META( "artist", Artist );
372 TRY_META( "genre", Genre );
373 TRY_META( "copyright", Copyright );
374 TRY_META( "album", Album );
375 TRY_META( "tracknum", TrackNum );
376 TRY_META( "description", Description );
377 TRY_META( "rating", Rating );
378 TRY_META( "date", Date );
379 TRY_META( "setting", Setting );
380 TRY_META( "url", URL );
381 TRY_META( "language", Language );
382 TRY_META( "nowplaying", NowPlaying );
383 TRY_META( "publisher", Publisher );
384 TRY_META( "encodedby", EncodedBy );
385 TRY_META( "arturl", ArtURL );
386 TRY_META( "trackid", TrackID );
387 TRY_META( "director", Director );
388 TRY_META( "season", Season );
389 TRY_META( "episode", Episode );
390 TRY_META( "show_name", ShowName );
391 TRY_META( "actors", Actors );
392 }
393
394 #undef vlclua_read_custom_meta_data
395 void vlclua_read_custom_meta_data( vlc_object_t *p_this, lua_State *L,
396 input_item_t *p_input )
397 {
398 /* Lock the input item and create the meta table if needed */
399 vlc_mutex_lock( &p_input->lock );
400
401 if( !p_input->p_meta )
402 p_input->p_meta = vlc_meta_New();
403
404 /* ... item */
405 lua_getfield( L, -1, "meta" );
406 /* ... item meta */
407 if( lua_istable( L, -1 ) )
408 {
409 lua_pushnil( L );
410 /* ... item meta nil */
411 while( lua_next( L, -2 ) )
412 {
413 /* ... item meta key value */
414 if( !lua_isstring( L, -2 ) || !lua_isstring( L, -1 ) )
415 {
416 msg_Err( p_this, "'meta' keys and values must be strings");
417 lua_pop( L, 1 ); /* pop "value" */
418 continue;
419 }
420 const char *psz_key = lua_tostring( L, -2 );
421 const char *psz_value = lua_tostring( L, -1 );
422
423 vlc_meta_AddExtra( p_input->p_meta, psz_key, psz_value );
424
425 lua_pop( L, 1 ); /* pop "value" */
426 }
427 }
428 lua_pop( L, 1 ); /* pop "meta" */
429 /* ... item -> back to original stack */
430
431 vlc_mutex_unlock( &p_input->lock );
432 }
433
434 /*****************************************************************************
435 * Playlist utilities
436 ****************************************************************************/
437 /**
438 * Playlist item table should be on top of the stack when this is called
439 */
440 #undef vlclua_read_options
441 void vlclua_read_options( vlc_object_t *p_this, lua_State *L,
442 int *pi_options, char ***pppsz_options )
443 {
444 lua_getfield( L, -1, "options" );
445 if( lua_istable( L, -1 ) )
446 {
447 lua_pushnil( L );
448 while( lua_next( L, -2 ) )
449 {
450 if( lua_isstring( L, -1 ) )
451 {
452 char *psz_option = strdup( lua_tostring( L, -1 ) );
453 msg_Dbg( p_this, "Option: %s", psz_option );
454 INSERT_ELEM( *pppsz_options, *pi_options, *pi_options,
455 psz_option );
456 }
457 else
458 {
459 msg_Warn( p_this, "Option should be a string" );
460 }
461 lua_pop( L, 1 ); /* pop option */
462 }
463 }
464 lua_pop( L, 1 ); /* pop "options" */
465 }
466
467 #undef vlclua_playlist_add_internal
468 int vlclua_playlist_add_internal( vlc_object_t *p_this, lua_State *L,
469 playlist_t *p_playlist,
470 input_item_t *p_parent, bool b_play )
471 {
472 int i_count = 0;
473 input_item_node_t *p_parent_node = NULL;
474
475 assert( p_parent || p_playlist );
476
477 /* playlist */
478 if( lua_istable( L, -1 ) )
479 {
480 if( p_parent ) p_parent_node = input_item_node_Create( p_parent );
481 lua_pushnil( L );
482 /* playlist nil */
483 while( lua_next( L, -2 ) )
484 {
485 /* playlist key item */
486 /* <Parse playlist item> */
487 if( lua_istable( L, -1 ) )
488 {
489 lua_getfield( L, -1, "path" );
490 /* playlist key item path */
491 if( lua_isstring( L, -1 ) )
492 {
493 char *psz_oldurl = NULL;
494 const char *psz_path = NULL;
495 char *psz_u8path = NULL;
496 const char *psz_name = NULL;
497 char **ppsz_options = NULL;
498 int i_options = 0;
499 mtime_t i_duration = -1;
500 input_item_t *p_input;
501
502 /* Read path and name */
503 if (p_parent) {
504 psz_oldurl = input_item_GetURI( p_parent );
505 msg_Dbg( p_this, "old path: %s", psz_oldurl );
506 }
507 psz_path = lua_tostring( L, -1 );
508 msg_Dbg( p_this, "Path: %s", psz_path );
509 lua_getfield( L, -2, "name" );
510 /* playlist key item path name */
511 if( lua_isstring( L, -1 ) )
512 {
513 psz_name = lua_tostring( L, -1 );
514 msg_Dbg( p_this, "Name: %s", psz_name );
515 }
516 else
517 {
518 if( !lua_isnil( L, -1 ) )
519 msg_Warn( p_this, "Playlist item name should be a string." );
520 psz_name = NULL;
521 }
522
523 /* Read duration */
524 lua_getfield( L, -3, "duration" );
525 /* playlist key item path name duration */
526 if( lua_isnumber( L, -1 ) )
527 {
528 i_duration = (mtime_t)(lua_tonumber( L, -1 )*1e6);
529 }
530 else if( !lua_isnil( L, -1 ) )
531 {
532 msg_Warn( p_this, "Playlist item duration should be a number (in seconds)." );
533 }
534 lua_pop( L, 1 ); /* pop "duration" */
535
536 /* playlist key item path name */
537
538 /* Read options: item must be on top of stack */
539 lua_pushvalue( L, -3 );
540 /* playlist key item path name item */
541 vlclua_read_options( p_this, L, &i_options, &ppsz_options );
542
543 /* Create input item */
544 p_input = input_item_NewExt( psz_path, psz_name, i_options,
545 (const char **)ppsz_options,
546 VLC_INPUT_OPTION_TRUSTED,
547 i_duration );
548 lua_pop( L, 3 ); /* pop "path name item" */
549 /* playlist key item */
550
551 /* Read meta data: item must be on top of stack */
552 vlclua_read_meta_data( p_this, L, p_input );
553
554 /* copy the original URL to the meta data, if "URL" is still empty */
555 char* url = input_item_GetURL( p_input );
556 if( url == NULL && p_parent)
557 {
558 EnsureUTF8( psz_oldurl );
559 msg_Dbg( p_this, "meta-URL: %s", psz_oldurl );
560 input_item_SetURL ( p_input, psz_oldurl );
561 }
562 free( psz_oldurl );
563 free( url );
564
565 /* copy the psz_name to the meta data, if "Title" is still empty */
566 char* title = input_item_GetTitle( p_input );
567 if( title == NULL )
568 input_item_SetTitle ( p_input, psz_name );
569 free( title );
570
571 /* Read custom meta data: item must be on top of stack*/
572 vlclua_read_custom_meta_data( p_this, L, p_input );
573
574 /* Append item to playlist */
575 if( p_parent ) /* Add to node */
576 {
577 input_item_CopyOptions( p_parent, p_input );
578 input_item_node_AppendItem( p_parent_node, p_input );
579 }
580 else /* Play or Enqueue (preparse) */
581 /* FIXME: playlist_AddInput() can fail */
582 playlist_AddInput( p_playlist, p_input,
583 PLAYLIST_APPEND |
584 ( b_play ? PLAYLIST_GO : PLAYLIST_PREPARSE ),
585 PLAYLIST_END, true, false );
586 i_count ++; /* increment counter */
587 vlc_gc_decref( p_input );
588 while( i_options > 0 )
589 free( ppsz_options[--i_options] );
590 free( ppsz_options );
591 free( psz_u8path );
592 }
593 else
594 {
595 lua_pop( L, 1 ); /* pop "path" */
596 msg_Warn( p_this,
597 "Playlist item's path should be a string" );
598 }
599 /* playlist key item */
600 }
601 else
602 {
603 msg_Warn( p_this, "Playlist item should be a table" );
604 }
605 /* <Parse playlist item> */
606 lua_pop( L, 1 ); /* pop the value, keep the key for
607 * the next lua_next() call */
608 /* playlist key */
609 }
610 /* playlist */
611 if( p_parent )
612 {
613 if( i_count ) input_item_node_PostAndDelete( p_parent_node );
614 else input_item_node_Delete( p_parent_node );
615 }
616 }
617 else
618 {
619 msg_Warn( p_this, "Playlist should be a table." );
620 }
621 return i_count;
622 }
623
624 static int vlc_sd_probe_Open( vlc_object_t *obj )
625 {
626 vlc_probe_t *probe = (vlc_probe_t *)obj;
627 char **ppsz_filelist = NULL;
628 char **ppsz_fileend = NULL;
629 char **ppsz_file;
630 char *psz_name;
631 char **ppsz_dir_list = NULL;
632 char **ppsz_dir;
633 lua_State *L = NULL;
634 vlclua_dir_list( "sd", &ppsz_dir_list );
635 for( ppsz_dir = ppsz_dir_list; *ppsz_dir; ppsz_dir++ )
636 {
637 int i_files;
638 if( ppsz_filelist )
639 {
640 for( ppsz_file = ppsz_filelist; ppsz_file < ppsz_fileend;
641 ppsz_file++ )
642 free( *ppsz_file );
643 free( ppsz_filelist );
644 ppsz_filelist = NULL;
645 }
646 i_files = vlc_scandir( *ppsz_dir, &ppsz_filelist, file_select,
647 file_compare );
648 if( i_files < 1 ) continue;
649 ppsz_fileend = ppsz_filelist + i_files;
650 for( ppsz_file = ppsz_filelist; ppsz_file < ppsz_fileend; ppsz_file++ )
651 {
652 char *psz_filename;
653 if( asprintf( &psz_filename,
654 "%s" DIR_SEP "%s", *ppsz_dir, *ppsz_file ) < 0 )
655 {
656 goto error;
657 }
658 L = luaL_newstate();
659 if( !L )
660 {
661 msg_Err( probe, "Could not create new Lua State" );
662 free( psz_filename );
663 goto error;
664 }
665 luaL_openlibs( L );
666 if( vlclua_add_modules_path( L, psz_filename ) )
667 {
668 msg_Err( probe, "Error while setting the module search path for %s",
669 psz_filename );
670 free( psz_filename );
671 goto error;
672 }
673 if( vlclua_dofile( VLC_OBJECT(probe), L, psz_filename ) )
674 {
675
676 msg_Err( probe, "Error loading script %s: %s", psz_filename,
677 lua_tostring( L, lua_gettop( L ) ) );
678 lua_pop( L, 1 );
679 free( psz_filename );
680 lua_close( L );
681 continue;
682 }
683 char *psz_longname;
684 char *temp = strchr( *ppsz_file, '.' );
685 if( temp )
686 *temp = '\0';
687 lua_getglobal( L, "descriptor" );
688 if( !lua_isfunction( L, lua_gettop( L ) ) || lua_pcall( L, 0, 1, 0 ) )
689 {
690 msg_Warn( probe, "No 'descriptor' function in '%s'", psz_filename );
691 lua_pop( L, 1 );
692 if( !( psz_longname = strdup( *ppsz_file ) ) )
693 {
694 free( psz_filename );
695 goto error;
696 }
697 }
698 else
699 {
700 lua_getfield( L, -1, "title" );
701 if( !lua_isstring( L, -1 ) ||
702 !( psz_longname = strdup( lua_tostring( L, -1 ) ) ) )
703 {
704 free( psz_filename );
705 goto error;
706 }
707 }
708
709 char *psz_file_esc = config_StringEscape( *ppsz_file );
710 char *psz_longname_esc = config_StringEscape( psz_longname );
711 if( asprintf( &psz_name, "lua{sd='%s',longname='%s'}",
712 psz_file_esc, psz_longname_esc ) < 0 )
713 {
714 free( psz_file_esc );
715 free( psz_longname_esc );
716 free( psz_filename );
717 free( psz_longname );
718 goto error;
719 }
720 free( psz_file_esc );
721 free( psz_longname_esc );
722 vlc_sd_probe_Add( probe, psz_name, psz_longname, SD_CAT_INTERNET );
723 free( psz_name );
724 free( psz_longname );
725 free( psz_filename );
726 lua_close( L );
727 }
728 }
729 if( ppsz_filelist )
730 {
731 for( ppsz_file = ppsz_filelist; ppsz_file < ppsz_fileend;
732 ppsz_file++ )
733 free( *ppsz_file );
734 free( ppsz_filelist );
735 }
736 vlclua_dir_list_free( ppsz_dir_list );
737 return VLC_PROBE_CONTINUE;
738 error:
739 if( ppsz_filelist )
740 {
741 for( ppsz_file = ppsz_filelist; ppsz_file < ppsz_fileend;
742 ppsz_file++ )
743 free( *ppsz_file );
744 free( ppsz_filelist );
745 }
746 if( L )
747 lua_close( L );
748 vlclua_dir_list_free( ppsz_dir_list );
749 return VLC_ENOMEM;
750 }
751
752 static int vlclua_add_modules_path_inner( lua_State *L, const char *psz_path )
753 {
754 int count = 0;
755 for( const char **ppsz_ext = ppsz_lua_exts; *ppsz_ext; ppsz_ext++ )
756 {
757 lua_pushfstring( L, "%s"DIR_SEP"modules"DIR_SEP"?%s;",
758 psz_path, *ppsz_ext );
759 count ++;
760 }
761
762 return count;
763 }
764
765 int vlclua_add_modules_path( lua_State *L, const char *psz_filename )
766 {
767 /* Setup the module search path:
768 * * "The script's directory"/modules
769 * * "The script's parent directory"/modules
770 * * and so on for all the next directories in the directory list
771 */
772 char *psz_path = strdup( psz_filename );
773 if( !psz_path )
774 return 1;
775
776 char *psz_char = strrchr( psz_path, DIR_SEP_CHAR );
777 if( !psz_char )
778 {
779 free( psz_path );
780 return 1;
781 }
782 *psz_char = '\0';
783
784 /* psz_path now holds the file's directory */
785 psz_char = strrchr( psz_path, DIR_SEP_CHAR );
786 if( !psz_char )
787 {
788 free( psz_path );
789 return 1;
790 }
791 *psz_char = '\0';
792
793 /* Push package on stack */
794 int count = 0;
795 lua_getglobal( L, "package" );
796
797 /* psz_path now holds the file's parent directory */
798 count += vlclua_add_modules_path_inner( L, psz_path );
799 *psz_char = DIR_SEP_CHAR;
800
801 /* psz_path now holds the file's directory */
802 count += vlclua_add_modules_path_inner( L, psz_path );
803
804 char **ppsz_dir_list = NULL;
805 vlclua_dir_list( psz_char+1/* gruik? */, &ppsz_dir_list );
806 char **ppsz_dir = ppsz_dir_list;
807
808 for( ; *ppsz_dir && strcmp( *ppsz_dir, psz_path ); ppsz_dir++ );
809 free( psz_path );
810
811 for( ; *ppsz_dir; ppsz_dir++ )
812 {
813 psz_path = *ppsz_dir;
814 /* FIXME: doesn't work well with meta/... modules due to the double
815 * directory depth */
816 psz_char = strrchr( psz_path, DIR_SEP_CHAR );
817 if( !psz_char )
818 {
819 vlclua_dir_list_free( ppsz_dir_list );
820 return 1;
821 }
822
823 *psz_char = '\0';
824 count += vlclua_add_modules_path_inner( L, psz_path );
825 *psz_char = DIR_SEP_CHAR;
826 count += vlclua_add_modules_path_inner( L, psz_path );
827 }
828
829 lua_getfield( L, -(count+1), "path" ); /* Get package.path */
830 lua_concat( L, count+1 ); /* Concat vlc module paths and package.path */
831 lua_setfield( L, -2, "path"); /* Set package.path */
832 lua_pop( L, 1 ); /* Pop the package module */
833
834 vlclua_dir_list_free( ppsz_dir_list );
835 return 0;
836 }
837
838 /** Replacement for luaL_dofile, using VLC's input capabilities */
839 int vlclua_dofile( vlc_object_t *p_this, lua_State *L, const char *curi )
840 {
841 char *uri = ToLocaleDup( curi );
842 if( !strstr( uri, "://" ) ) {
843 int ret = luaL_dofile( L, uri );
844 free( uri );
845 return ret;
846 }
847 if( !strncasecmp( uri, "file://", 7 ) ) {
848 int ret = luaL_dofile( L, uri + 7 );
849 free( uri );
850 return ret;
851 }
852 stream_t *s = stream_UrlNew( p_this, uri );
853 if( !s )
854 {
855 free( uri );
856 return 1;
857 }
858 int64_t i_size = stream_Size( s );
859 char *p_buffer = ( i_size > 0 ) ? malloc( i_size ) : NULL;
860 if( !p_buffer )
861 {
862 // FIXME: read the whole stream until we reach the end (if no size)
863 stream_Delete( s );
864 free( uri );
865 return 1;
866 }
867 int64_t i_read = stream_Read( s, p_buffer, (int) i_size );
868 int i_ret = ( i_read == i_size ) ? 0 : 1;
869 if( !i_ret )
870 i_ret = luaL_loadbuffer( L, p_buffer, (size_t) i_size, uri );
871 if( !i_ret )
872 i_ret = lua_pcall( L, 0, LUA_MULTRET, 0 );
873 stream_Delete( s );
874 free( p_buffer );
875 free( uri );
876 return i_ret;
877 }
VLC Development tree mirror