git.sesse.net Git - vlc/blob - modules/services_discovery/sap.c

   1 /*****************************************************************************
   2  * sap.c :  SAP interface module
   3  *****************************************************************************
   4  * Copyright (C) 2004-2005 the VideoLAN team
   5  * $Id$
   6  *
   7  * Authors: Clément Stenac <zorglub@videolan.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  * Includes
  26  *****************************************************************************/
  27 #define _GNU_SOURCE
  28 #include <stdlib.h>                                      /* malloc(), free() */
  29
  30 #include <vlc/vlc.h>
  31 #include <vlc/input.h>
  32 #include <vlc/intf.h>
  33
  34 #include <network.h>
  35 #include <charset.h>
  36
  37 #include <ctype.h>
  38 #include <errno.h>
  39
  40 #ifdef HAVE_UNISTD_H
  41 #    include <unistd.h>
  42 #endif
  43 #ifdef HAVE_SYS_TIME_H
  44 #    include <sys/time.h>
  45 #endif
  46
  47 #ifdef HAVE_ZLIB_H
  48 #   include <zlib.h>
  49 #endif
  50
  51 /************************************************************************
  52  * Macros and definitions
  53  ************************************************************************/
  54
  55 #define MAX_LINE_LENGTH 256
  56
  57 /* SAP is always on that port */
  58 #define SAP_PORT 9875
  59 /* Global-scope SAP address */
  60 #define SAP_V4_GLOBAL_ADDRESS   "224.2.127.254"
  61 /* Organization-local SAP address */
  62 #define SAP_V4_ORG_ADDRESS      "239.195.255.255"
  63 /* Local (smallest non-link-local scope) SAP address */
  64 #define SAP_V4_LOCAL_ADDRESS    "239.255.255.255"
  65 /* Link-local SAP address */
  66 #define SAP_V4_LINK_ADDRESS     "224.0.0.255"
  67 #define ADD_SESSION 1
  68
  69 #define SAP_V6_1 "FF0"
  70 /* Scope is inserted between them */
  71 #define SAP_V6_2 "::2:7FFE"
  72 /* See RFC3513 for list of valid scopes */
  73 /* FIXME: find a way to listen to link-local scope */
  74 static const char ipv6_scopes[] = "1456789ABCDE";
  75
  76
  77 /*****************************************************************************
  78  * Module descriptor
  79  *****************************************************************************/
  80 #define SAP_ADDR_TEXT N_( "SAP multicast address" )
  81 #define SAP_ADDR_LONGTEXT N_( "The SAP module normally chooses itself the " \
  82                               "right addresses to listen to. However, you " \
  83                               "can specify a specific address." )
  84 #define SAP_IPV4_TEXT N_( "IPv4 SAP" )
  85 #define SAP_IPV4_LONGTEXT N_( \
  86       "Listen to IPv4 announcements " \
  87       "on the standard address." )
  88 #define SAP_IPV6_TEXT N_( "IPv6 SAP" )
  89 #define SAP_IPV6_LONGTEXT N_( \
  90       "Listen to IPv6 announcements " \
  91       "on the standard addresses." )
  92 #define SAP_SCOPE_TEXT N_( "IPv6 SAP scope" )
  93 #define SAP_SCOPE_LONGTEXT N_( \
  94        "Scope for IPv6 announcements (default is 8)." )
  95 #define SAP_TIMEOUT_TEXT N_( "SAP timeout (seconds)" )
  96 #define SAP_TIMEOUT_LONGTEXT N_( \
  97        "Delay after which SAP items get deleted if no new announcement " \
  98        "is received." )
  99 #define SAP_PARSE_TEXT N_( "Try to parse the announce" )
 100 #define SAP_PARSE_LONGTEXT N_( \
 101        "This enables actual parsing of the announces by the SAP module. " \
 102        "Otherwise, all announcements are parsed by the \"livedotcom\" " \
 103        "(RTP/RTSP) module." )
 104 #define SAP_STRICT_TEXT N_( "SAP Strict mode" )
 105 #define SAP_STRICT_LONGTEXT N_( \
 106        "When this is set, the SAP parser will discard some non-compliant " \
 107        "announcements." )
 108 #define SAP_CACHE_TEXT N_("Use SAP cache")
 109 #define SAP_CACHE_LONGTEXT N_( \
 110        "This enables a SAP caching mechanism. " \
 111        "This will result in lower SAP startup time, but you could end up " \
 112        "with items corresponding to legacy streams." )
 113 #define SAP_TIMESHIFT_TEXT N_("Allow timeshifting")
 114 #define SAP_TIMESHIFT_LONGTEXT N_( "This automatically enables timeshifting " \
 115         "for streams discovered through SAP announcements." )
 116
 117 /* Callbacks */
 118     static int  Open ( vlc_object_t * );
 119     static void Close( vlc_object_t * );
 120     static int  OpenDemux ( vlc_object_t * );
 121     static void CloseDemux ( vlc_object_t * );
 122
 123 vlc_module_begin();
 124     set_shortname( _("SAP"));
 125     set_description( _("SAP Announcements") );
 126     set_category( CAT_PLAYLIST );
 127     set_subcategory( SUBCAT_PLAYLIST_SD );
 128
 129     add_string( "sap-addr", NULL, NULL,
 130                 SAP_ADDR_TEXT, SAP_ADDR_LONGTEXT, VLC_TRUE );
 131     add_bool( "sap-ipv4", 1 , NULL,
 132                SAP_IPV4_TEXT,SAP_IPV4_LONGTEXT, VLC_TRUE );
 133     add_bool( "sap-ipv6", 1 , NULL,
 134               SAP_IPV6_TEXT, SAP_IPV6_LONGTEXT, VLC_TRUE );
 135     add_integer( "sap-timeout", 1800, NULL,
 136                  SAP_TIMEOUT_TEXT, SAP_TIMEOUT_LONGTEXT, VLC_TRUE );
 137     add_bool( "sap-parse", 1 , NULL,
 138                SAP_PARSE_TEXT,SAP_PARSE_LONGTEXT, VLC_TRUE );
 139     add_bool( "sap-strict", 0 , NULL,
 140                SAP_STRICT_TEXT,SAP_STRICT_LONGTEXT, VLC_TRUE );
 141 #if 0
 142     add_bool( "sap-cache", 0 , NULL,
 143                SAP_CACHE_TEXT,SAP_CACHE_LONGTEXT, VLC_TRUE );
 144 #endif
 145     add_bool( "sap-timeshift", 0 , NULL,
 146               SAP_TIMESHIFT_TEXT,SAP_TIMESHIFT_LONGTEXT, VLC_TRUE );
 147
 148     set_capability( "services_discovery", 0 );
 149     set_callbacks( Open, Close );
 150
 151     add_submodule();
 152         set_description( _("SDP file parser for UDP") );
 153         add_shortcut( "sdp" );
 154         set_capability( "demux2", 51 );
 155         set_callbacks( OpenDemux, CloseDemux );
 156 vlc_module_end();
 157
 158
 159 /*****************************************************************************
 160  * Local structures
 161  *****************************************************************************/
 162
 163 typedef struct sdp_t sdp_t;
 164 typedef struct attribute_t attribute_t;
 165 typedef struct sap_announce_t sap_announce_t;
 166
 167 /* The structure that contains sdp information */
 168 struct  sdp_t
 169 {
 170     char *psz_sdp;
 171
 172     /* s= field */
 173     char *psz_sessionname;
 174
 175     /* Raw m= and c= fields */
 176     char *psz_connection;
 177     char *psz_media;
 178
 179     /* o field */
 180     char *psz_username;
 181     char *psz_network_type;
 182     char *psz_address_type;
 183     char *psz_address;
 184     int64_t i_session_id;
 185
 186     /* "computed" URI */
 187     char *psz_uri;
 188
 189     int           i_in; /* IP version */
 190
 191     int           i_media;
 192     int           i_media_type;
 193
 194     int           i_attributes;
 195     attribute_t  **pp_attributes;
 196 };
 197
 198 struct attribute_t
 199 {
 200     char *psz_field;
 201     char *psz_value;
 202 };
 203
 204 struct sap_announce_t
 205 {
 206     mtime_t i_last;
 207
 208     uint16_t    i_hash;
 209     uint32_t    i_source[4];
 210
 211     /* SAP annnounces must only contain one SDP */
 212     sdp_t       *p_sdp;
 213
 214     int i_input_id;
 215     int i_item_id_cat;
 216     int i_item_id_one;
 217 };
 218
 219 struct services_discovery_sys_t
 220 {
 221     /* Socket descriptors */
 222     int i_fd;
 223     int *pi_fd;
 224
 225     /* playlist node */
 226     playlist_item_t *p_node_cat;
 227     playlist_item_t *p_node_one;
 228
 229     /* Table of announces */
 230     int i_announces;
 231     struct sap_announce_t **pp_announces;
 232
 233     /* Modes */
 234     vlc_bool_t  b_strict;
 235     vlc_bool_t  b_parse;
 236     vlc_bool_t  b_timeshift;
 237
 238     int i_timeout;
 239 };
 240
 241 struct demux_sys_t
 242 {
 243     sdp_t *p_sdp;
 244 };
 245
 246 /*****************************************************************************
 247  * Local prototypes
 248  *****************************************************************************/
 249
 250
 251 /* Main functions */
 252     static int Demux( demux_t *p_demux );
 253     static int Control( demux_t *, int, va_list );
 254     static void Run    ( services_discovery_t *p_sd );
 255
 256 /* Main parsing functions */
 257     static int ParseConnection( vlc_object_t *p_obj, sdp_t *p_sdp );
 258     static int ParseSAP( services_discovery_t *p_sd, const uint8_t *p_buffer, size_t i_read );
 259     static sdp_t *ParseSDP (vlc_object_t *p_sd, const char *psz_sdp);
 260     static sap_announce_t *CreateAnnounce( services_discovery_t *, uint16_t, sdp_t * );
 261     static int RemoveAnnounce( services_discovery_t *p_sd, sap_announce_t *p_announce );
 262
 263 /* Helper functions */
 264     static char *GetAttribute( sdp_t *p_sdp, const char *psz_search );
 265     static vlc_bool_t IsSameSession( sdp_t *p_sdp1, sdp_t *p_sdp2 );
 266     static int InitSocket( services_discovery_t *p_sd, const char *psz_address, int i_port );
 267     static int Decompress( const unsigned char *psz_src, unsigned char **_dst, int i_len );
 268     static void FreeSDP( sdp_t *p_sdp );
 269
 270 /*****************************************************************************
 271  * Open: initialize and create stuff
 272  *****************************************************************************/
 273 static int Open( vlc_object_t *p_this )
 274 {
 275     services_discovery_t *p_sd = ( services_discovery_t* )p_this;
 276     services_discovery_sys_t *p_sys  = (services_discovery_sys_t *)
 277                                 malloc( sizeof( services_discovery_sys_t ) );
 278
 279     p_sys->i_timeout = var_CreateGetInteger( p_sd, "sap-timeout" );
 280
 281     p_sd->pf_run = Run;
 282     p_sd->p_sys  = p_sys;
 283
 284     p_sys->pi_fd = NULL;
 285     p_sys->i_fd = 0;
 286
 287     p_sys->b_strict = var_CreateGetInteger( p_sd, "sap-strict");
 288     p_sys->b_parse = var_CreateGetInteger( p_sd, "sap-parse" );
 289
 290 #if 0
 291     if( var_CreateGetInteger( p_sd, "sap-cache" ) )
 292     {
 293         CacheLoad( p_sd );
 294     }
 295 #endif
 296
 297     /* Cache sap_timeshift value */
 298     p_sys->b_timeshift = var_CreateGetInteger( p_sd, "sap-timeshift" )
 299             ? VLC_TRUE : VLC_FALSE;
 300
 301     /* Create our playlist node */
 302     pl_Yield( p_sd );
 303
 304     playlist_NodesPairCreate( pl_Get( p_sd ), _("SAP sessions"),
 305                               &p_sys->p_node_cat, &p_sys->p_node_one,
 306                               VLC_TRUE );
 307
 308     p_sys->i_announces = 0;
 309     p_sys->pp_announces = NULL;
 310
 311     return VLC_SUCCESS;
 312 }
 313
 314 /*****************************************************************************
 315  * OpenDemux: initialize and create stuff
 316  *****************************************************************************/
 317 static int OpenDemux( vlc_object_t *p_this )
 318 {
 319     demux_t *p_demux = (demux_t *)p_this;
 320     uint8_t *p_peek;
 321     int i_max_sdp = 1024;
 322     int i_sdp = 0;
 323     char *psz_sdp = NULL;
 324     sdp_t *p_sdp = NULL;
 325
 326     if( !var_CreateGetInteger( p_demux, "sap-parse" ) )
 327     {
 328         /* We want livedotcom module to parse this SDP file */
 329         return VLC_EGENERIC;
 330     }
 331
 332     /* Probe for SDP */
 333     if( p_demux->s )
 334     {
 335         if( stream_Peek( p_demux->s, &p_peek, 7 ) < 7 ) return VLC_EGENERIC;
 336
 337         if( strncmp( (char*)p_peek, "v=0\r\n", 5 ) &&
 338             strncmp( (char*)p_peek, "v=0\n", 4 ) &&
 339             ( p_peek[0] < 'a' || p_peek[0] > 'z' || p_peek[1] != '=' ) )
 340         {
 341             return VLC_EGENERIC;
 342         }
 343     }
 344
 345     psz_sdp = (char *)malloc( i_max_sdp );
 346     if( !psz_sdp ) return VLC_EGENERIC;
 347
 348     /* Gather the complete sdp file */
 349     for( ;; )
 350     {
 351         int i_read = stream_Read( p_demux->s,
 352                                   &psz_sdp[i_sdp], i_max_sdp - i_sdp - 1 );
 353
 354         if( i_read < 0 )
 355         {
 356             msg_Err( p_demux, "failed to read SDP" );
 357             goto error;
 358         }
 359
 360         i_sdp += i_read;
 361
 362         if( i_read < i_max_sdp - i_sdp - 1 )
 363         {
 364             psz_sdp[i_sdp] = '\0';
 365             break;
 366         }
 367
 368         i_max_sdp += 1000;
 369         psz_sdp = (char *)realloc( psz_sdp, i_max_sdp );
 370     }
 371
 372     p_sdp = ParseSDP( VLC_OBJECT(p_demux), psz_sdp );
 373
 374     if( !p_sdp )
 375     {
 376         msg_Warn( p_demux, "invalid SDP");
 377         goto error;
 378     }
 379
 380     if( p_sdp->i_media > 1 )
 381     {
 382         goto error;
 383     }
 384
 385     if( ParseConnection( VLC_OBJECT( p_demux ), p_sdp ) )
 386     {
 387         p_sdp->psz_uri = NULL;
 388     }
 389     if( p_sdp->i_media_type != 33 && p_sdp->i_media_type != 32 &&
 390         p_sdp->i_media_type != 14 )
 391         goto error;
 392
 393     if( p_sdp->psz_uri == NULL ) goto error;
 394
 395     p_demux->p_sys = (demux_sys_t *)malloc( sizeof(demux_sys_t) );
 396     p_demux->p_sys->p_sdp = p_sdp;
 397     p_demux->pf_control = Control;
 398     p_demux->pf_demux = Demux;
 399
 400     FREENULL( psz_sdp );
 401     return VLC_SUCCESS;
 402
 403 error:
 404     FREENULL( psz_sdp );
 405     if( p_sdp ) FreeSDP( p_sdp ); p_sdp = NULL;
 406     stream_Seek( p_demux->s, 0 );
 407     return VLC_EGENERIC;
 408 }
 409
 410 /*****************************************************************************
 411  * Close:
 412  *****************************************************************************/
 413 static void Close( vlc_object_t *p_this )
 414 {
 415     services_discovery_t *p_sd = ( services_discovery_t* )p_this;
 416     services_discovery_sys_t    *p_sys  = p_sd->p_sys;
 417
 418     int i;
 419
 420     for( i = p_sys->i_fd-1 ; i >= 0 ; i-- )
 421     {
 422         net_Close( p_sys->pi_fd[i] );
 423     }
 424     FREENULL( p_sys->pi_fd );
 425
 426 #if 0
 427     if( config_GetInt( p_sd, "sap-cache" ) )
 428     {
 429         CacheSave( p_sd );
 430     }
 431 #endif
 432
 433     for( i = p_sys->i_announces  - 1;  i>= 0; i-- )
 434     {
 435         RemoveAnnounce( p_sd, p_sys->pp_announces[i] );
 436     }
 437     FREENULL( p_sys->pp_announces );
 438
 439     playlist_NodeDelete( pl_Get(p_sd), p_sys->p_node_cat, VLC_TRUE,
 440                          VLC_TRUE );
 441     playlist_NodeDelete( pl_Get(p_sd), p_sys->p_node_one, VLC_TRUE,
 442                          VLC_TRUE );
 443     pl_Release( p_sd );
 444     free( p_sys );
 445 }
 446
 447 /*****************************************************************************
 448  * CloseDemux: Close the demuxer
 449  *****************************************************************************/
 450 static void CloseDemux( vlc_object_t *p_this )
 451 {
 452     demux_t *p_demux = (demux_t *)p_this;
 453     if( p_demux->p_sys )
 454     {
 455         if( p_demux->p_sys->p_sdp ) { FreeSDP( p_demux->p_sys->p_sdp ); p_demux->p_sys->p_sdp = NULL; }
 456         free( p_demux->p_sys );
 457     }
 458 }
 459
 460 /*****************************************************************************
 461  * Run: main SAP thread
 462  *****************************************************************************
 463  * Listens to SAP packets, and sends them to packet_handle
 464  *****************************************************************************/
 465 #define MAX_SAP_BUFFER 5000
 466
 467 static void Run( services_discovery_t *p_sd )
 468 {
 469     char *psz_addr;
 470     int i;
 471
 472     /* Braindead Winsock DNS resolver will get stuck over 2 seconds per failed
 473      * DNS queries, even if the DNS server returns an error with milliseconds.
 474      * You don't want to know why the bug (as of XP SP2) wasn't fixed since
 475      * Winsock 1.1 from Windows 95, if not Windows 3.1.
 476      * Anyway, to avoid a 30 seconds delay for failed IPv6 socket creation,
 477      * we have to open sockets in Run() rather than Open(). */
 478     if( var_CreateGetInteger( p_sd, "sap-ipv4" ) )
 479     {
 480         InitSocket( p_sd, SAP_V4_GLOBAL_ADDRESS, SAP_PORT );
 481         InitSocket( p_sd, SAP_V4_ORG_ADDRESS, SAP_PORT );
 482         InitSocket( p_sd, SAP_V4_LOCAL_ADDRESS, SAP_PORT );
 483         InitSocket( p_sd, SAP_V4_LINK_ADDRESS, SAP_PORT );
 484     }
 485     if( var_CreateGetInteger( p_sd, "sap-ipv6" ) )
 486     {
 487         char psz_address[] = SAP_V6_1"0"SAP_V6_2;
 488         const char *c_scope;
 489
 490         for( c_scope = ipv6_scopes; *c_scope; c_scope++ )
 491         {
 492             psz_address[sizeof(SAP_V6_1) - 1] = *c_scope;
 493             InitSocket( p_sd, psz_address, SAP_PORT );
 494         }
 495     }
 496
 497     psz_addr = var_CreateGetString( p_sd, "sap-addr" );
 498     if( psz_addr && *psz_addr )
 499     {
 500         InitSocket( p_sd, psz_addr, SAP_PORT );
 501         free( psz_addr );
 502     }
 503
 504     if( p_sd->p_sys->i_fd == 0 )
 505     {
 506         msg_Err( p_sd, "unable to listen on any address" );
 507         return;
 508     }
 509
 510     /* read SAP packets */
 511     while( !p_sd->b_die )
 512     {
 513         int i_read;
 514         uint8_t p_buffer[MAX_SAP_BUFFER+1];
 515
 516         i_read = net_Select( p_sd, p_sd->p_sys->pi_fd, NULL,
 517                              p_sd->p_sys->i_fd, p_buffer,
 518                              MAX_SAP_BUFFER, 500000 );
 519
 520         /* Check for items that need deletion */
 521         for( i = 0; i < p_sd->p_sys->i_announces; i++ )
 522         {
 523             mtime_t i_timeout = ( mtime_t ) 1000000 * p_sd->p_sys->i_timeout;
 524
 525             if( mdate() - p_sd->p_sys->pp_announces[i]->i_last > i_timeout )
 526             {
 527                 RemoveAnnounce( p_sd, p_sd->p_sys->pp_announces[i] );
 528             }
 529         }
 530
 531         /* Minimum length is > 6 */
 532         if( i_read <= 6 )
 533         {
 534             if( i_read < 0 )
 535             {
 536                 msg_Warn( p_sd, "socket read error" );
 537             }
 538             continue;
 539         }
 540
 541         p_buffer[i_read] = '\0';
 542
 543         /* Parse the packet */
 544         ParseSAP( p_sd, p_buffer, i_read );
 545     }
 546 }
 547
 548 /**********************************************************************
 549  * Demux: reads and demuxes data packets
 550  * Return -1 if error, 0 if EOF, 1 else
 551  **********************************************************************/
 552 static int Demux( demux_t *p_demux )
 553 {
 554     sdp_t *p_sdp = p_demux->p_sys->p_sdp;
 555     input_thread_t *p_input;
 556     input_item_t *p_parent_input;
 557
 558     playlist_t *p_playlist = pl_Yield( p_demux );
 559     p_input = (input_thread_t *)vlc_object_find( p_demux, VLC_OBJECT_INPUT,
 560                                                  FIND_PARENT );
 561     assert( p_input );
 562     if( !p_input )
 563     {
 564         msg_Err( p_demux, "parent input could not be found" );
 565         return VLC_EGENERIC;
 566     }
 567
 568     p_parent_input = p_input->input.p_item;
 569
 570     vlc_mutex_lock( &p_parent_input->lock );
 571     FREENULL( p_parent_input->psz_uri );
 572     p_parent_input->psz_uri = strdup( p_sdp->psz_uri );
 573     FREENULL( p_parent_input->psz_name );
 574     p_parent_input->psz_name = strdup( EnsureUTF8( p_sdp->psz_sessionname ) );
 575     p_parent_input->i_type = ITEM_TYPE_NET;
 576
 577     if( p_playlist->status.p_item &&
 578              p_playlist->status.p_item->p_input == p_parent_input )
 579     {
 580         playlist_Control( p_playlist, PLAYLIST_VIEWPLAY,
 581                           p_playlist->status.p_node, p_playlist->status.p_item );
 582     }
 583
 584     vlc_mutex_unlock( &p_parent_input->lock );
 585     vlc_object_release( p_input );
 586     vlc_object_release( p_playlist );
 587
 588     return VLC_SUCCESS;
 589 }
 590
 591 static int Control( demux_t *p_demux, int i_query, va_list args )
 592 {
 593     return VLC_EGENERIC;
 594 }
 595
 596 /**************************************************************
 597  * Local functions
 598  **************************************************************/
 599
 600 /* i_read is at least > 6 */
 601 static int ParseSAP( services_discovery_t *p_sd, const uint8_t *buf,
 602                      size_t len )
 603 {
 604     int i;
 605     const char          *psz_sdp;
 606     const uint8_t *end = buf + len;
 607     sdp_t               *p_sdp;
 608
 609     assert (buf[len] == '\0');
 610
 611     if (len < 4)
 612         return VLC_EGENERIC;
 613
 614     uint8_t flags = buf[0];
 615
 616     /* First, check the sap announce is correct */
 617     if ((flags >> 5) != 1)
 618         return VLC_EGENERIC;
 619
 620     vlc_bool_t b_ipv6 = (flags & 0x10) != 0;
 621     vlc_bool_t b_need_delete = (flags & 0x04) != 0;
 622
 623     if (flags & 0x02)
 624     {
 625         msg_Dbg( p_sd, "encrypted packet, unsupported" );
 626         return VLC_EGENERIC;
 627     }
 628
 629     vlc_bool_t b_compressed = (flags & 0x01) != 0;
 630
 631     uint16_t i_hash = U16_AT (buf + 2);
 632
 633     if( p_sd->p_sys->b_strict && i_hash == 0 )
 634     {
 635         msg_Dbg( p_sd, "strict mode, discarding announce with null id hash");
 636         return VLC_EGENERIC;
 637     }
 638
 639     // Skips source address and auth data
 640     buf += 4 + (b_ipv6 ? 16 : 4) + buf[1];
 641     if (buf > end)
 642         return VLC_EGENERIC;
 643
 644     uint8_t *decomp = NULL;
 645     if( b_compressed )
 646     {
 647         int newsize = Decompress (buf, &decomp, end - buf);
 648         if (newsize < 0)
 649         {
 650             msg_Warn( p_sd, "decompression of sap packet failed" );
 651             return VLC_EGENERIC;
 652         }
 653
 654         decomp = realloc (decomp, newsize + 1);
 655         decomp[newsize] = '\0';
 656
 657         psz_sdp = (const char *)decomp;
 658         len = newsize;
 659     }
 660     else
 661     {
 662         psz_sdp = (const char *)buf;
 663         len = end - buf;
 664     }
 665
 666     /* len is a strlen here here. both buf and decomp are len+1 where the 1 should be a \0 */
 667     assert( psz_sdp[len] == '\0');
 668
 669     /* Skip payload type */
 670     /* SAPv1 has implicit "application/sdp" payload type: first line is v=0 */
 671     if (strncmp (psz_sdp, "v=0", 3))
 672     {
 673         size_t clen = strlen (psz_sdp) + 1;
 674
 675         if (strcmp (psz_sdp, "application/sdp"))
 676         {
 677             msg_Dbg (p_sd, "unsupported content type: %s", psz_sdp);
 678             return VLC_EGENERIC;
 679         }
 680
 681         // skips content type
 682         if (len <= clen)
 683             return VLC_EGENERIC;
 684
 685         len -= clen;
 686         psz_sdp += clen;
 687     }
 688
 689     /* Parse SDP info */
 690     p_sdp = ParseSDP( VLC_OBJECT(p_sd), psz_sdp );
 691
 692     if( p_sdp == NULL )
 693     {
 694         return VLC_EGENERIC;
 695     }
 696
 697     /* Decide whether we should add a playlist item for this SDP */
 698     /* Parse connection information (c= & m= ) */
 699     if( ParseConnection( VLC_OBJECT(p_sd), p_sdp ) )
 700     {
 701         p_sdp->psz_uri = NULL;
 702     }
 703
 704     /* Multi-media or no-parse -> pass to LIVE.COM */
 705     if( p_sdp->i_media > 1 || ( p_sdp->i_media_type != 14 &&
 706                                 p_sdp->i_media_type != 32 &&
 707                                 p_sdp->i_media_type != 33) ||
 708         p_sd->p_sys->b_parse == VLC_FALSE )
 709     {
 710         if( p_sdp->psz_uri ) free( p_sdp->psz_uri );
 711         asprintf( &p_sdp->psz_uri, "sdp://%s", p_sdp->psz_sdp );
 712     }
 713
 714     if( p_sdp->psz_uri == NULL ) return VLC_EGENERIC;
 715
 716     for( i = 0 ; i< p_sd->p_sys->i_announces ; i++ )
 717     {
 718         /* FIXME: slow */
 719         /* FIXME: we create a new announce each time the sdp changes */
 720         if( IsSameSession( p_sd->p_sys->pp_announces[i]->p_sdp, p_sdp ) )
 721         {
 722             if( b_need_delete )
 723             {
 724                 RemoveAnnounce( p_sd, p_sd->p_sys->pp_announces[i]);
 725             }
 726             else
 727             {
 728                 p_sd->p_sys->pp_announces[i]->i_last = mdate();
 729             }
 730             FreeSDP( p_sdp ); p_sdp = NULL;
 731             return VLC_SUCCESS;
 732         }
 733     }
 734     /* Add item */
 735     if( p_sdp->i_media > 1 )
 736     {
 737         msg_Dbg( p_sd, "passing to liveMedia" );
 738     }
 739
 740     CreateAnnounce( p_sd, i_hash, p_sdp );
 741
 742     FREENULL (decomp);
 743     return VLC_SUCCESS;
 744 }
 745
 746 sap_announce_t *CreateAnnounce( services_discovery_t *p_sd, uint16_t i_hash,
 747                                 sdp_t *p_sdp )
 748 {
 749     input_item_t *p_input;
 750     playlist_item_t     *p_item, *p_child;
 751     char *psz_value;
 752     sap_announce_t *p_sap = (sap_announce_t *)malloc(
 753                                         sizeof(sap_announce_t ) );
 754     services_discovery_sys_t *p_sys;
 755     if( p_sap == NULL )
 756         return NULL;
 757
 758     p_sys = p_sd->p_sys;
 759
 760     EnsureUTF8( p_sdp->psz_sessionname );
 761     p_sap->i_last = mdate();
 762     p_sap->i_hash = i_hash;
 763     p_sap->p_sdp = p_sdp;
 764
 765     /* Create the actual playlist item here */
 766     p_input = input_ItemNewWithType( VLC_OBJECT(p_sd),
 767                                      p_sap->p_sdp->psz_uri,
 768                                      p_sdp->psz_sessionname,
 769                                      0, NULL, -1, ITEM_TYPE_NET );
 770     p_sap->i_input_id = p_input->i_id;
 771     if( !p_input )
 772     {
 773         free( p_sap );
 774         return NULL;
 775     }
 776
 777     if( p_sys->b_timeshift )
 778         input_ItemAddOption( p_input, ":access-filter=timeshift" );
 779
 780     psz_value = GetAttribute( p_sap->p_sdp, "tool" );
 781     if( psz_value != NULL )
 782     {
 783         input_ItemAddInfo( p_input, _("Session"),_("Tool"), psz_value );
 784     }
 785     if( strcmp( p_sdp->psz_username, "-" ) )
 786     {
 787         input_ItemAddInfo( p_input, _("Session"),
 788                                 _("User"), p_sdp->psz_username );
 789     }
 790
 791     /* Handle group */
 792     psz_value = GetAttribute( p_sap->p_sdp, "x-plgroup" );
 793     if( psz_value == NULL )
 794         psz_value = GetAttribute( p_sap->p_sdp, "plgroup" );
 795
 796     if( psz_value != NULL )
 797     {
 798         EnsureUTF8( psz_value );
 799
 800         p_child = playlist_ChildSearchName( p_sys->p_node_cat, psz_value );
 801
 802         if( p_child == NULL )
 803         {
 804             p_child = playlist_NodeCreate( pl_Get( p_sd ), psz_value,
 805                                            p_sys->p_node_cat );
 806             p_child->i_flags &= ~PLAYLIST_SKIP_FLAG;
 807         }
 808     }
 809     else
 810     {
 811         p_child = p_sys->p_node_cat;
 812     }
 813
 814     p_item = playlist_NodeAddInput( pl_Get( p_sd ), p_input, p_child,
 815                                     PLAYLIST_APPEND, PLAYLIST_END );
 816     p_item->i_flags &= ~PLAYLIST_SKIP_FLAG;
 817     p_item->i_flags &= ~PLAYLIST_SAVE_FLAG;
 818     p_sap->i_item_id_cat = p_item->i_id;
 819
 820     p_item = playlist_NodeAddInput( pl_Get( p_sd ), p_input,
 821                         p_sys->p_node_one, PLAYLIST_APPEND, PLAYLIST_END );
 822     p_item->i_flags &= ~PLAYLIST_SKIP_FLAG;
 823     p_item->i_flags &= ~PLAYLIST_SAVE_FLAG;
 824     p_sap->i_item_id_one = p_item->i_id;
 825
 826     TAB_APPEND( p_sys->i_announces, p_sys->pp_announces, p_sap );
 827
 828     return p_sap;
 829 }
 830
 831 static char *GetAttribute( sdp_t *p_sdp, const char *psz_search )
 832 {
 833     int i;
 834
 835     for( i = 0 ; i< p_sdp->i_attributes; i++ )
 836     {
 837         if( !strncmp( p_sdp->pp_attributes[i]->psz_field, psz_search,
 838                       strlen( p_sdp->pp_attributes[i]->psz_field ) ) )
 839         {
 840             return p_sdp->pp_attributes[i]->psz_value;
 841         }
 842     }
 843     return NULL;
 844 }
 845
 846
 847 /* Fill p_sdp->psz_uri */
 848 static int ParseConnection( vlc_object_t *p_obj, sdp_t *p_sdp )
 849 {
 850     char *psz_eof = NULL;
 851     char *psz_parse = NULL;
 852     char psz_uri[1026];
 853     char *psz_proto = NULL;
 854     int i_port = 0;
 855
 856     /* Parse c= field */
 857     if( p_sdp->psz_connection )
 858     {
 859         char hostname[1024];
 860         int ipv;
 861
 862         /*
 863          * NOTE: we ignore the TTL parameter on-purpose, as some SAP
 864          * advertisers don't include it (and it is utterly useless).
 865          */
 866         if (sscanf (p_sdp->psz_connection, "IN IP%d %1023[^/]", &ipv,
 867                     hostname) != 2)
 868         {
 869             msg_Warn (p_obj, "unable to parse c field: \"%s\"",
 870                       p_sdp->psz_connection);
 871             return VLC_EGENERIC;
 872         }
 873
 874         switch (ipv)
 875         {
 876             case 4:
 877             case 6:
 878                 break;
 879
 880             default:
 881                 msg_Warn (p_obj, "unknown IP version %d", ipv);
 882                 return VLC_EGENERIC;
 883         }
 884
 885         if (strchr (hostname, ':') != NULL)
 886             sprintf (psz_uri, "[%s]", hostname);
 887         else
 888             strcpy (psz_uri, hostname);
 889     }
 890
 891     /* Parse m= field */
 892     if( p_sdp->psz_media )
 893     {
 894         psz_parse = p_sdp->psz_media;
 895
 896         psz_eof = strchr( psz_parse, ' ' );
 897
 898         if( psz_eof )
 899         {
 900             *psz_eof = '\0';
 901
 902             /*
 903              * That's ugly. We should go through every media, and make sure
 904              * at least one of them is audio or video. In the mean time, I
 905              * need to accept data too.
 906              */
 907             if( strncmp( psz_parse, "audio", 5 )
 908              && strncmp( psz_parse, "video", 5 )
 909              && strncmp( psz_parse, "data", 4 ) )
 910             {
 911                 msg_Warn( p_obj, "unhandled media type \"%s\"", psz_parse );
 912                 return VLC_EGENERIC;
 913             }
 914
 915             psz_parse = psz_eof + 1;
 916         }
 917         else
 918         {
 919             msg_Warn( p_obj, "unable to parse m field (1)");
 920             return VLC_EGENERIC;
 921         }
 922
 923         psz_eof = strchr( psz_parse, ' ' );
 924
 925         if( psz_eof )
 926         {
 927             *psz_eof = '\0';
 928
 929             /* FIXME : multiple port ! */
 930             i_port = atoi( psz_parse );
 931
 932             if( i_port <= 0 || i_port >= 65536 )
 933             {
 934                 msg_Warn( p_obj, "invalid transport port %i", i_port );
 935             }
 936
 937             psz_parse = psz_eof + 1;
 938         }
 939         else
 940         {
 941             msg_Warn( p_obj, "unable to parse m field (2)");
 942             return VLC_EGENERIC;
 943         }
 944
 945         psz_eof = strchr( psz_parse, ' ' );
 946
 947         if( psz_eof )
 948         {
 949             *psz_eof = '\0';
 950             psz_proto = strdup( psz_parse );
 951
 952             psz_parse = psz_eof + 1;
 953             p_sdp->i_media_type = atoi( psz_parse );
 954
 955         }
 956         else
 957         {
 958             msg_Dbg( p_obj, "incorrect m field, %s", p_sdp->psz_media );
 959             p_sdp->i_media_type = 33;
 960             psz_proto = strdup( psz_parse );
 961         }
 962     }
 963
 964     if( psz_proto && !strncmp( psz_proto, "RTP/AVP", 7 ) )
 965     {
 966         free( psz_proto );
 967         psz_proto = strdup( "rtp" );
 968     }
 969     if( psz_proto && !strncasecmp( psz_proto, "UDP", 3 ) )
 970     {
 971         free( psz_proto );
 972         psz_proto = strdup( "udp" );
 973     }
 974
 975     /* FIXME: HTTP support */
 976
 977     if( i_port == 0 )
 978     {
 979         i_port = 1234;
 980     }
 981
 982     /* handle SSM case */
 983     psz_parse = GetAttribute( p_sdp, "source-filter" );
 984     char psz_source[258] = "";
 985     if (psz_parse != NULL)
 986     {
 987         char psz_source_ip[256];
 988
 989         if (sscanf (psz_parse, " incl IN IP%*c %*s %255s ", psz_source_ip) == 1)
 990         {
 991             if (strchr (psz_source_ip, ':') != NULL)
 992                 sprintf (psz_source, "[%s]", psz_source_ip);
 993             else
 994                 strcpy (psz_source, psz_source_ip);
 995         }
 996     }
 997
 998     asprintf( &p_sdp->psz_uri, "%s://%s@%s:%i", psz_proto, psz_source,
 999               psz_uri, i_port );
1000
1001     FREENULL( psz_proto );
1002     return VLC_SUCCESS;
1003 }
1004
1005 /***********************************************************************
1006  * ParseSDP : SDP parsing
1007  * *********************************************************************
1008  * Validate SDP and parse all fields
1009  ***********************************************************************/
1010 static sdp_t *ParseSDP (vlc_object_t *p_obj, const char *psz_sdp)
1011 {
1012     sdp_t *p_sdp;
1013     vlc_bool_t b_invalid = VLC_FALSE;
1014     vlc_bool_t b_end = VLC_FALSE;
1015     if( psz_sdp == NULL )
1016     {
1017         return NULL;
1018     }
1019
1020     if( psz_sdp[0] != 'v' || psz_sdp[1] != '=' )
1021     {
1022         msg_Warn( p_obj, "bad packet" );
1023         return NULL;
1024     }
1025
1026     p_sdp = (sdp_t *)malloc( sizeof( sdp_t ) );
1027     if( p_sdp == NULL )
1028         return NULL;
1029
1030     /* init to 0 */
1031     memset( p_sdp, 0, sizeof( sdp_t ) );
1032
1033     p_sdp->psz_sdp = strdup( psz_sdp );
1034     if( p_sdp->psz_sdp == NULL )
1035     {
1036         free( p_sdp );
1037         return NULL;
1038     }
1039
1040     while( *psz_sdp != '\0' && b_end == VLC_FALSE  )
1041     {
1042         char *psz_eol = NULL;
1043         char *psz_eof = NULL;
1044         char *psz_parse = NULL;
1045         char *psz_sess_id = NULL;
1046
1047         while( *psz_sdp == '\r' || *psz_sdp == '\n' ||
1048                *psz_sdp == ' ' || *psz_sdp == '\t' )
1049         {
1050             psz_sdp++;
1051         }
1052
1053         if( ( psz_eol = strchr( psz_sdp, '\n' ) ) == NULL )
1054         {
1055             psz_eol = (char *)psz_sdp + strlen( psz_sdp );
1056             b_end = VLC_TRUE;
1057         }
1058         if( psz_eol > psz_sdp && *( psz_eol - 1 ) == '\r' )
1059         {
1060             psz_eol--;
1061         }
1062
1063         if( psz_eol <= psz_sdp )
1064         {
1065             break;
1066         }
1067         *psz_eol++ = '\0';
1068
1069         /* no space allowed between fields */
1070         if( psz_sdp[1] != '=' )
1071         {
1072             msg_Warn( p_obj, "invalid packet" ) ;
1073             FreeSDP( p_sdp ); p_sdp = NULL;
1074             return NULL;
1075         }
1076
1077         /* Now parse each line */
1078         switch( psz_sdp[0] )
1079         {
1080             case( 'v' ):
1081                 break;
1082             case( 's' ):
1083                 p_sdp->psz_sessionname = strdup( &psz_sdp[2] );
1084                 break;
1085             case ( 'o' ):
1086             {
1087                 int i_field = 0;
1088                 /* o field is <username> <session id> <version>
1089                  *  <network type> <address type> <address> */
1090
1091 #define GET_FIELD( store ) \
1092                 psz_eof = strchr( psz_parse, ' ' ); \
1093                 if( psz_eof ) \
1094                 { \
1095                     *psz_eof=0; store = strdup( psz_parse ); \
1096                 } \
1097                 else \
1098                 { \
1099                     if( i_field != 5 ) \
1100                     { \
1101                         b_invalid = VLC_TRUE; break; \
1102                     } \
1103                     else \
1104                     { \
1105                         store = strdup( psz_parse ); \
1106                     } \
1107                 }; \
1108                 psz_parse = psz_eof + 1; i_field++;
1109
1110
1111                 psz_parse = (char *)&psz_sdp[2];
1112                 GET_FIELD( p_sdp->psz_username );
1113                 GET_FIELD( psz_sess_id );
1114
1115                 p_sdp->i_session_id = atoll( psz_sess_id );
1116
1117                 FREENULL( psz_sess_id );
1118
1119                 GET_FIELD( psz_sess_id );
1120                 FREENULL( psz_sess_id );
1121
1122                 GET_FIELD( p_sdp->psz_network_type );
1123                 GET_FIELD( p_sdp->psz_address_type );
1124                 GET_FIELD( p_sdp->psz_address );
1125
1126                 break;
1127             }
1128             case( 'i' ):
1129             case( 'u' ):
1130             case( 'e' ):
1131             case( 'p' ):
1132             case( 't' ):
1133             case( 'r' ):
1134                 break;
1135             case( 'a' ): /* attribute */
1136             {
1137                 char *psz_eon = strchr( &psz_sdp[2], ':' );
1138                 attribute_t *p_attr = malloc( sizeof( attribute_t ) );
1139
1140                 /* Attribute with value */
1141                 if( psz_eon )
1142                 {
1143                     *psz_eon++ = '\0';
1144
1145                     p_attr->psz_field = strdup( &psz_sdp[2] );
1146                     p_attr->psz_value = strdup( psz_eon );
1147                 }
1148                 else /* Attribute without value */
1149                 {
1150                     p_attr->psz_field = strdup( &psz_sdp[2] );
1151                     p_attr->psz_value = NULL;
1152                 }
1153
1154                 TAB_APPEND( p_sdp->i_attributes, p_sdp->pp_attributes, p_attr );
1155                 break;
1156             }
1157
1158             case( 'm' ): /* Media announcement */
1159             {
1160                 /* If we have several medias, we pass the announcement to
1161                  * LIVE.COM, so just count them */
1162                 p_sdp->i_media++;
1163                 if( p_sdp->i_media == 1 )
1164                 {
1165                     p_sdp->psz_media = strdup( &psz_sdp[2] );
1166                 }
1167                 break;
1168             }
1169
1170             case( 'c' ):
1171             {
1172                 if( p_sdp->psz_connection != NULL ) // FIXME
1173                     break;
1174
1175                 p_sdp->psz_connection = strdup( &psz_sdp[2] );
1176                 break;
1177             }
1178
1179             default:
1180                break;
1181         }
1182
1183         if( b_invalid )
1184         {
1185             FreeSDP( p_sdp ); p_sdp = NULL;
1186             return NULL;
1187         }
1188
1189         psz_sdp = psz_eol;
1190     }
1191
1192     return p_sdp;
1193 }
1194
1195 static int InitSocket( services_discovery_t *p_sd, const char *psz_address,
1196                        int i_port )
1197 {
1198     int i_fd = net_ListenUDP1 ((vlc_object_t *)p_sd, psz_address, i_port);
1199     if (i_fd == -1)
1200         return VLC_EGENERIC;
1201
1202     net_StopSend( i_fd );
1203     INSERT_ELEM (p_sd->p_sys->pi_fd, p_sd->p_sys->i_fd,
1204                  p_sd->p_sys->i_fd, i_fd);
1205     return VLC_SUCCESS;
1206 }
1207
1208 static int Decompress( const unsigned char *psz_src, unsigned char **_dst, int i_len )
1209 {
1210 #ifdef HAVE_ZLIB_H
1211     int i_result, i_dstsize, n = 0;
1212     unsigned char *psz_dst = NULL;
1213     z_stream d_stream;
1214
1215     memset (&d_stream, 0, sizeof (d_stream));
1216
1217     i_result = inflateInit(&d_stream);
1218     if( i_result != Z_OK )
1219         return( -1 );
1220
1221     d_stream.next_in = (Bytef *)psz_src;
1222     d_stream.avail_in = i_len;
1223
1224     do
1225     {
1226         n++;
1227         psz_dst = (unsigned char *)realloc( psz_dst, n * 1000 );
1228         d_stream.next_out = (Bytef *)&psz_dst[(n - 1) * 1000];
1229         d_stream.avail_out = 1000;
1230
1231         i_result = inflate(&d_stream, Z_NO_FLUSH);
1232         if( ( i_result != Z_OK ) && ( i_result != Z_STREAM_END ) )
1233         {
1234             inflateEnd( &d_stream );
1235             return( -1 );
1236         }
1237     }
1238     while( ( d_stream.avail_out == 0 ) && ( d_stream.avail_in != 0 ) &&
1239            ( i_result != Z_STREAM_END ) );
1240
1241     i_dstsize = d_stream.total_out;
1242     inflateEnd( &d_stream );
1243
1244     *_dst = (unsigned char *)realloc( psz_dst, i_dstsize );
1245
1246     return i_dstsize;
1247 #else
1248     (void)psz_src;
1249     (void)_dst;
1250     (void)i_len;
1251     return -1;
1252 #endif
1253 }
1254
1255
1256 static void FreeSDP( sdp_t *p_sdp )
1257 {
1258     int i;
1259     FREENULL( p_sdp->psz_sdp );
1260     FREENULL( p_sdp->psz_sessionname );
1261     FREENULL( p_sdp->psz_connection );
1262     FREENULL( p_sdp->psz_media );
1263     FREENULL( p_sdp->psz_uri );
1264     FREENULL( p_sdp->psz_username );
1265     FREENULL( p_sdp->psz_network_type );
1266
1267     FREENULL( p_sdp->psz_address );
1268     FREENULL( p_sdp->psz_address_type );
1269
1270     for( i= p_sdp->i_attributes - 1; i >= 0 ; i-- )
1271     {
1272         struct attribute_t *p_attr = p_sdp->pp_attributes[i];
1273         FREENULL( p_sdp->pp_attributes[i]->psz_field );
1274         FREENULL( p_sdp->pp_attributes[i]->psz_value );
1275         REMOVE_ELEM( p_sdp->pp_attributes, p_sdp->i_attributes, i);
1276         FREENULL( p_attr );
1277     }
1278     FREENULL( p_sdp );
1279 }
1280
1281 static int RemoveAnnounce( services_discovery_t *p_sd,
1282                            sap_announce_t *p_announce )
1283 {
1284     int i;
1285
1286     if( p_announce->p_sdp )
1287     {
1288         FreeSDP( p_announce->p_sdp );
1289         p_announce->p_sdp = NULL;
1290     }
1291
1292     if( p_announce->i_input_id > -1 )
1293         playlist_LockDeleteAllFromInput( pl_Get(p_sd), p_announce->i_input_id );
1294
1295     for( i = 0; i< p_sd->p_sys->i_announces; i++)
1296     {
1297         if( p_sd->p_sys->pp_announces[i] == p_announce )
1298         {
1299             REMOVE_ELEM( p_sd->p_sys->pp_announces, p_sd->p_sys->i_announces,
1300                          i);
1301             break;
1302         }
1303     }
1304
1305     free( p_announce );
1306
1307     return VLC_SUCCESS;
1308 }
1309
1310 static vlc_bool_t IsSameSession( sdp_t *p_sdp1, sdp_t *p_sdp2 )
1311 {
1312     /* A session is identified by
1313      * username, session_id, network type, address type and address */
1314     if( p_sdp1->psz_username && p_sdp2->psz_username &&
1315         p_sdp1->psz_network_type && p_sdp2->psz_network_type &&
1316         p_sdp1->psz_address_type && p_sdp2->psz_address_type &&
1317         p_sdp1->psz_address &&  p_sdp2->psz_address )
1318     {
1319         if(!strcmp( p_sdp1->psz_username , p_sdp2->psz_username ) &&
1320            !strcmp( p_sdp1->psz_network_type , p_sdp2->psz_network_type ) &&
1321            !strcmp( p_sdp1->psz_address_type , p_sdp2->psz_address_type ) &&
1322            !strcmp( p_sdp1->psz_address , p_sdp2->psz_address ) &&
1323            p_sdp1->i_session_id == p_sdp2->i_session_id )
1324         {
1325             return VLC_TRUE;
1326         }
1327         else
1328         {
1329             return VLC_FALSE;
1330         }
1331     }
1332     else
1333     {
1334         return VLC_FALSE;
1335     }
1336 }