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  * Copyright © 2007 Rémi Denis-Courmont
   6  * $Id$
   7  *
   8  * Authors: Clément Stenac <zorglub@videolan.org>
   9  *          Rémi Denis-Courmont
  10  *
  11  * This program is free software; you can redistribute it and/or modify
  12  * it under the terms of the GNU General Public License as published by
  13  * the Free Software Foundation; either version 2 of the License, or
  14  * (at your option) any later version.
  15  *
  16  * This program is distributed in the hope that it will be useful,
  17  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  18  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  19  * GNU General Public License for more details.
  20  *
  21  * You should have received a copy of the GNU General Public License
  22  * along with this program; if not, write to the Free Software
  23  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
  24  *****************************************************************************/
  25
  26 /*****************************************************************************
  27  * Includes
  28  *****************************************************************************/
  29 #ifdef HAVE_CONFIG_H
  30 # include "config.h"
  31 #endif
  32
  33 #include <vlc_common.h>
  34 #include <vlc_plugin.h>
  35 #include <assert.h>
  36
  37 #include <vlc_demux.h>
  38 #include <vlc_services_discovery.h>
  39
  40 #include <vlc_network.h>
  41 #include <vlc_charset.h>
  42
  43 #include <ctype.h>
  44 #include <errno.h>
  45
  46 #ifdef HAVE_UNISTD_H
  47 #    include <unistd.h>
  48 #endif
  49 #ifdef HAVE_SYS_TIME_H
  50 #    include <sys/time.h>
  51 #endif
  52 #ifdef HAVE_POLL
  53 # include <poll.h>
  54 #endif
  55
  56 #ifdef HAVE_ZLIB_H
  57 #   include <zlib.h>
  58 #endif
  59
  60 #ifndef WIN32
  61 #   include <net/if.h>
  62 #endif
  63
  64 /************************************************************************
  65  * Macros and definitions
  66  ************************************************************************/
  67
  68 #define MAX_LINE_LENGTH 256
  69
  70 /* SAP is always on that port */
  71 #define SAP_PORT 9875
  72 /* Global-scope SAP address */
  73 #define SAP_V4_GLOBAL_ADDRESS   "224.2.127.254"
  74 /* Organization-local SAP address */
  75 #define SAP_V4_ORG_ADDRESS      "239.195.255.255"
  76 /* Local (smallest non-link-local scope) SAP address */
  77 #define SAP_V4_LOCAL_ADDRESS    "239.255.255.255"
  78 /* Link-local SAP address */
  79 #define SAP_V4_LINK_ADDRESS     "224.0.0.255"
  80 #define ADD_SESSION 1
  81
  82 /*****************************************************************************
  83  * Module descriptor
  84  *****************************************************************************/
  85 #define SAP_ADDR_TEXT N_( "SAP multicast address" )
  86 #define SAP_ADDR_LONGTEXT N_( "The SAP module normally chooses itself the " \
  87                               "right addresses to listen to. However, you " \
  88                               "can specify a specific address." )
  89 #define SAP_IPV4_TEXT N_( "IPv4 SAP" )
  90 #define SAP_IPV4_LONGTEXT N_( \
  91       "Listen to IPv4 announcements on the standard addresses." )
  92 #define SAP_IPV6_TEXT N_( "IPv6 SAP" )
  93 #define SAP_IPV6_LONGTEXT N_( \
  94       "Listen to IPv6 announcements on the standard addresses." )
  95 #define SAP_SCOPE_TEXT N_( "IPv6 SAP scope" )
  96 #define SAP_SCOPE_LONGTEXT N_( \
  97        "Scope for IPv6 announcements (default is 8)." )
  98 #define SAP_TIMEOUT_TEXT N_( "SAP timeout (seconds)" )
  99 #define SAP_TIMEOUT_LONGTEXT N_( \
 100        "Delay after which SAP items get deleted if no new announcement " \
 101        "is received." )
 102 #define SAP_PARSE_TEXT N_( "Try to parse the announce" )
 103 #define SAP_PARSE_LONGTEXT N_( \
 104        "This enables actual parsing of the announces by the SAP module. " \
 105        "Otherwise, all announcements are parsed by the \"live555\" " \
 106        "(RTP/RTSP) module." )
 107 #define SAP_STRICT_TEXT N_( "SAP Strict mode" )
 108 #define SAP_STRICT_LONGTEXT N_( \
 109        "When this is set, the SAP parser will discard some non-compliant " \
 110        "announcements." )
 111 #define SAP_CACHE_TEXT N_("Use SAP cache")
 112 #define SAP_CACHE_LONGTEXT N_( \
 113        "This enables a SAP caching mechanism. " \
 114        "This will result in lower SAP startup time, but you could end up " \
 115        "with items corresponding to legacy streams." )
 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( N_("SAP"))
 125     set_description( N_("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, true )
 131     add_bool( "sap-ipv4", 1 , NULL,
 132                SAP_IPV4_TEXT,SAP_IPV4_LONGTEXT, true )
 133     add_bool( "sap-ipv6", 1 , NULL,
 134               SAP_IPV6_TEXT, SAP_IPV6_LONGTEXT, true )
 135     add_integer( "sap-timeout", 1800, NULL,
 136                  SAP_TIMEOUT_TEXT, SAP_TIMEOUT_LONGTEXT, true )
 137     add_bool( "sap-parse", 1 , NULL,
 138                SAP_PARSE_TEXT,SAP_PARSE_LONGTEXT, true )
 139     add_bool( "sap-strict", 0 , NULL,
 140                SAP_STRICT_TEXT,SAP_STRICT_LONGTEXT, true )
 141 #if 0
 142     add_bool( "sap-cache", 0 , NULL,
 143                SAP_CACHE_TEXT,SAP_CACHE_LONGTEXT, true )
 144 #endif
 145     add_obsolete_bool( "sap-timeshift" ) /* Redumdant since 1.0.0 */
 146
 147     set_capability( "services_discovery", 0 )
 148     set_callbacks( Open, Close )
 149
 150     add_submodule ()
 151         set_description( N_("SDP Descriptions parser") )
 152         add_shortcut( "sdp" )
 153         set_capability( "demux", 51 )
 154         set_callbacks( OpenDemux, CloseDemux )
 155 vlc_module_end ()
 156
 157
 158 /*****************************************************************************
 159  * Local structures
 160  *****************************************************************************/
 161
 162 typedef struct sdp_t sdp_t;
 163 typedef struct attribute_t attribute_t;
 164 typedef struct sap_announce_t sap_announce_t;
 165
 166
 167 struct sdp_media_t
 168 {
 169     struct sdp_t           *parent;
 170     char                   *fmt;
 171     struct sockaddr_storage addr;
 172     socklen_t               addrlen;
 173     unsigned                n_addr;
 174     int           i_attributes;
 175     attribute_t  **pp_attributes;
 176 };
 177
 178
 179 /* The structure that contains sdp information */
 180 struct  sdp_t
 181 {
 182     const char *psz_sdp;
 183
 184     /* o field */
 185     char     username[64];
 186     uint64_t session_id;
 187     uint64_t session_version;
 188     unsigned orig_ip_version;
 189     char     orig_host[1024];
 190
 191     /* s= field */
 192     char *psz_sessionname;
 193
 194     /* old cruft */
 195     /* "computed" URI */
 196     char *psz_uri;
 197     int           i_media_type;
 198     unsigned rtcp_port;
 199
 200     /* a= global attributes */
 201     int           i_attributes;
 202     attribute_t  **pp_attributes;
 203
 204     /* medias (well, we only support one atm) */
 205     unsigned            mediac;
 206     struct sdp_media_t *mediav;
 207 };
 208
 209 struct attribute_t
 210 {
 211     const char *value;
 212     char name[];
 213 };
 214
 215 struct sap_announce_t
 216 {
 217     mtime_t i_last;
 218     mtime_t i_period;
 219     uint8_t i_period_trust;
 220
 221     uint16_t    i_hash;
 222     uint32_t    i_source[4];
 223
 224     /* SAP annnounces must only contain one SDP */
 225     sdp_t       *p_sdp;
 226
 227     input_item_t * p_item;
 228 };
 229
 230 struct services_discovery_sys_t
 231 {
 232     vlc_thread_t thread;
 233
 234     /* Socket descriptors */
 235     int i_fd;
 236     int *pi_fd;
 237
 238     /* Table of announces */
 239     int i_announces;
 240     struct sap_announce_t **pp_announces;
 241
 242     /* Modes */
 243     bool  b_strict;
 244     bool  b_parse;
 245
 246     int i_timeout;
 247 };
 248
 249 struct demux_sys_t
 250 {
 251     sdp_t *p_sdp;
 252 };
 253
 254 /*****************************************************************************
 255  * Local prototypes
 256  *****************************************************************************/
 257
 258
 259 /* Main functions */
 260     static int Demux( demux_t *p_demux );
 261     static int Control( demux_t *, int, va_list );
 262     static void *Run  ( void *p_sd );
 263
 264 /* Main parsing functions */
 265     static int ParseConnection( vlc_object_t *p_obj, sdp_t *p_sdp );
 266     static int ParseSAP( services_discovery_t *p_sd, const uint8_t *p_buffer, size_t i_read );
 267     static sdp_t *ParseSDP (vlc_object_t *p_sd, const char *psz_sdp);
 268     static sap_announce_t *CreateAnnounce( services_discovery_t *, uint16_t, sdp_t * );
 269     static int RemoveAnnounce( services_discovery_t *p_sd, sap_announce_t *p_announce );
 270
 271 /* Helper functions */
 272     static inline attribute_t *MakeAttribute (const char *str);
 273     static const char *GetAttribute (attribute_t **tab, unsigned n, const char *name);
 274     static inline void FreeAttribute (attribute_t *a);
 275     static const char *FindAttribute (const sdp_t *sdp, unsigned media,
 276                                       const char *name);
 277
 278     static bool IsSameSession( sdp_t *p_sdp1, sdp_t *p_sdp2 );
 279     static int InitSocket( services_discovery_t *p_sd, const char *psz_address, int i_port );
 280     static int Decompress( const unsigned char *psz_src, unsigned char **_dst, int i_len );
 281     static void FreeSDP( sdp_t *p_sdp );
 282
 283 static inline int min_int( int a, int b )
 284 {
 285     return a > b ? b : a;
 286 }
 287
 288 static bool IsWellKnownPayload (int type)
 289 {
 290     switch (type)
 291     {   /* Should be in sync with modules/demux/rtp.c */
 292         case  0: /* PCMU/8000 */
 293         case  3:
 294         case  8: /* PCMA/8000 */
 295         case 10: /* L16/44100/2 */
 296         case 11: /* L16/44100 */
 297         case 12:
 298         case 14: /* MPA/90000 */
 299         case 32: /* MPV/90000 */
 300         case 33: /* MP2/90000 */
 301             return true;
 302    }
 303    return false;
 304 }
 305
 306 /*****************************************************************************
 307  * Open: initialize and create stuff
 308  *****************************************************************************/
 309 static int Open( vlc_object_t *p_this )
 310 {
 311     services_discovery_t *p_sd = ( services_discovery_t* )p_this;
 312     services_discovery_sys_t *p_sys  = (services_discovery_sys_t *)
 313                                 malloc( sizeof( services_discovery_sys_t ) );
 314     if( !p_sys )
 315         return VLC_ENOMEM;
 316
 317     p_sys->i_timeout = var_CreateGetInteger( p_sd, "sap-timeout" );
 318
 319     p_sd->p_sys  = p_sys;
 320
 321     p_sys->pi_fd = NULL;
 322     p_sys->i_fd = 0;
 323
 324     p_sys->b_strict = var_CreateGetInteger( p_sd, "sap-strict");
 325     p_sys->b_parse = var_CreateGetInteger( p_sd, "sap-parse" );
 326
 327 #if 0
 328     if( var_CreateGetInteger( p_sd, "sap-cache" ) )
 329     {
 330         CacheLoad( p_sd );
 331     }
 332 #endif
 333
 334     p_sys->i_announces = 0;
 335     p_sys->pp_announces = NULL;
 336     /* TODO: create sockets here, and fix racy sockets table */
 337     if (vlc_clone (&p_sys->thread, Run, p_sd, VLC_THREAD_PRIORITY_LOW))
 338     {
 339         free (p_sys);
 340         return VLC_EGENERIC;
 341     }
 342
 343     return VLC_SUCCESS;
 344 }
 345
 346 /*****************************************************************************
 347  * OpenDemux: initialize and create stuff
 348  *****************************************************************************/
 349 static int OpenDemux( vlc_object_t *p_this )
 350 {
 351     demux_t *p_demux = (demux_t *)p_this;
 352     const uint8_t *p_peek;
 353     char *psz_sdp = NULL;
 354     sdp_t *p_sdp = NULL;
 355     int errval = VLC_EGENERIC;
 356     size_t i_len;
 357
 358     if( !var_CreateGetInteger( p_demux, "sap-parse" ) )
 359     {
 360         /* We want livedotcom module to parse this SDP file */
 361         return VLC_EGENERIC;
 362     }
 363
 364     assert( p_demux->s ); /* this is NOT an access_demux */
 365
 366     /* Probe for SDP */
 367     if( stream_Peek( p_demux->s, &p_peek, 7 ) < 7 )
 368         return VLC_EGENERIC;
 369
 370     if( memcmp( p_peek, "v=0\r\no=", 7 ) && memcmp( p_peek, "v=0\no=", 6 ) )
 371         return VLC_EGENERIC;
 372
 373     /* Gather the complete sdp file */
 374     for( i_len = 0, psz_sdp = NULL; i_len < 65536; )
 375     {
 376         const int i_read_max = 1024;
 377         char *psz_sdp_new = realloc( psz_sdp, i_len + i_read_max );
 378         size_t i_read;
 379         if( psz_sdp_new == NULL )
 380         {
 381             errval = VLC_ENOMEM;
 382             goto error;
 383         }
 384         psz_sdp = psz_sdp_new;
 385
 386         i_read = stream_Read( p_demux->s, &psz_sdp[i_len], i_read_max );
 387         if( (int)i_read < 0 )
 388         {
 389             msg_Err( p_demux, "cannot read SDP" );
 390             goto error;
 391         }
 392         i_len += i_read;
 393
 394         psz_sdp[i_len] = '\0';
 395
 396         if( (int)i_read < i_read_max )
 397             break; // EOF
 398     }
 399
 400     p_sdp = ParseSDP( VLC_OBJECT(p_demux), psz_sdp );
 401
 402     if( !p_sdp )
 403     {
 404         msg_Warn( p_demux, "invalid SDP");
 405         goto error;
 406     }
 407
 408     if( ParseConnection( VLC_OBJECT( p_demux ), p_sdp ) )
 409     {
 410         p_sdp->psz_uri = NULL;
 411     }
 412     if (!IsWellKnownPayload (p_sdp->i_media_type))
 413         goto error;
 414     if( p_sdp->psz_uri == NULL ) goto error;
 415
 416     p_demux->p_sys = (demux_sys_t *)malloc( sizeof(demux_sys_t) );
 417     p_demux->p_sys->p_sdp = p_sdp;
 418     p_demux->pf_control = Control;
 419     p_demux->pf_demux = Demux;
 420
 421     FREENULL( psz_sdp );
 422     return VLC_SUCCESS;
 423
 424 error:
 425     FREENULL( psz_sdp );
 426     if( p_sdp ) FreeSDP( p_sdp ); p_sdp = NULL;
 427     stream_Seek( p_demux->s, 0 );
 428     return errval;
 429 }
 430
 431 /*****************************************************************************
 432  * Close:
 433  *****************************************************************************/
 434 static void Close( vlc_object_t *p_this )
 435 {
 436     services_discovery_t *p_sd = ( services_discovery_t* )p_this;
 437     services_discovery_sys_t    *p_sys  = p_sd->p_sys;
 438     int i;
 439
 440     vlc_cancel (p_sys->thread);
 441     vlc_join (p_sys->thread, NULL);
 442
 443     for( i = p_sys->i_fd-1 ; i >= 0 ; i-- )
 444     {
 445         net_Close( p_sys->pi_fd[i] );
 446     }
 447     FREENULL( p_sys->pi_fd );
 448
 449 #if 0
 450     if( config_GetInt( p_sd, "sap-cache" ) )
 451     {
 452         CacheSave( p_sd );
 453     }
 454 #endif
 455
 456     for( i = p_sys->i_announces  - 1;  i>= 0; i-- )
 457     {
 458         RemoveAnnounce( p_sd, p_sys->pp_announces[i] );
 459     }
 460     FREENULL( p_sys->pp_announces );
 461
 462     free( p_sys );
 463 }
 464
 465 /*****************************************************************************
 466  * CloseDemux: Close the demuxer
 467  *****************************************************************************/
 468 static void CloseDemux( vlc_object_t *p_this )
 469 {
 470     demux_t *p_demux = (demux_t *)p_this;
 471     if( p_demux->p_sys )
 472     {
 473         if( p_demux->p_sys->p_sdp ) { FreeSDP( p_demux->p_sys->p_sdp ); p_demux->p_sys->p_sdp = NULL; }
 474         free( p_demux->p_sys );
 475     }
 476 }
 477
 478 /*****************************************************************************
 479  * Run: main SAP thread
 480  *****************************************************************************
 481  * Listens to SAP packets, and sends them to packet_handle
 482  *****************************************************************************/
 483 #define MAX_SAP_BUFFER 5000
 484
 485 static void *Run( void *data )
 486 {
 487     services_discovery_t *p_sd = data;
 488     char *psz_addr;
 489     int i;
 490     int timeout = -1;
 491     int canc = vlc_savecancel ();
 492
 493     /* Braindead Winsock DNS resolver will get stuck over 2 seconds per failed
 494      * DNS queries, even if the DNS server returns an error with milliseconds.
 495      * You don't want to know why the bug (as of XP SP2) wasn't fixed since
 496      * Winsock 1.1 from Windows 95, if not Windows 3.1.
 497      * Anyway, to avoid a 30 seconds delay for failed IPv6 socket creation,
 498      * we have to open sockets in Run() rather than Open(). */
 499     if( var_CreateGetInteger( p_sd, "sap-ipv4" ) )
 500     {
 501         InitSocket( p_sd, SAP_V4_GLOBAL_ADDRESS, SAP_PORT );
 502         InitSocket( p_sd, SAP_V4_ORG_ADDRESS, SAP_PORT );
 503         InitSocket( p_sd, SAP_V4_LOCAL_ADDRESS, SAP_PORT );
 504         InitSocket( p_sd, SAP_V4_LINK_ADDRESS, SAP_PORT );
 505     }
 506     if( var_CreateGetInteger( p_sd, "sap-ipv6" ) )
 507     {
 508         char psz_address[NI_MAXNUMERICHOST] = "ff02::2:7ffe%";
 509
 510 #ifndef WIN32
 511         struct if_nameindex *l = if_nameindex ();
 512         if (l != NULL)
 513         {
 514             char *ptr = strchr (psz_address, '%') + 1;
 515             for (unsigned i = 0; l[i].if_index; i++)
 516             {
 517                 strcpy (ptr, l[i].if_name);
 518                 InitSocket (p_sd, psz_address, SAP_PORT);
 519             }
 520             if_freenameindex (l);
 521         }
 522 #else
 523         /* this is the Winsock2 equivalant of SIOCGIFCONF on BSD stacks,
 524            which if_nameindex uses internally anyway */
 525
 526         // first create a dummy socket to pin down the protocol family
 527         SOCKET s = socket(PF_INET6, SOCK_DGRAM, IPPROTO_UDP);
 528         if( s != INVALID_SOCKET )
 529         {
 530             INTERFACE_INFO ifaces[10]; // Assume there will be no more than 10 IP interfaces
 531             size_t len = sizeof(ifaces);
 532
 533             if( SOCKET_ERROR != WSAIoctl(s, SIO_GET_INTERFACE_LIST, NULL, 0, &ifaces, len, &len, NULL, NULL) )
 534             {
 535                 unsigned ifcount = len/sizeof(INTERFACE_INFO);
 536                 char *ptr = strchr (psz_address, '%') + 1;
 537                 for(unsigned i = 1; i<=ifcount; ++i )
 538                 {
 539                     // append link-local zone identifier
 540                     sprintf(ptr, "%d", i);
 541                 }
 542             }
 543             closesocket(s);
 544         }
 545 #endif
 546         *strchr (psz_address, '%') = '\0';
 547
 548         static const char ipv6_scopes[] = "1456789ABCDE";
 549         for (const char *c_scope = ipv6_scopes; *c_scope; c_scope++)
 550         {
 551             psz_address[3] = *c_scope;
 552             InitSocket( p_sd, psz_address, SAP_PORT );
 553         }
 554     }
 555
 556     psz_addr = var_CreateGetString( p_sd, "sap-addr" );
 557     if( psz_addr && *psz_addr )
 558         InitSocket( p_sd, psz_addr, SAP_PORT );
 559     free( psz_addr );
 560
 561     if( p_sd->p_sys->i_fd == 0 )
 562     {
 563         msg_Err( p_sd, "unable to listen on any address" );
 564         return NULL;
 565     }
 566
 567     /* read SAP packets */
 568     for (;;)
 569     {
 570         vlc_restorecancel (canc);
 571         unsigned n = p_sd->p_sys->i_fd;
 572         struct pollfd ufd[n];
 573
 574         for (unsigned i = 0; i < n; i++)
 575         {
 576             ufd[i].fd = p_sd->p_sys->pi_fd[i];
 577             ufd[i].events = POLLIN;
 578             ufd[i].revents = 0;
 579         }
 580
 581         int val = poll (ufd, n, timeout);
 582         canc = vlc_savecancel ();
 583         if (val > 0)
 584         {
 585             for (unsigned i = 0; i < n; i++)
 586             {
 587                 if (ufd[i].revents)
 588                 {
 589                     uint8_t p_buffer[MAX_SAP_BUFFER+1];
 590                     ssize_t i_read;
 591
 592                     i_read = net_Read (p_sd, ufd[i].fd, NULL, p_buffer,
 593                                        MAX_SAP_BUFFER, false);
 594                     if (i_read < 0)
 595                         msg_Warn (p_sd, "receive error: %m");
 596                     if (i_read > 6)
 597                     {
 598                         /* Parse the packet */
 599                         p_buffer[i_read] = '\0';
 600                         ParseSAP (p_sd, p_buffer, i_read);
 601                     }
 602                 }
 603             }
 604         }
 605
 606         mtime_t now = mdate();
 607
 608         /* A 1 hour timeout correspong to the RFC Implicit timeout.
 609          * This timeout is tuned in the following loop. */
 610         timeout = 1000 * 60 * 60;
 611
 612         /* Check for items that need deletion */
 613         for( i = 0; i < p_sd->p_sys->i_announces; i++ )
 614         {
 615             mtime_t i_timeout = ( mtime_t ) 1000000 * p_sd->p_sys->i_timeout;
 616             sap_announce_t * p_announce = p_sd->p_sys->pp_announces[i];
 617             mtime_t i_last_period = now - p_announce->i_last;
 618
 619             /* Remove the annoucement, if the last announcement was 1 hour ago
 620              * or if the last packet emitted was 3 times the average time
 621              * between two packets */
 622             if( ( p_announce->i_period_trust > 5 && i_last_period > 3 * p_announce->i_period ) ||
 623                 i_last_period > i_timeout )
 624             {
 625                 RemoveAnnounce( p_sd, p_announce );
 626             }
 627             else
 628             {
 629                 /* Compute next timeout */
 630                 if( p_announce->i_period_trust > 5 )
 631                     timeout = min_int((3 * p_announce->i_period - i_last_period) / 1000, timeout);
 632                 timeout = min_int((i_timeout - i_last_period)/1000, timeout);
 633             }
 634         }
 635
 636         if( !p_sd->p_sys->i_announces )
 637             timeout = -1; /* We can safely poll indefinitly. */
 638         else if( timeout < 200 )
 639             timeout = 200; /* Don't wakeup too fast. */
 640     }
 641     assert (0);
 642 }
 643
 644 /**********************************************************************
 645  * Demux: reads and demuxes data packets
 646  * Return -1 if error, 0 if EOF, 1 else
 647  **********************************************************************/
 648 static int Demux( demux_t *p_demux )
 649 {
 650     sdp_t *p_sdp = p_demux->p_sys->p_sdp;
 651     input_thread_t *p_input;
 652     input_item_t *p_parent_input;
 653
 654     p_input = (input_thread_t *)vlc_object_find( p_demux, VLC_OBJECT_INPUT,
 655                                                  FIND_PARENT );
 656     assert( p_input );
 657     if( !p_input )
 658     {
 659         msg_Err( p_demux, "parent input could not be found" );
 660         return VLC_EGENERIC;
 661     }
 662
 663     /* This item hasn't been held by input_GetItem
 664      * don't release it */
 665     p_parent_input = input_GetItem( p_input );
 666
 667     input_item_SetURI( p_parent_input, p_sdp->psz_uri );
 668     input_item_SetName( p_parent_input, p_sdp->psz_sessionname );
 669     if( p_sdp->rtcp_port )
 670     {
 671         char *rtcp;
 672         if( asprintf( &rtcp, ":rtcp-port=%u", p_sdp->rtcp_port ) != -1 )
 673         {
 674             input_item_AddOption( p_parent_input, rtcp, VLC_INPUT_OPTION_TRUSTED );
 675             free( rtcp );
 676         }
 677     }
 678
 679     vlc_mutex_lock( &p_parent_input->lock );
 680
 681     p_parent_input->i_type = ITEM_TYPE_NET;
 682
 683     vlc_mutex_unlock( &p_parent_input->lock );
 684     return VLC_SUCCESS;
 685 }
 686
 687 static int Control( demux_t *p_demux, int i_query, va_list args )
 688 {
 689     VLC_UNUSED(p_demux); VLC_UNUSED(i_query); VLC_UNUSED(args);
 690     return VLC_EGENERIC;
 691 }
 692
 693 /**************************************************************
 694  * Local functions
 695  **************************************************************/
 696
 697 /* i_read is at least > 6 */
 698 static int ParseSAP( services_discovery_t *p_sd, const uint8_t *buf,
 699                      size_t len )
 700 {
 701     int i;
 702     const char          *psz_sdp;
 703     const uint8_t *end = buf + len;
 704     sdp_t               *p_sdp;
 705
 706     assert (buf[len] == '\0');
 707
 708     if (len < 4)
 709         return VLC_EGENERIC;
 710
 711     uint8_t flags = buf[0];
 712
 713     /* First, check the sap announce is correct */
 714     if ((flags >> 5) != 1)
 715         return VLC_EGENERIC;
 716
 717     bool b_ipv6 = (flags & 0x10) != 0;
 718     bool b_need_delete = (flags & 0x04) != 0;
 719
 720     if (flags & 0x02)
 721     {
 722         msg_Dbg( p_sd, "encrypted packet, unsupported" );
 723         return VLC_EGENERIC;
 724     }
 725
 726     bool b_compressed = (flags & 0x01) != 0;
 727
 728     uint16_t i_hash = U16_AT (buf + 2);
 729
 730     if( p_sd->p_sys->b_strict && i_hash == 0 )
 731     {
 732         msg_Dbg( p_sd, "strict mode, discarding announce with null id hash");
 733         return VLC_EGENERIC;
 734     }
 735
 736     // Skips source address and auth data
 737     buf += 4 + (b_ipv6 ? 16 : 4) + buf[1];
 738     if (buf > end)
 739         return VLC_EGENERIC;
 740
 741     uint8_t *decomp = NULL;
 742     if( b_compressed )
 743     {
 744         int newsize = Decompress (buf, &decomp, end - buf);
 745         if (newsize < 0)
 746         {
 747             msg_Dbg( p_sd, "decompression of SAP packet failed" );
 748             return VLC_EGENERIC;
 749         }
 750
 751         decomp = realloc (decomp, newsize + 1);
 752         decomp[newsize] = '\0';
 753         psz_sdp = (const char *)decomp;
 754         len = newsize;
 755     }
 756     else
 757     {
 758         psz_sdp = (const char *)buf;
 759         len = end - buf;
 760     }
 761
 762     /* len is a strlen here here. both buf and decomp are len+1 where the 1 should be a \0 */
 763     assert( psz_sdp[len] == '\0');
 764
 765     /* Skip payload type */
 766     /* SAPv1 has implicit "application/sdp" payload type: first line is v=0 */
 767     if (strncmp (psz_sdp, "v=0", 3))
 768     {
 769         size_t clen = strlen (psz_sdp) + 1;
 770
 771         if (strcmp (psz_sdp, "application/sdp"))
 772         {
 773             msg_Dbg (p_sd, "unsupported content type: %s", psz_sdp);
 774             return VLC_EGENERIC;
 775         }
 776
 777         // skips content type
 778         if (len <= clen)
 779             return VLC_EGENERIC;
 780
 781         len -= clen;
 782         psz_sdp += clen;
 783     }
 784
 785     /* Parse SDP info */
 786     p_sdp = ParseSDP( VLC_OBJECT(p_sd), psz_sdp );
 787
 788     if( p_sdp == NULL )
 789         return VLC_EGENERIC;
 790
 791     p_sdp->psz_sdp = psz_sdp;
 792
 793     /* Decide whether we should add a playlist item for this SDP */
 794     /* Parse connection information (c= & m= ) */
 795     if( ParseConnection( VLC_OBJECT(p_sd), p_sdp ) )
 796         p_sdp->psz_uri = NULL;
 797
 798     /* Multi-media or no-parse -> pass to LIVE.COM */
 799     if( !IsWellKnownPayload( p_sdp->i_media_type ) || !p_sd->p_sys->b_parse )
 800     {
 801         free( p_sdp->psz_uri );
 802         if (asprintf( &p_sdp->psz_uri, "sdp://%s", p_sdp->psz_sdp ) == -1)
 803             p_sdp->psz_uri = NULL;
 804     }
 805
 806     if( p_sdp->psz_uri == NULL )
 807     {
 808         FreeSDP( p_sdp );
 809         return VLC_EGENERIC;
 810     }
 811
 812     for( i = 0 ; i< p_sd->p_sys->i_announces ; i++ )
 813     {
 814         sap_announce_t * p_announce = p_sd->p_sys->pp_announces[i];
 815         /* FIXME: slow */
 816         /* FIXME: we create a new announce each time the sdp changes */
 817         if( IsSameSession( p_announce->p_sdp, p_sdp ) )
 818         {
 819             /* We don't support delete announcement as they can easily
 820              * Be used to highjack an announcement by a third party.
 821              * Intead we cleverly implement Implicit Announcement removal.
 822              *
 823              * if( b_need_delete )
 824              *    RemoveAnnounce( p_sd, p_sd->p_sys->pp_announces[i]);
 825              * else
 826              */
 827
 828             if( !b_need_delete )
 829             {
 830                 /* No need to go after six, as we start to trust the
 831                  * average period at six */
 832                 if( p_announce->i_period_trust <= 5 )
 833                     p_announce->i_period_trust++;
 834
 835                 /* Compute the average period */
 836                 mtime_t now = mdate();
 837                 p_announce->i_period = (p_announce->i_period + (now - p_announce->i_last)) / 2;
 838                 p_announce->i_last = now;
 839             }
 840             FreeSDP( p_sdp ); p_sdp = NULL;
 841             return VLC_SUCCESS;
 842         }
 843     }
 844
 845     CreateAnnounce( p_sd, i_hash, p_sdp );
 846
 847     FREENULL (decomp);
 848     return VLC_SUCCESS;
 849 }
 850
 851 sap_announce_t *CreateAnnounce( services_discovery_t *p_sd, uint16_t i_hash,
 852                                 sdp_t *p_sdp )
 853 {
 854     input_item_t *p_input;
 855     const char *psz_value;
 856     sap_announce_t *p_sap = (sap_announce_t *)malloc(
 857                                         sizeof(sap_announce_t ) );
 858     services_discovery_sys_t *p_sys;
 859     if( p_sap == NULL )
 860         return NULL;
 861
 862     p_sys = p_sd->p_sys;
 863
 864     p_sap->i_last = mdate();
 865     p_sap->i_period = 0;
 866     p_sap->i_period_trust = 0;
 867     p_sap->i_hash = i_hash;
 868     p_sap->p_sdp = p_sdp;
 869
 870     /* Released in RemoveAnnounce */
 871     p_input = input_item_NewWithType( VLC_OBJECT(p_sd),
 872                                      p_sap->p_sdp->psz_uri,
 873                                      p_sdp->psz_sessionname,
 874                                      0, NULL, 0, -1, ITEM_TYPE_NET );
 875     p_sap->p_item = p_input;
 876     if( !p_input )
 877     {
 878         free( p_sap );
 879         return NULL;
 880     }
 881
 882     if( p_sdp->rtcp_port )
 883     {
 884         char *rtcp;
 885         if( asprintf( &rtcp, ":rtcp-port=%u", p_sdp->rtcp_port ) != -1 )
 886         {
 887             input_item_AddOption( p_input, rtcp, VLC_INPUT_OPTION_TRUSTED );
 888             free( rtcp );
 889         }
 890     }
 891
 892     psz_value = GetAttribute( p_sap->p_sdp->pp_attributes, p_sap->p_sdp->i_attributes, "tool" );
 893     if( psz_value != NULL )
 894     {
 895         input_item_AddInfo( p_input, _("Session"), _("Tool"), "%s", psz_value );
 896     }
 897     if( strcmp( p_sdp->username, "-" ) )
 898     {
 899         input_item_AddInfo( p_input, _("Session"), _("User"), "%s",
 900                            p_sdp->username );
 901     }
 902
 903     /* Handle group */
 904     if (p_sap->p_sdp->mediac >= 1)
 905         psz_value = FindAttribute (p_sap->p_sdp, 0, "x-plgroup");
 906     else
 907         psz_value = GetAttribute( p_sap->p_sdp->pp_attributes, p_sap->p_sdp->i_attributes, "x-plgroup" );
 908
 909     services_discovery_AddItem( p_sd, p_input, psz_value /* category name */ );
 910
 911     TAB_APPEND( p_sys->i_announces, p_sys->pp_announces, p_sap );
 912
 913     return p_sap;
 914 }
 915
 916
 917 static const char *FindAttribute (const sdp_t *sdp, unsigned media,
 918                                   const char *name)
 919 {
 920     /* Look for media attribute, and fallback to session */
 921     const char *attr = GetAttribute (sdp->mediav[media].pp_attributes,
 922                                      sdp->mediav[media].i_attributes, name);
 923     if (attr == NULL)
 924         attr = GetAttribute (sdp->pp_attributes, sdp->i_attributes, name);
 925     return attr;
 926 }
 927
 928
 929 /* Fill p_sdp->psz_uri */
 930 static int ParseConnection( vlc_object_t *p_obj, sdp_t *p_sdp )
 931 {
 932     if (p_sdp->mediac == 0)
 933     {
 934         msg_Dbg (p_obj, "Ignoring SDP with no media");
 935         return VLC_EGENERIC;
 936     }
 937
 938     for (unsigned i = 1; i < p_sdp->mediac; i++)
 939     {
 940         if ((p_sdp->mediav[i].n_addr != p_sdp->mediav->n_addr)
 941          || (p_sdp->mediav[i].addrlen != p_sdp->mediav->addrlen)
 942          || memcmp (&p_sdp->mediav[i].addr, &p_sdp->mediav->addr,
 943                     p_sdp->mediav->addrlen))
 944         {
 945             msg_Dbg (p_obj, "Multiple media ports not supported -> live555");
 946             return VLC_EGENERIC;
 947         }
 948     }
 949
 950     if (p_sdp->mediav->n_addr != 1)
 951     {
 952         msg_Dbg (p_obj, "Layered encoding not supported -> live555");
 953         return VLC_EGENERIC;
 954     }
 955
 956     char psz_uri[1026];
 957     const char *host;
 958     int port;
 959
 960     psz_uri[0] = '[';
 961     if (vlc_getnameinfo ((struct sockaddr *)&(p_sdp->mediav->addr),
 962                          p_sdp->mediav->addrlen, psz_uri + 1,
 963                          sizeof (psz_uri) - 2, &port, NI_NUMERICHOST))
 964         return VLC_EGENERIC;
 965
 966     if (strchr (psz_uri + 1, ':'))
 967     {
 968         host = psz_uri;
 969         strcat (psz_uri, "]");
 970     }
 971     else
 972         host = psz_uri + 1;
 973
 974     /* Parse m= field */
 975     char *sdp_proto = strdup (p_sdp->mediav[0].fmt);
 976     if (sdp_proto == NULL)
 977         return VLC_ENOMEM;
 978
 979     char *subtype = strchr (sdp_proto, ' ');
 980     if (subtype == NULL)
 981     {
 982         msg_Dbg (p_obj, "missing SDP media subtype: %s", sdp_proto);
 983         free (sdp_proto);
 984         return VLC_EGENERIC;
 985     }
 986     else
 987     {
 988         *subtype++ = '\0';
 989         /* FIXME: check for multiple payload types in RTP/AVP case.
 990          * FIXME: check for "mpeg" subtype in raw udp case. */
 991         if (!strcasecmp (sdp_proto, "udp"))
 992             p_sdp->i_media_type = 33;
 993         else
 994             p_sdp->i_media_type = atoi (subtype);
 995     }
 996
 997     /* RTP protocol, nul, VLC shortcut, nul, flags byte as follow:
 998      * 0x1: Connection-Oriented media. */
 999     static const char proto_match[] =
1000         "udp\0"             "udp\0\0"
1001         "RTP/AVP\0"         "rtp\0\0"
1002         "UDPLite/RTP/AVP\0" "udplite\0\0"
1003         "DCCP/RTP/AVP\0"    "dccp\0\1"
1004         "TCP/RTP/AVP\0"     "rtptcp\0\1"
1005         "\0";
1006
1007     const char *vlc_proto = NULL;
1008     uint8_t flags = 0;
1009     for (const char *proto = proto_match; *proto;)
1010     {
1011         if (strcasecmp (proto, sdp_proto) == 0)
1012         {
1013             vlc_proto = proto + strlen (proto) + 1;
1014             flags = vlc_proto[strlen (vlc_proto) + 1];
1015             break;
1016         }
1017         proto += strlen (proto) + 1;
1018         proto += strlen (proto) + 2;
1019     }
1020
1021     free (sdp_proto);
1022     if (vlc_proto == NULL)
1023     {
1024         msg_Dbg (p_obj, "unknown SDP media protocol: %s",
1025                  p_sdp->mediav[0].fmt);
1026         return VLC_EGENERIC;
1027     }
1028
1029     if (!strcmp (vlc_proto, "udp") || FindAttribute (p_sdp, 0, "rtcp-mux"))
1030         p_sdp->rtcp_port = 0;
1031     else
1032     {
1033         const char *rtcp = FindAttribute (p_sdp, 0, "rtcp");
1034         if (rtcp)
1035             p_sdp->rtcp_port = atoi (rtcp);
1036         else
1037         if (port & 1) /* odd port -> RTCP; next even port -> RTP */
1038             p_sdp->rtcp_port = port++;
1039         else /* even port -> RTP; next odd port -> RTCP */
1040             p_sdp->rtcp_port = port + 1;
1041     }
1042
1043     if (flags & 1)
1044     {
1045         /* Connection-oriented media */
1046         const char *setup = FindAttribute (p_sdp, 0, "setup");
1047         if (setup == NULL)
1048             setup = "active"; /* default value */
1049
1050         if (strcmp (setup, "actpass") && strcmp (setup, "passive"))
1051         {
1052             msg_Dbg (p_obj, "unsupported COMEDIA mode: %s", setup);
1053             return VLC_EGENERIC;
1054         }
1055
1056         if (asprintf (&p_sdp->psz_uri, "%s://%s:%d", vlc_proto,
1057                       host, port) == -1)
1058             return VLC_ENOMEM;
1059     }
1060     else
1061     {
1062         /* Non-connected (normally multicast) media */
1063         char psz_source[258] = "";
1064         const char *sfilter = FindAttribute (p_sdp, 0, "source-filter");
1065         if (sfilter != NULL)
1066         {
1067             char psz_source_ip[256];
1068             unsigned ipv;
1069
1070             if (sscanf (sfilter, " incl IN IP%u %*s %255s ", &ipv,
1071                         psz_source_ip) == 2)
1072             {
1073                 /* According to RFC4570, FQDNs can be used for source-filters,
1074                  * but -seriously- this is impractical */
1075                 switch (ipv)
1076                 {
1077 #ifdef AF_INET6
1078                     case 6:
1079                     {
1080                         struct in6_addr addr;
1081                         if ((inet_pton (AF_INET6, psz_source_ip, &addr) > 0)
1082                         && (inet_ntop (AF_INET6, &addr, psz_source + 1,
1083                                         sizeof (psz_source) - 2) != NULL))
1084                         {
1085                             psz_source[0] = '[';
1086                             psz_source[strlen (psz_source)] = ']';
1087                         }
1088                         break;
1089                     }
1090 #endif
1091                     case 4:
1092                     {
1093                         struct in_addr addr;
1094                         if ((inet_pton (AF_INET, psz_source_ip, &addr) > 0)
1095                         && (inet_ntop (AF_INET, &addr, psz_source,
1096                                         sizeof (psz_source)) == NULL))
1097                             *psz_source = '\0';
1098                         break;
1099                     }
1100                 }
1101             }
1102         }
1103
1104         if (asprintf (&p_sdp->psz_uri, "%s://%s@%s:%i", vlc_proto, psz_source,
1105                      host, port) == -1)
1106             return VLC_ENOMEM;
1107     }
1108
1109     return VLC_SUCCESS;
1110 }
1111
1112
1113 static int ParseSDPConnection (const char *str, struct sockaddr_storage *addr,
1114                                socklen_t *addrlen, unsigned *number)
1115 {
1116     char host[60];
1117     unsigned fam, n1, n2;
1118
1119     int res = sscanf (str, "IN IP%u %59[^/]/%u/%u", &fam, host, &n1, &n2);
1120     if (res < 2)
1121         return -1;
1122
1123     switch (fam)
1124     {
1125 #ifdef AF_INET6
1126         case 6:
1127             addr->ss_family = AF_INET6;
1128 # ifdef HAVE_SA_LEN
1129             addr->ss_len =
1130 # endif
1131            *addrlen = sizeof (struct sockaddr_in6);
1132
1133             if (inet_pton (AF_INET6, host,
1134                            &((struct sockaddr_in6 *)addr)->sin6_addr) <= 0)
1135                 return -1;
1136
1137             *number = (res >= 3) ? n1 : 1;
1138             break;
1139 #endif
1140
1141         case 4:
1142             addr->ss_family = AF_INET;
1143 # ifdef HAVE_SA_LEN
1144             addr->ss_len =
1145 # endif
1146            *addrlen = sizeof (struct sockaddr_in);
1147
1148             if (inet_pton (AF_INET, host,
1149                            &((struct sockaddr_in *)addr)->sin_addr) <= 0)
1150                 return -1;
1151
1152             *number = (res >= 4) ? n2 : 1;
1153             break;
1154
1155         default:
1156             return -1;
1157     }
1158     return 0;
1159 }
1160
1161
1162 /***********************************************************************
1163  * ParseSDP : SDP parsing
1164  * *********************************************************************
1165  * Validate SDP and parse all fields
1166  ***********************************************************************/
1167 static sdp_t *ParseSDP (vlc_object_t *p_obj, const char *psz_sdp)
1168 {
1169     if( psz_sdp == NULL )
1170         return NULL;
1171
1172     sdp_t *p_sdp = calloc (1, sizeof (*p_sdp));
1173     if (p_sdp == NULL)
1174         return NULL;
1175
1176     char expect = 'V';
1177     struct sockaddr_storage glob_addr;
1178     memset (&glob_addr, 0, sizeof (glob_addr));
1179     socklen_t glob_len = 0;
1180     unsigned glob_count = 1;
1181     int port = 0;
1182
1183     /* TODO: use iconv and charset attribute instead of EnsureUTF8 */
1184     while (*psz_sdp)
1185     {
1186         /* Extract one line */
1187         char *eol = strchr (psz_sdp, '\n');
1188         size_t linelen = eol ? (size_t)(eol - psz_sdp) : strlen (psz_sdp);
1189         char line[linelen + 1];
1190         memcpy (line, psz_sdp, linelen);
1191         line[linelen] = '\0';
1192
1193         psz_sdp += linelen + 1;
1194
1195         /* Remove carriage return if present */
1196         eol = strchr (line, '\r');
1197         if (eol != NULL)
1198         {
1199             linelen = eol - line;
1200             line[linelen] = '\0';
1201         }
1202
1203         /* Validate line */
1204         char cat = line[0], *data = line + 2;
1205         if (!cat || (strchr ("vosiuepcbtrzkam", cat) == NULL))
1206         {
1207             /* MUST ignore SDP with unknown line type */
1208             msg_Dbg (p_obj, "unknown SDP line type: 0x%02x", (int)cat);
1209             goto error;
1210         }
1211         if (line[1] != '=')
1212         {
1213             msg_Dbg (p_obj, "invalid SDP line: %s", line);
1214             goto error;
1215         }
1216
1217         assert (linelen >= 2);
1218
1219         /* SDP parsing state machine
1220          * We INTERNALLY use uppercase for session, lowercase for media
1221          */
1222         switch (expect)
1223         {
1224             /* Session description */
1225             case 'V':
1226                 expect = 'O';
1227                 if (cat != 'v')
1228                 {
1229                     msg_Dbg (p_obj, "missing SDP version");
1230                     goto error;
1231                 }
1232                 if (strcmp (data, "0"))
1233                 {
1234                     msg_Dbg (p_obj, "unknown SDP version: %s", data);
1235                     goto error;
1236                 }
1237                 break;
1238
1239             case 'O':
1240             {
1241                 expect = 'S';
1242                 if (cat != 'o')
1243                 {
1244                     msg_Dbg (p_obj, "missing SDP originator");
1245                     goto error;
1246                 }
1247
1248                 if ((sscanf (data, "%63s %"PRIu64" %"PRIu64" IN IP%u %1023s",
1249                              p_sdp->username, &p_sdp->session_id,
1250                              &p_sdp->session_version, &p_sdp->orig_ip_version,
1251                              p_sdp->orig_host) != 5)
1252                  || ((p_sdp->orig_ip_version != 4)
1253                   && (p_sdp->orig_ip_version != 6)))
1254                 {
1255                     msg_Dbg (p_obj, "SDP origin not supported: %s", data);
1256                     /* Or maybe out-of-range, but this looks suspicious */
1257                     return NULL;
1258                 }
1259                 EnsureUTF8 (p_sdp->orig_host);
1260                 break;
1261             }
1262
1263             case 'S':
1264             {
1265                 expect = 'I';
1266                 if ((cat != 's') || !*data)
1267                 {
1268                     /* MUST be present AND non-empty */
1269                     msg_Dbg (p_obj, "missing SDP session name");
1270                     goto error;
1271                 }
1272                 assert (p_sdp->psz_sessionname == NULL); // no memleak here
1273                 p_sdp->psz_sessionname = strdup (data);
1274                 if (p_sdp->psz_sessionname == NULL)
1275                     goto error;
1276                 EnsureUTF8 (p_sdp->psz_sessionname);
1277                 break;
1278             }
1279
1280             case 'I':
1281                 expect = 'U';
1282                 if (cat == 'i')
1283                     break;
1284             case 'U':
1285                 expect = 'E';
1286                 if (cat == 'u')
1287                     break;
1288             case 'E':
1289                 expect = 'E';
1290                 if (cat == 'e')
1291                     break;
1292             case 'P':
1293                 expect = 'P';
1294                 if (cat == 'p')
1295                     break;
1296             case 'C':
1297                 expect = 'B';
1298                 if (cat == 'c')
1299                 {
1300                     if (ParseSDPConnection (data, &glob_addr, &glob_len,
1301                                             &glob_count))
1302                     {
1303                         msg_Dbg (p_obj, "SDP connection infos not supported: "
1304                                  "%s", data);
1305                         goto error;
1306                     }
1307                     break;
1308                 }
1309             case 'B':
1310                 assert (expect == 'B');
1311                 if (cat == 'b')
1312                     break;
1313             case 'T':
1314                 expect = 'R';
1315                 if (cat != 't')
1316                 {
1317                     msg_Dbg (p_obj, "missing SDP time description");
1318                     goto error;
1319                 }
1320                 break;
1321
1322             case 'R':
1323                 if ((cat == 't') || (cat == 'r'))
1324                     break;
1325
1326             case 'Z':
1327                 expect = 'K';
1328                 if (cat == 'z')
1329                     break;
1330             case 'K':
1331                 expect = 'A';
1332                 if (cat == 'k')
1333                     break;
1334             case 'A':
1335                 //expect = 'A';
1336                 if (cat == 'a')
1337                 {
1338                     attribute_t *p_attr = MakeAttribute (data);
1339                     TAB_APPEND( p_sdp->i_attributes, p_sdp->pp_attributes, p_attr );
1340                     break;
1341                 }
1342
1343             /* Media description */
1344             case 'm':
1345             media:
1346             {
1347                 expect = 'i';
1348                 if (cat != 'm')
1349                 {
1350                     msg_Dbg (p_obj, "missing SDP media description");
1351                     goto error;
1352                 }
1353                 struct sdp_media_t *m;
1354                 m = realloc (p_sdp->mediav, (p_sdp->mediac + 1) * sizeof (*m));
1355                 if (m == NULL)
1356                     goto error;
1357
1358                 p_sdp->mediav = m;
1359                 m += p_sdp->mediac;
1360                 p_sdp->mediac++;
1361
1362                 memset (m, 0, sizeof (*m));
1363                 memcpy (&m->addr, &glob_addr, m->addrlen = glob_len);
1364                 m->n_addr = glob_count;
1365
1366                 /* TODO: remember media type (if we need multiple medias) */
1367                 data = strchr (data, ' ');
1368                 if (data == NULL)
1369                 {
1370                     msg_Dbg (p_obj, "missing SDP media port");
1371                     goto error;
1372                 }
1373                 port = atoi (++data);
1374                 if (port <= 0 || port >= 65536)
1375                 {
1376                     msg_Dbg (p_obj, "invalid transport port %d", port);
1377                     goto error;
1378                 }
1379                 net_SetPort ((struct sockaddr *)&m->addr, htons (port));
1380
1381                 data = strchr (data, ' ');
1382                 if (data == NULL)
1383                 {
1384                     msg_Dbg (p_obj, "missing SDP media format");
1385                     goto error;
1386                 }
1387                 m->fmt = strdup (++data);
1388                 if (m->fmt == NULL)
1389                     goto error;
1390
1391                 break;
1392             }
1393             case 'i':
1394                 expect = 'c';
1395                 if (cat == 'i')
1396                     break;
1397             case 'c':
1398                 expect = 'b';
1399                 if (cat == 'c')
1400                 {
1401                     struct sdp_media_t *m = p_sdp->mediav + p_sdp->mediac - 1;
1402                     if (ParseSDPConnection (data, &m->addr, &m->addrlen,
1403                                             &m->n_addr))
1404                     {
1405                         msg_Dbg (p_obj, "SDP connection infos not supported: "
1406                                  "%s", data);
1407                         goto error;
1408                     }
1409                     net_SetPort ((struct sockaddr *)&m->addr, htons (port));
1410                     break;
1411                 }
1412             case 'b':
1413                 expect = 'b';
1414                 if (cat == 'b')
1415                     break;
1416             case 'k':
1417                 expect = 'a';
1418                 if (cat == 'k')
1419                     break;
1420             case 'a':
1421                 assert (expect == 'a');
1422                 if (cat == 'a')
1423                 {
1424                     attribute_t *p_attr = MakeAttribute (data);
1425                     if (p_attr == NULL)
1426                         goto error;
1427
1428                     TAB_APPEND (p_sdp->mediav[p_sdp->mediac - 1].i_attributes,
1429                                 p_sdp->mediav[p_sdp->mediac - 1].pp_attributes, p_attr);
1430                     break;
1431                 }
1432
1433                 if (cat == 'm')
1434                     goto media;
1435
1436                 if (cat != 'm')
1437                 {
1438                     msg_Dbg (p_obj, "unexpected SDP line: 0x%02x", (int)cat);
1439                     goto error;
1440                 }
1441                 break;
1442
1443             default:
1444                 msg_Err (p_obj, "*** BUG in SDP parser! ***");
1445                 goto error;
1446         }
1447     }
1448
1449     return p_sdp;
1450
1451 error:
1452     FreeSDP (p_sdp);
1453     return NULL;
1454 }
1455
1456 static int InitSocket( services_discovery_t *p_sd, const char *psz_address,
1457                        int i_port )
1458 {
1459     int i_fd = net_ListenUDP1 ((vlc_object_t *)p_sd, psz_address, i_port);
1460     if (i_fd == -1)
1461         return VLC_EGENERIC;
1462
1463     shutdown( i_fd, SHUT_WR );
1464     INSERT_ELEM (p_sd->p_sys->pi_fd, p_sd->p_sys->i_fd,
1465                  p_sd->p_sys->i_fd, i_fd);
1466     return VLC_SUCCESS;
1467 }
1468
1469 static int Decompress( const unsigned char *psz_src, unsigned char **_dst, int i_len )
1470 {
1471 #ifdef HAVE_ZLIB_H
1472     int i_result, i_dstsize, n = 0;
1473     unsigned char *psz_dst = NULL;
1474     z_stream d_stream;
1475
1476     memset (&d_stream, 0, sizeof (d_stream));
1477
1478     i_result = inflateInit(&d_stream);
1479     if( i_result != Z_OK )
1480         return( -1 );
1481
1482     d_stream.next_in = (Bytef *)psz_src;
1483     d_stream.avail_in = i_len;
1484
1485     do
1486     {
1487         n++;
1488         psz_dst = (unsigned char *)realloc( psz_dst, n * 1000 );
1489         d_stream.next_out = (Bytef *)&psz_dst[(n - 1) * 1000];
1490         d_stream.avail_out = 1000;
1491
1492         i_result = inflate(&d_stream, Z_NO_FLUSH);
1493         if( ( i_result != Z_OK ) && ( i_result != Z_STREAM_END ) )
1494         {
1495             inflateEnd( &d_stream );
1496             free( psz_dst );
1497             return( -1 );
1498         }
1499     }
1500     while( ( d_stream.avail_out == 0 ) && ( d_stream.avail_in != 0 ) &&
1501            ( i_result != Z_STREAM_END ) );
1502
1503     i_dstsize = d_stream.total_out;
1504     inflateEnd( &d_stream );
1505
1506     *_dst = (unsigned char *)realloc( psz_dst, i_dstsize );
1507
1508     return i_dstsize;
1509 #else
1510     (void)psz_src;
1511     (void)_dst;
1512     (void)i_len;
1513     return -1;
1514 #endif
1515 }
1516
1517
1518 static void FreeSDP( sdp_t *p_sdp )
1519 {
1520     free( p_sdp->psz_sessionname );
1521     free( p_sdp->psz_uri );
1522
1523     for (unsigned j = 0; j < p_sdp->mediac; j++)
1524     {
1525         free (p_sdp->mediav[j].fmt);
1526         for (int i = 0; i < p_sdp->mediav[j].i_attributes; i++)
1527             FreeAttribute (p_sdp->mediav[j].pp_attributes[i]);
1528         free (p_sdp->mediav[j].pp_attributes);
1529     }
1530     free (p_sdp->mediav);
1531
1532     for (int i = 0; i < p_sdp->i_attributes; i++)
1533         FreeAttribute (p_sdp->pp_attributes[i]);
1534
1535     free (p_sdp->pp_attributes);
1536     free (p_sdp);
1537 }
1538
1539 static int RemoveAnnounce( services_discovery_t *p_sd,
1540                            sap_announce_t *p_announce )
1541 {
1542     int i;
1543
1544     if( p_announce->p_sdp )
1545     {
1546         FreeSDP( p_announce->p_sdp );
1547         p_announce->p_sdp = NULL;
1548     }
1549
1550     if( p_announce->p_item )
1551     {
1552         services_discovery_RemoveItem( p_sd, p_announce->p_item );
1553         vlc_gc_decref( p_announce->p_item );
1554         p_announce->p_item = NULL;
1555     }
1556
1557     for( i = 0; i< p_sd->p_sys->i_announces; i++)
1558     {
1559         if( p_sd->p_sys->pp_announces[i] == p_announce )
1560         {
1561             REMOVE_ELEM( p_sd->p_sys->pp_announces, p_sd->p_sys->i_announces,
1562                          i);
1563             break;
1564         }
1565     }
1566
1567     free( p_announce );
1568
1569     return VLC_SUCCESS;
1570 }
1571
1572 static bool IsSameSession( sdp_t *p_sdp1, sdp_t *p_sdp2 )
1573 {
1574     /* A session is identified by
1575      * - username,
1576      * - session_id,
1577      * - network type (which is always IN),
1578      * - address type (currently, this means IP version),
1579      * - and hostname.
1580      */
1581     if (strcmp (p_sdp1->username, p_sdp2->username)
1582      || (p_sdp1->session_id != p_sdp2->session_id)
1583      || (p_sdp1->orig_ip_version != p_sdp2->orig_ip_version)
1584      || strcmp (p_sdp1->orig_host, p_sdp2->orig_host))
1585         return false;
1586
1587     return true;
1588 }
1589
1590
1591 static inline attribute_t *MakeAttribute (const char *str)
1592 {
1593     attribute_t *a = malloc (sizeof (*a) + strlen (str) + 1);
1594     if (a == NULL)
1595         return NULL;
1596
1597     strcpy (a->name, str);
1598     EnsureUTF8 (a->name);
1599     char *value = strchr (a->name, ':');
1600     if (value != NULL)
1601     {
1602         *value++ = '\0';
1603         a->value = value;
1604     }
1605     else
1606         a->value = "";
1607     return a;
1608 }
1609
1610
1611 static const char *GetAttribute (attribute_t **tab, unsigned n,
1612                                  const char *name)
1613 {
1614     for (unsigned i = 0; i < n; i++)
1615         if (strcasecmp (tab[i]->name, name) == 0)
1616             return tab[i]->value;
1617     return NULL;
1618 }
1619
1620
1621 static inline void FreeAttribute (attribute_t *a)
1622 {
1623     free (a);
1624 }