git.sesse.net Git - vlc/blob - 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