git.sesse.net Git - vlc/blob - modules/lua/vlc.c

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