90d664b40c258214909e2196614e82a1dcd08ccd
[vlc.git] / src / stream_output / announce.c
1 /*****************************************************************************
2  * announce.c : Session announcement
3  *****************************************************************************
4  * Copyright (C) 2002 VideoLAN
5  *
6  * Authors: ClĂ©ment Stenac <zorglub@via.ecp.fr>
7  *          Damien Lucas <nitrox@via.ecp.fr>
8  *
9  * This program is free software; you can redistribute it and/or modify
10  * it under the terms of the GNU General Public License as published by
11  * the Free Software Foundation; either version 2 of the License, or
12  * (at your option) any later version.
13  *
14  * This program is distributed in the hope that it will be useful,
15  * but WITHOUT ANY WARRANTY; without even the implied warranty of
16  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
17  * GNU General Public License for more details.
18  *
19  * You should have received a copy of the GNU General Public License
20  * along with this program; if not, write to the Free Software
21  * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111, USA.
22  *****************************************************************************/
23
24 /*****************************************************************************
25  * Preamble
26  *****************************************************************************/
27 #include <errno.h>                                                 /* ENOMEM */
28 #include <stdlib.h>                                                /* free() */
29 #include <stdio.h>                                              /* sprintf() */
30 #include <string.h>                                            /* strerror() */
31
32 #include <vlc/vlc.h>
33 #include <vlc/sout.h>
34
35 #ifdef HAVE_UNISTD_H
36 #   include <unistd.h>
37 #endif
38
39 #ifdef WIN32
40 #   include <winsock2.h>
41 #   include <ws2tcpip.h>
42 #   ifndef IN_MULTICAST
43 #       define IN_MULTICAST(a) IN_CLASSD(a)
44 #   endif
45 #else
46 #   include <sys/socket.h>
47 #endif
48
49 #include "announce.h"
50 #include "network.h"
51
52 #define SAP_IPV4_ADDR "224.2.127.254" /* Standard port and address for SAP */
53 #define SAP_PORT 9875
54
55 #define SAP_IPV6_ADDR_1 "FF0"
56 #define SAP_IPV6_ADDR_2 "::2:7FFE"
57
58 #define DEFAULT_PORT "1234"
59
60 /****************************************************************************
61  *  Split : split a string into two parts: the one which is before the delim
62  *               and the one which is after.
63  *               NULL is returned if delim is not found
64  ****************************************************************************/
65
66 static char * split( char *psz_in, char *psz_out1, char *psz_out2, char delim)
67 {
68     unsigned int i_count = 0; /* pos in input string */
69     unsigned int i_pos1  = 0; /* pos in out2 string */
70     unsigned int i_pos2  = 0;
71     char *psz_cur; /* store the pos of the first delim found */
72
73     /* Skip spaces at the beginning */
74     while( psz_in[i_count] == ' ' )
75     {
76         i_count++;
77     }
78
79     if( psz_in[i_count] == '\0' )
80     {
81         return NULL;
82     }
83
84     /* Look for delim */
85     while( psz_in[i_count] && psz_in[i_count] != delim )
86     {
87         psz_out1[i_pos1] = psz_in[i_count];
88         i_count++;
89         i_pos1++;
90     }
91     /* Mark the end of out1 */
92     psz_out1[i_pos1] = '\0';
93
94     if( psz_in[i_count] == '\0' )
95     {
96         return NULL;
97     }
98
99     /* store pos of the first delim */
100     psz_cur = psz_in + i_count;
101
102     /* skip all delim and all spaces */
103     while( psz_in[i_count] == ' ' || psz_in[i_count] == delim )
104     {
105         i_count++;
106     }
107
108     if( psz_in[i_count] == '\0' )
109     {
110         return psz_cur;
111     }
112
113     /* Store the second string */
114     while( psz_in[i_count] )
115     {
116         psz_out2[i_pos2] = psz_in[i_count];
117         i_pos2++;
118         i_count++;
119     }
120     psz_out2[i_pos2] = '\0';
121
122     return psz_cur;
123 }
124
125 /*****************************************************************************
126  * sout_SAPNew: Creates a SAP Session
127  *****************************************************************************/
128 sap_session_t * sout_SAPNew ( sout_instance_t *p_sout, char * psz_url_arg,
129                               char * psz_name_arg, int ip_version,
130                               char * psz_v6_scope )
131 {
132     sap_session_t       *p_sap; /* The SAP structure */
133     module_t            *p_network; /* Network module */
134     network_socket_t    socket_desc; /* Socket descriptor */
135     char                *sap_ipv6_addr = NULL; /* IPv6 built address */
136     char                *psz_eol; /* Used to parse IPv6 URIs */
137     int                 i_port; /* Port in numerical format */
138
139     /* Allocate the SAP structure */
140     p_sap = (sap_session_t *) malloc( sizeof ( sap_session_t ) ) ;
141     if ( !p_sap )
142     {
143         msg_Err( p_sout, "out of memory" );
144         return NULL;
145     }
146
147     /* Fill the information in the structure */
148     if( strstr( psz_url_arg, "[" ) )
149     {
150         /* We have an IPv6 address. Do not use ':' as the port separator */
151         psz_eol = strchr( psz_url_arg, ']' );
152         if( !psz_eol )
153         {
154             msg_Warn( p_sout, "no matching ], unable to parse URI");
155             return NULL;
156         }
157
158         if( !psz_eol++ )
159         {
160             sprintf( p_sap->psz_url, "%s", psz_url_arg );
161             sprintf( p_sap->psz_port, "%s", DEFAULT_PORT );
162         }
163         else
164         {
165             *psz_eol = '\0';
166             sprintf( p_sap->psz_url, "%s", psz_url_arg );
167             psz_eol++;
168             if( psz_eol )
169             {
170                 sprintf( p_sap->psz_port, "%s", psz_eol );
171             }
172         }
173     }
174     else
175     {
176         split( psz_url_arg, p_sap->psz_url, p_sap->psz_port, ':' );
177     }
178
179     /* Check if we have a port */
180     if( !strlen( p_sap->psz_port ) )
181     {
182         sprintf( p_sap->psz_port, "%s", DEFAULT_PORT );
183     }
184
185     /* Make sure our port is valid and atoi it */
186     i_port = atoi( p_sap->psz_port );
187
188     if( !i_port )
189     {
190         sprintf( p_sap->psz_port, "%s", DEFAULT_PORT );
191     }
192     else
193     {
194         sprintf( p_sap->psz_port, "%i", i_port );
195     }
196
197     /* The name that we send */
198     sprintf( p_sap->psz_name, "%s", psz_name_arg );
199
200     p_sap->i_ip_version = ip_version;
201
202     /* Only "6" triggers IPv6. IPv4 is default */
203     if( ip_version != 6 )
204     {
205         msg_Dbg( p_sout, "creating IPv4 SAP socket" );
206
207         /* Fill the socket descriptor */
208         socket_desc.i_type            = NETWORK_UDP;
209         socket_desc.psz_bind_addr     = "";
210         socket_desc.i_bind_port       = 0;
211         socket_desc.psz_server_addr   = SAP_IPV4_ADDR;
212         socket_desc.i_server_port     = SAP_PORT;
213         socket_desc.i_handle          = 0;
214         socket_desc.i_ttl             = 0;
215
216         /* Call the network module */
217         p_sout->p_private = (void*) &socket_desc;
218         if( !( p_network = module_Need( p_sout, "network", "ipv4" ) ) )
219         {
220              msg_Warn( p_sout, "failed to open a connection (udp)" );
221              return NULL;
222         }
223         module_Unneed( p_sout, p_network );
224
225         p_sap->i_socket = socket_desc.i_handle;
226         if( p_sap->i_socket < 0 )
227         {
228             msg_Warn( p_sout, "unable to initialize SAP" );
229             return NULL;
230         }
231     }
232     else
233     {
234         msg_Dbg( p_sout, "creating IPv6 SAP socket with scope %s",
235                          psz_v6_scope );
236
237         /* Initialize and build the IPv6 address to broadcast to */
238         sap_ipv6_addr = (char *) malloc( 28 * sizeof(char) );
239         if ( !sap_ipv6_addr )
240         {
241             msg_Err( p_sout, "out of memory" );
242             return NULL;
243         }
244         sprintf( sap_ipv6_addr, "%s%c%s",
245                  SAP_IPV6_ADDR_1, psz_v6_scope[0], SAP_IPV6_ADDR_2 );
246
247         /* Fill the socket descriptor */
248         socket_desc.i_type          = NETWORK_UDP;
249         socket_desc.psz_bind_addr   = "";
250         socket_desc.i_bind_port     = 0;
251         socket_desc.psz_server_addr = sap_ipv6_addr;
252         socket_desc.i_server_port   = SAP_PORT;
253         socket_desc.i_handle        = 0;
254         socket_desc.i_ttl           = 0;
255
256         /* Call the network module */
257         p_sout->p_private = (void *) &socket_desc;
258         if( !( p_network = module_Need( p_sout, "network", "ipv6" ) ) )
259         {
260             msg_Warn( p_sout, "failed to open a connection (udp)" );
261             return NULL;
262         }
263         module_Unneed( p_sout, p_network );
264
265         p_sap->i_socket = socket_desc.i_handle;
266         if( p_sap->i_socket <= 0 )
267         {
268             msg_Warn( p_sout, "unable to initialize SAP" );
269             return NULL;
270         }
271
272         /* Free what we allocated */
273         if( sap_ipv6_addr )
274         {
275             free( sap_ipv6_addr );
276         }
277     }
278
279     msg_Dbg( p_sout, "SAP initialization complete" );
280
281     return p_sap;
282 }
283
284 /*****************************************************************************
285  * sout_SAPDelete: Deletes a SAP Session
286  *****************************************************************************/
287 void sout_SAPDelete( sout_instance_t *p_sout, sap_session_t * p_sap )
288 {
289     int i_ret;
290
291 #if defined( UNDER_CE )
292     i_ret = CloseHandle( (HANDLE)p_sap->i_socket );
293 #elif defined( WIN32 )
294     i_ret = closesocket( p_sap->i_socket );
295 #else
296     i_ret = close( p_sap->i_socket );
297 #endif
298
299     if( i_ret )
300     {
301         msg_Err( p_sout, "unable to close SAP socket" );
302     }
303
304     free( p_sap );
305 }
306
307 /*****************************************************************************
308  * sout_SAPSend: Sends a SAP packet
309  *****************************************************************************/
310 void sout_SAPSend( sout_instance_t *p_sout, sap_session_t * p_sap )
311 {
312     char psz_msg[1000];                     /* SDP content */
313     char *psz_head;                         /* SAP header */
314     char *psz_send;                         /* What we send */
315     char *psz_type = "application/sdp";
316     int i_header_size;                      /* SAP header size */
317     int i_msg_size;                         /* SDP content size */
318     int i_size;                             /* Total size */
319     int i_ret = 0;
320
321     /* We send a packet every 24 calls to the function */
322     if( p_sap->i_calls++ < 24 )
323     {
324         return;
325     }
326
327     i_header_size = 8 + strlen( psz_type ) + 1;
328     psz_head = (char *) malloc( i_header_size * sizeof( char ) );
329
330     if( ! psz_head )
331     {
332         msg_Err( p_sout, "out of memory" );
333         return;
334     }
335
336     /* Create the SAP headers */
337     psz_head[0] = 0x20; /* Means IPv4, not encrypted, not compressed */
338     psz_head[1] = 0x00; /* No authentification */
339     psz_head[2] = 0x42; /* Version */
340     psz_head[3] = 0x12; /* Version */
341
342     psz_head[4] = 0x01; /* Source IP  FIXME: we should get the real address */
343     psz_head[5] = 0x02; /* idem */
344     psz_head[6] = 0x03; /* idem */
345     psz_head[7] = 0x04; /* idem */
346
347     strncpy( psz_head + 8, psz_type, 15 );
348     psz_head[ i_header_size-1 ] = '\0';
349
350     /* Create the SDP content */
351     /* Do not add spaces at beginning of the lines ! */
352     sprintf( psz_msg, "v=0\n"
353                       "o=VideoLAN 3247692199 3247895918 IN IP4 VideoLAN\n"
354                       "s=%s\n"
355                       "u=VideoLAN\n"
356                       "t=0 0\n"
357                       "m=audio %s udp 14\n"
358                       "c=IN IP4 %s/15\n"
359                       "a=type:test\n",
360              p_sap->psz_name, p_sap->psz_port, p_sap->psz_url );
361
362     i_msg_size = strlen( psz_msg );
363     i_size = i_msg_size + i_header_size;
364
365     /* Create the message */
366     psz_send = (char *) malloc( i_size*sizeof(char) );
367     if( !psz_send )
368     {
369         msg_Err( p_sout, "out of memory" );
370         return;
371     }
372
373     memcpy( psz_send, psz_head, i_header_size );
374     memcpy( psz_send + i_header_size, psz_msg, i_msg_size );
375
376     if( i_size < 1024 ) /* We mustn't send packets larger than 1024B */
377     {
378         i_ret = send( p_sap->i_socket, psz_send, i_size, 0 );
379     }
380
381     if( i_ret <= 0 )
382     {
383         msg_Warn( p_sout, "SAP send failed on socket %i (%s)",
384                           p_sap->i_socket, strerror(errno) );
385     }
386
387     p_sap->i_calls = 0;
388
389     /* Free what we allocated */
390     free( psz_send );
391     free( psz_head );
392 }
393