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