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