lib/media_discoverer.c

   1 /*****************************************************************************
   2  * media_discoverer.c: libvlc new API media discoverer functions
   3  *****************************************************************************
   4  * Copyright (C) 2007 VLC authors and VideoLAN
   5  * $Id$
   6  *
   7  * Authors: Pierre d'Herbemont <pdherbemont # videolan.org>
   8  *
   9  * This program is free software; you can redistribute it and/or modify it
  10  * under the terms of the GNU Lesser General Public License as published by
  11  * the Free Software Foundation; either version 2.1 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 Lesser General Public License for more details.
  18  *
  19  * You should have received a copy of the GNU Lesser General Public License
  20  * along with this program; if not, write to the Free Software Foundation,
  21  * Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
  22  *****************************************************************************/
  23
  24 #ifdef HAVE_CONFIG_H
  25 # include "config.h"
  26 #endif
  27
  28 #include <assert.h>
  29
  30 #include <vlc/libvlc.h>
  31 #include <vlc/libvlc_picture.h>
  32 #include <vlc/libvlc_media.h>
  33 #include <vlc/libvlc_media_list.h>
  34 #include <vlc/libvlc_media_discoverer.h>
  35 #include <vlc/libvlc_events.h>
  36
  37 #include <vlc_services_discovery.h>
  38
  39 #include "libvlc_internal.h"
  40 #include "media_internal.h" // libvlc_media_new_from_input_item()
  41 #include "media_list_internal.h" // libvlc_media_list_internal_add_media()
  42
  43 struct libvlc_media_discoverer_t
  44 {
  45     libvlc_instance_t *      p_libvlc_instance;
  46     services_discovery_t *   p_sd;
  47     libvlc_media_list_t *    p_mlist;
  48     vlc_dictionary_t         catname_to_submedialist;
  49     char                     name[];
  50 };
  51
  52 /*
  53  * Private functions
  54  */
  55
  56 /**************************************************************************
  57  *       services_discovery_item_added (Private) (VLC event callback)
  58  **************************************************************************/
  59
  60 static void services_discovery_item_added( services_discovery_t *sd,
  61                                            input_item_t *parent,
  62                                            input_item_t *p_item,
  63                                            const char *psz_cat )
  64 {
  65     libvlc_media_t * p_md;
  66     libvlc_media_discoverer_t *p_mdis = sd->owner.sys;
  67     libvlc_media_list_t * p_mlist = p_mdis->p_mlist;
  68
  69     p_md = libvlc_media_new_from_input_item( p_mdis->p_libvlc_instance,
  70                                              p_item );
  71
  72     if( parent != NULL )
  73     {
  74         /* Flatten items list for now. TODO: tree support. */
  75     }
  76     else
  77     /* If we have a category, that mean we have to group the items having
  78      * that category in a media_list. */
  79     if( psz_cat )
  80     {
  81         p_mlist = vlc_dictionary_value_for_key( &p_mdis->catname_to_submedialist, psz_cat );
  82
  83         if( p_mlist == kVLCDictionaryNotFound )
  84         {
  85             libvlc_media_t * p_catmd;
  86             p_catmd = libvlc_media_new_as_node( p_mdis->p_libvlc_instance, psz_cat );
  87             p_mlist = libvlc_media_subitems( p_catmd );
  88             p_mlist->b_read_only = true;
  89
  90             /* Insert the newly created mlist in our dictionary */
  91             vlc_dictionary_insert( &p_mdis->catname_to_submedialist, psz_cat, p_mlist );
  92
  93             /* Insert the md into the root list */
  94             libvlc_media_list_lock( p_mdis->p_mlist );
  95             libvlc_media_list_internal_add_media( p_mdis->p_mlist, p_catmd );
  96             libvlc_media_list_unlock( p_mdis->p_mlist );
  97
  98             /* We don't release the mlist cause the dictionary
  99              * doesn't retain the object. But we release the md. */
 100             libvlc_media_release( p_catmd );
 101         }
 102     }
 103
 104     libvlc_media_list_lock( p_mlist );
 105     libvlc_media_list_internal_add_media( p_mlist, p_md );
 106     libvlc_media_list_unlock( p_mlist );
 107
 108     libvlc_media_release( p_md );
 109 }
 110
 111 /**************************************************************************
 112  *       services_discovery_item_removed (Private) (VLC event callback)
 113  **************************************************************************/
 114
 115 static void services_discovery_item_removed( services_discovery_t *sd,
 116                                              input_item_t *p_item )
 117 {
 118     libvlc_media_t * p_md;
 119     libvlc_media_discoverer_t *p_mdis = sd->owner.sys;
 120
 121     int i, count = libvlc_media_list_count( p_mdis->p_mlist );
 122     libvlc_media_list_lock( p_mdis->p_mlist );
 123     for( i = 0; i < count; i++ )
 124     {
 125         p_md = libvlc_media_list_item_at_index( p_mdis->p_mlist, i );
 126         assert(p_md != NULL);
 127         if( p_md->p_input_item == p_item )
 128         {
 129             libvlc_media_list_internal_remove_index( p_mdis->p_mlist, i );
 130             libvlc_media_release( p_md );
 131             break;
 132         }
 133         libvlc_media_release( p_md );
 134     }
 135     libvlc_media_list_unlock( p_mdis->p_mlist );
 136 }
 137
 138 /*
 139  * Public libvlc functions
 140  */
 141
 142 /**************************************************************************
 143  *       new (Public)
 144  **************************************************************************/
 145 libvlc_media_discoverer_t *
 146 libvlc_media_discoverer_new( libvlc_instance_t * p_inst, const char * psz_name )
 147 {
 148     /* podcast SD is a hack and only works with custom playlist callbacks. */
 149     if( !strncasecmp( psz_name, "podcast", 7 ) )
 150         return NULL;
 151
 152     libvlc_media_discoverer_t *p_mdis;
 153
 154     p_mdis = malloc(sizeof(*p_mdis) + strlen(psz_name) + 1);
 155     if( unlikely(p_mdis == NULL) )
 156     {
 157         libvlc_printerr( "Not enough memory" );
 158         return NULL;
 159     }
 160
 161     p_mdis->p_libvlc_instance = p_inst;
 162     p_mdis->p_mlist = libvlc_media_list_new( p_inst );
 163     p_mdis->p_mlist->b_read_only = true;
 164     p_mdis->p_sd = NULL;
 165
 166     vlc_dictionary_init( &p_mdis->catname_to_submedialist, 0 );
 167
 168     libvlc_retain( p_inst );
 169     strcpy( p_mdis->name, psz_name );
 170     return p_mdis;
 171 }
 172
 173 static const struct services_discovery_callbacks sd_cbs = {
 174     .item_added = services_discovery_item_added,
 175     .item_removed = services_discovery_item_removed,
 176 };
 177
 178 /**************************************************************************
 179  *       start (Public)
 180  **************************************************************************/
 181 LIBVLC_API int
 182 libvlc_media_discoverer_start( libvlc_media_discoverer_t * p_mdis )
 183 {
 184     struct services_discovery_owner_t owner = {
 185         &sd_cbs,
 186         p_mdis,
 187     };
 188
 189     /* Here we go */
 190     p_mdis->p_sd = vlc_sd_Create( (vlc_object_t *)p_mdis->p_libvlc_instance->p_libvlc_int,
 191                                   p_mdis->name, &owner );
 192     if( p_mdis->p_sd == NULL )
 193     {
 194         libvlc_printerr( "%s: no such discovery module found", p_mdis->name );
 195         return -1;
 196     }
 197
 198     return 0;
 199 }
 200
 201 /**************************************************************************
 202  *       stop (Public)
 203  **************************************************************************/
 204 LIBVLC_API void
 205 libvlc_media_discoverer_stop( libvlc_media_discoverer_t * p_mdis )
 206 {
 207     libvlc_media_list_t * p_mlist = p_mdis->p_mlist;
 208     libvlc_media_list_lock( p_mlist );
 209     libvlc_media_list_internal_end_reached( p_mlist );
 210     libvlc_media_list_unlock( p_mlist );
 211
 212     vlc_sd_Destroy( p_mdis->p_sd );
 213     p_mdis->p_sd = NULL;
 214 }
 215
 216 /**************************************************************************
 217  * release (Public)
 218  **************************************************************************/
 219 static void
 220 MediaListDictValueRelease( void* mlist, void* obj )
 221 {
 222     libvlc_media_list_release( mlist );
 223     (void)obj;
 224 }
 225
 226 void
 227 libvlc_media_discoverer_release( libvlc_media_discoverer_t * p_mdis )
 228 {
 229     if( p_mdis->p_sd != NULL )
 230         libvlc_media_discoverer_stop( p_mdis );
 231
 232     libvlc_media_list_release( p_mdis->p_mlist );
 233
 234     vlc_dictionary_clear( &p_mdis->catname_to_submedialist,
 235         MediaListDictValueRelease, NULL );
 236
 237     libvlc_release( p_mdis->p_libvlc_instance );
 238
 239     free( p_mdis );
 240 }
 241
 242 /**************************************************************************
 243  * media_list (Public)
 244  **************************************************************************/
 245 libvlc_media_list_t *
 246 libvlc_media_discoverer_media_list( libvlc_media_discoverer_t * p_mdis )
 247 {
 248     libvlc_media_list_retain( p_mdis->p_mlist );
 249     return p_mdis->p_mlist;
 250 }
 251
 252 /**************************************************************************
 253  * running (Public)
 254  **************************************************************************/
 255 int
 256 libvlc_media_discoverer_is_running( libvlc_media_discoverer_t * p_mdis )
 257 {
 258     return p_mdis->p_sd != NULL;
 259 }
 260
 261 void
 262 libvlc_media_discoverer_list_release( libvlc_media_discoverer_description_t **pp_services,
 263                                       size_t i_count )
 264 {
 265     if( i_count > 0 )
 266     {
 267         for( size_t i = 0; i < i_count; ++i )
 268         {
 269             free( pp_services[i]->psz_name );
 270             free( pp_services[i]->psz_longname );
 271         }
 272         free( *pp_services );
 273         free( pp_services );
 274     }
 275 }
 276
 277 size_t
 278 libvlc_media_discoverer_list_get( libvlc_instance_t *p_inst,
 279                                   libvlc_media_discoverer_category_t i_cat,
 280                                   libvlc_media_discoverer_description_t ***ppp_services )
 281 {
 282     assert( p_inst != NULL && ppp_services != NULL );
 283
 284     int i_core_cat;
 285     switch( i_cat )
 286     {
 287     case libvlc_media_discoverer_devices:
 288         i_core_cat = SD_CAT_DEVICES;
 289         break;
 290     case libvlc_media_discoverer_lan:
 291         i_core_cat = SD_CAT_LAN;
 292         break;
 293     case libvlc_media_discoverer_podcasts:
 294         i_core_cat = SD_CAT_INTERNET;
 295         break;
 296     case libvlc_media_discoverer_localdirs:
 297         i_core_cat = SD_CAT_MYCOMPUTER;
 298         break;
 299     default:
 300         vlc_assert_unreachable();
 301         *ppp_services = NULL;
 302         return 0;
 303     }
 304
 305     /* Fetch all sd names, longnames and categories */
 306     char **ppsz_names, **ppsz_longnames;
 307     int *p_categories;
 308     ppsz_names = vlc_sd_GetNames( p_inst->p_libvlc_int, &ppsz_longnames,
 309                                   &p_categories );
 310
 311     if( ppsz_names == NULL )
 312     {
 313         *ppp_services = NULL;
 314         return 0;
 315     }
 316
 317     /* Count the number of sd matching our category (i_cat/i_core_cat) */
 318     size_t i_nb_services = 0;
 319     char **ppsz_name = ppsz_names;
 320     int *p_category = p_categories;
 321     for( ; *ppsz_name != NULL; ppsz_name++, p_category++ )
 322     {
 323         if( *p_category == i_core_cat )
 324             i_nb_services++;
 325     }
 326
 327     libvlc_media_discoverer_description_t **pp_services = NULL, *p_services = NULL;
 328     if( i_nb_services > 0 )
 329     {
 330         /* Double alloc here, so that the caller iterates through pointers of
 331          * struct instead of structs. This allows us to modify the struct
 332          * without breaking the API. */
 333
 334         pp_services = malloc( i_nb_services
 335                               * sizeof(libvlc_media_discoverer_description_t *) );
 336         p_services = malloc( i_nb_services
 337                              * sizeof(libvlc_media_discoverer_description_t) );
 338         if( pp_services == NULL || p_services == NULL )
 339         {
 340             free( pp_services );
 341             free( p_services );
 342             pp_services = NULL;
 343             p_services = NULL;
 344             i_nb_services = 0;
 345             /* Even if alloc fails, the next loop must be run in order to free
 346              * names returned by vlc_sd_GetNames */
 347         }
 348     }
 349
 350     /* Fill output pp_services or free unused name, longname */
 351     char **ppsz_longname = ppsz_longnames;
 352     ppsz_name = ppsz_names;
 353     p_category = p_categories;
 354     unsigned int i_service_idx = 0;
 355     libvlc_media_discoverer_description_t *p_service = p_services;
 356     for( ; *ppsz_name != NULL; ppsz_name++, ppsz_longname++, p_category++ )
 357     {
 358         if( pp_services != NULL && *p_category == i_core_cat )
 359         {
 360             p_service->psz_name = *ppsz_name;
 361             p_service->psz_longname = *ppsz_longname;
 362             p_service->i_cat = i_cat;
 363             pp_services[i_service_idx++] = p_service++;
 364         }
 365         else
 366         {
 367             free( *ppsz_name );
 368             free( *ppsz_longname );
 369         }
 370     }
 371     free( ppsz_names );
 372     free( ppsz_longnames );
 373     free( p_categories );
 374
 375     *ppp_services = pp_services;
 376     return i_nb_services;
 377 }