git.sesse.net Git - vlc/blob - modules/misc/rtsp.c

   1 /*****************************************************************************
   2  * rtsp.c: rtsp VoD server module
   3  *****************************************************************************
   4  * Copyright (C) 2003-2006 the VideoLAN team
   5  * $Id$
   6  *
   7  * Authors: Laurent Aimar <fenrir@via.ecp.fr>
   8  *          Gildas Bazin <gbazin@videolan.org>
   9  *
  10  * This program is free software; you can redistribute it and/or modify
  11  * it under the terms of the GNU General Public License as published by
  12  * the Free Software Foundation; either version 2 of the License, or
  13  * (at your option) any later version.
  14  *
  15  * This program is distributed in the hope that it will be useful,
  16  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  17  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  18  * GNU General Public License for more details.
  19  *
  20  * You should have received a copy of the GNU General Public License
  21  * along with this program; if not, write to the Free Software
  22  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
  23  *****************************************************************************/
  24
  25 /*****************************************************************************
  26  * Preamble
  27  *****************************************************************************/
  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 <vlc_input.h>
  36 #include <vlc_sout.h>
  37 #include <vlc_block.h>
  38
  39 #include "vlc_httpd.h"
  40 #include "vlc_vod.h"
  41 #include "vlc_url.h"
  42 #include <vlc_network.h>
  43 #include <vlc_charset.h>
  44 #include <vlc_strings.h>
  45
  46 #include <errno.h>
  47
  48 #ifndef WIN32
  49 # include <locale.h>
  50 #endif
  51
  52 #ifdef HAVE_XLOCALE_H
  53 # include <xlocale.h>
  54 #endif
  55
  56 /*****************************************************************************
  57  * Module descriptor
  58  *****************************************************************************/
  59 static int  Open ( vlc_object_t * );
  60 static void Close( vlc_object_t * );
  61
  62 #define HOST_TEXT N_( "RTSP host address" )
  63 #define HOST_LONGTEXT N_( \
  64     "This defines the address, port and path the RTSP VOD server will listen " \
  65     "on.\nSyntax is address:port/path. The default is to listen on all "\
  66     "interfaces (address 0.0.0.0), on port 554, with no path.\nTo listen " \
  67     "only on the local interface, use \"localhost\" as address." )
  68
  69 #define THROTLE_TEXT N_( "Maximum number of connections" )
  70 #define THROTLE_LONGTEXT N_( "This limits the maximum number of clients " \
  71     "that can connect to the RTSP VOD. 0 means no limit."  )
  72
  73 #define RAWMUX_TEXT N_( "MUX for RAW RTSP transport" )
  74
  75 #define SESSION_TIMEOUT_TEXT N_( "Sets the timeout option in the RTSP " \
  76     "session string" )
  77 #define SESSION_TIMEOUT_LONGTEXT N_( "Defines what timeout option to add " \
  78     "to the RTSP session ID string. Setting it to a negative number removes " \
  79     "the timeout option entirely. This is needed by some IPTV STBs (such as " \
  80     "those made by HansunTech) which get confused by it. The default is 5." )
  81
  82 vlc_module_begin();
  83     set_shortname( N_("RTSP VoD" ) );
  84     set_description( N_("RTSP VoD server") );
  85     set_category( CAT_SOUT );
  86     set_subcategory( SUBCAT_SOUT_VOD );
  87     set_capability( "vod server", 1 );
  88     set_callbacks( Open, Close );
  89     add_shortcut( "rtsp" );
  90     add_string ( "rtsp-host", NULL, NULL, HOST_TEXT, HOST_LONGTEXT, true );
  91     add_string( "rtsp-raw-mux", "ts", NULL, RAWMUX_TEXT,
  92                 RAWMUX_TEXT, true );
  93     add_integer( "rtsp-throttle-users", 0, NULL, THROTLE_TEXT,
  94                                            THROTLE_LONGTEXT, true );
  95     add_integer( "rtsp-session-timeout", 5, NULL, SESSION_TIMEOUT_TEXT,
  96                  SESSION_TIMEOUT_LONGTEXT, true );
  97 vlc_module_end();
  98
  99 /*****************************************************************************
 100  * Exported prototypes
 101  *****************************************************************************/
 102
 103 typedef struct media_es_t media_es_t;
 104
 105 typedef struct
 106 {
 107     media_es_t *p_media_es;
 108     char *psz_ip;
 109     int i_port;
 110
 111 } rtsp_client_es_t;
 112
 113 typedef struct
 114 {
 115     char *psz_session;
 116     int64_t i_last; /* for timeout */
 117
 118     bool b_playing; /* is it in "play" state */
 119     bool b_paused; /* is it in "pause" state */
 120
 121     int i_es;
 122     rtsp_client_es_t **es;
 123
 124 } rtsp_client_t;
 125
 126 struct media_es_t
 127 {
 128     /* VoD server */
 129     vod_t *p_vod;
 130
 131     /* RTSP server */
 132     httpd_url_t *p_rtsp_url;
 133
 134     vod_media_t *p_media;
 135
 136     es_format_t fmt;
 137     int         i_port;
 138     uint8_t     i_payload_type;
 139     char        *psz_rtpmap;
 140     char        *psz_fmtp;
 141
 142 };
 143
 144 struct vod_media_t
 145 {
 146     int id;
 147
 148     /* VoD server */
 149     vod_t *p_vod;
 150
 151     /* RTSP server */
 152     httpd_url_t  *p_rtsp_url;
 153     char         *psz_rtsp_control_v4;
 154     char         *psz_rtsp_control_v6;
 155     char         *psz_rtsp_path;
 156
 157     int  i_port;
 158     int  i_port_audio;
 159     int  i_port_video;
 160     int  i_ttl;
 161     int  i_payload_type;
 162
 163     int64_t i_sdp_id;
 164     int     i_sdp_version;
 165
 166     bool b_multicast;
 167
 168     vlc_mutex_t lock;
 169
 170     /* ES list */
 171     int        i_es;
 172     media_es_t **es;
 173     char       *psz_mux;
 174     bool  b_raw;
 175
 176     /* RTSP client */
 177     int           i_rtsp;
 178     rtsp_client_t **rtsp;
 179
 180     /* Infos */
 181     char *psz_session_name;
 182     char *psz_session_description;
 183     char *psz_session_url;
 184     char *psz_session_email;
 185     mtime_t i_length;
 186 };
 187
 188 struct vod_sys_t
 189 {
 190     /* RTSP server */
 191     httpd_host_t *p_rtsp_host;
 192     char *psz_path;
 193     int i_port;
 194     int i_throttle_users;
 195     int i_connections;
 196
 197     char *psz_raw_mux;
 198
 199     int i_session_timeout;
 200
 201     /* List of media */
 202     vlc_mutex_t lock_media;
 203     int i_media_id;
 204     int i_media;
 205     vod_media_t **media;
 206
 207     /* */
 208     block_fifo_t *p_fifo_cmd;
 209 };
 210
 211 /* rtsp delayed command (to avoid deadlock between vlm/httpd) */
 212 typedef enum
 213 {
 214     RTSP_CMD_TYPE_NONE,  /* Exit requested */
 215
 216     RTSP_CMD_TYPE_PLAY,
 217     RTSP_CMD_TYPE_PAUSE,
 218     RTSP_CMD_TYPE_STOP,
 219     RTSP_CMD_TYPE_SEEK,
 220     RTSP_CMD_TYPE_REWIND,
 221     RTSP_CMD_TYPE_FORWARD,
 222 } rtsp_cmd_type_t;
 223
 224 static vod_media_t *MediaNew( vod_t *, const char *, input_item_t * );
 225 static void         MediaDel( vod_t *, vod_media_t * );
 226 static int          MediaAddES( vod_t *, vod_media_t *, es_format_t * );
 227 static void         MediaDelES( vod_t *, vod_media_t *, es_format_t * );
 228
 229 static void* CommandThread( vlc_object_t *p_this );
 230 static void  CommandPush( vod_t *, rtsp_cmd_type_t, vod_media_t *, const char *psz_session,
 231                           double f_arg, const char *psz_arg );
 232
 233 static rtsp_client_t *RtspClientNew( vod_media_t *, char * );
 234 static rtsp_client_t *RtspClientGet( vod_media_t *, const char * );
 235 static void           RtspClientDel( vod_media_t *, rtsp_client_t * );
 236
 237 static int RtspCallback( httpd_callback_sys_t *, httpd_client_t *,
 238                          httpd_message_t *, const httpd_message_t * );
 239 static int RtspCallbackES( httpd_callback_sys_t *, httpd_client_t *,
 240                            httpd_message_t *, const httpd_message_t * );
 241
 242 static char *SDPGenerate( const vod_media_t *, httpd_client_t *cl );
 243
 244 static void sprintf_hexa( char *s, uint8_t *p_data, int i_data )
 245 {
 246     static const char hex[16] = "0123456789abcdef";
 247     int i;
 248
 249     for( i = 0; i < i_data; i++ )
 250     {
 251         s[2*i+0] = hex[(p_data[i]>>4)&0xf];
 252         s[2*i+1] = hex[(p_data[i]   )&0xf];
 253     }
 254     s[2*i_data] = '\0';
 255 }
 256
 257 /*****************************************************************************
 258  * Open: Starts the RTSP server module
 259  *****************************************************************************/
 260 static int Open( vlc_object_t *p_this )
 261 {
 262     vod_t *p_vod = (vod_t *)p_this;
 263     vod_sys_t *p_sys = 0;
 264     char *psz_url = 0;
 265     vlc_url_t url;
 266
 267     psz_url = config_GetPsz( p_vod, "rtsp-host" );
 268     vlc_UrlParse( &url, psz_url, 0 );
 269     free( psz_url );
 270
 271     if( url.i_port <= 0 ) url.i_port = 554;
 272
 273     p_vod->p_sys = p_sys = malloc( sizeof( vod_sys_t ) );
 274     if( !p_sys ) goto error;
 275     p_sys->p_rtsp_host = 0;
 276
 277     p_sys->i_session_timeout = var_CreateGetInteger( p_this, "rtsp-session-timeout" );
 278
 279     p_sys->i_throttle_users = var_CreateGetInteger( p_this, "rtsp-throttle-users" );
 280     msg_Dbg( p_this, "allowing up to %d connections", p_sys->i_throttle_users );
 281     p_sys->i_connections = 0;
 282
 283     p_sys->psz_raw_mux = var_CreateGetString( p_this, "rtsp-raw-mux" );
 284
 285     p_sys->p_rtsp_host =
 286         httpd_HostNew( VLC_OBJECT(p_vod), url.psz_host, url.i_port );
 287     if( !p_sys->p_rtsp_host )
 288     {
 289         msg_Err( p_vod, "cannot create RTSP server (%s:%i)",
 290                  url.psz_host, url.i_port );
 291         goto error;
 292     }
 293
 294     p_sys->psz_path = strdup( url.psz_path ? url.psz_path : "/" );
 295     p_sys->i_port = url.i_port;
 296
 297     vlc_UrlClean( &url );
 298
 299     vlc_mutex_init( &p_sys->lock_media );
 300
 301     TAB_INIT( p_sys->i_media, p_sys->media );
 302     p_sys->i_media_id = 0;
 303
 304     p_vod->pf_media_new = MediaNew;
 305     p_vod->pf_media_del = MediaDel;
 306     p_vod->pf_media_add_es = MediaAddES;
 307     p_vod->pf_media_del_es = MediaDelES;
 308
 309     p_sys->p_fifo_cmd = block_FifoNew();
 310     if( vlc_thread_create( p_vod, "rtsp vod thread", CommandThread,
 311                            VLC_THREAD_PRIORITY_LOW, false ) )
 312     {
 313         msg_Err( p_vod, "cannot spawn rtsp vod thread" );
 314         block_FifoRelease( p_sys->p_fifo_cmd );
 315         free( p_sys->psz_path );
 316         goto error;
 317     }
 318
 319     return VLC_SUCCESS;
 320
 321 error:
 322     if( p_sys )
 323     {
 324         if( p_sys->p_rtsp_host ) httpd_HostDelete( p_sys->p_rtsp_host );
 325         free( p_sys->psz_raw_mux );
 326         free( p_sys );
 327     }
 328     vlc_UrlClean( &url );
 329
 330     return VLC_EGENERIC;
 331 }
 332
 333 /*****************************************************************************
 334  * Close:
 335  *****************************************************************************/
 336 static void Close( vlc_object_t * p_this )
 337 {
 338     vod_t *p_vod = (vod_t *)p_this;
 339     vod_sys_t *p_sys = p_vod->p_sys;
 340
 341     /* Stop command thread */
 342     vlc_object_kill( p_vod );
 343     CommandPush( p_vod, RTSP_CMD_TYPE_NONE, NULL, NULL, 0.0, NULL );
 344     vlc_thread_join( p_vod );
 345
 346     block_FifoRelease( p_sys->p_fifo_cmd );
 347
 348     httpd_HostDelete( p_sys->p_rtsp_host );
 349     var_Destroy( p_this, "rtsp-session-timeout" );
 350     var_Destroy( p_this, "rtsp-throttle-users" );
 351     var_Destroy( p_this, "rtsp-raw-mux" );
 352
 353     /* Check VLM is not buggy */
 354     if( p_sys->i_media > 0 )
 355         msg_Err( p_vod, "rtsp vod leaking %d medias", p_sys->i_media );
 356     TAB_CLEAN( p_sys->i_media, p_sys->media );
 357
 358     vlc_mutex_destroy( &p_sys->lock_media );
 359
 360     free( p_sys->psz_path );
 361     free( p_sys->psz_raw_mux );
 362     free( p_sys );
 363 }
 364
 365 /*****************************************************************************
 366  * Media handling
 367  *****************************************************************************/
 368 static vod_media_t *MediaNew( vod_t *p_vod, const char *psz_name,
 369                               input_item_t *p_item )
 370 {
 371     vod_sys_t *p_sys = p_vod->p_sys;
 372     vod_media_t *p_media = malloc( sizeof(vod_media_t) );
 373     int i;
 374
 375     if( !p_media )
 376         return NULL;
 377
 378     memset( p_media, 0, sizeof(vod_media_t) );
 379     p_media->id = p_sys->i_media_id++;
 380     TAB_INIT( p_media->i_es, p_media->es );
 381     p_media->psz_mux = 0;
 382     TAB_INIT( p_media->i_rtsp, p_media->rtsp );
 383     p_media->b_raw = false;
 384
 385     if( asprintf( &p_media->psz_rtsp_path, "%s%s",
 386                   p_sys->psz_path, psz_name ) <0 )
 387         return NULL;
 388     p_media->p_rtsp_url =
 389         httpd_UrlNewUnique( p_sys->p_rtsp_host, p_media->psz_rtsp_path, NULL,
 390                             NULL, NULL );
 391
 392     if( !p_media->p_rtsp_url )
 393     {
 394         msg_Err( p_vod, "cannot create RTSP url (%s)", p_media->psz_rtsp_path);
 395         free( p_media->psz_rtsp_path );
 396         free( p_media );
 397         return NULL;
 398     }
 399
 400     msg_Dbg( p_vod, "created RTSP url: %s", p_media->psz_rtsp_path );
 401
 402     if( asprintf( &p_media->psz_rtsp_control_v4,
 403                "a=control:rtsp://%%s:%d%s/trackID=%%d\r\n",
 404                p_sys->i_port, p_media->psz_rtsp_path ) < 0 )
 405     {
 406         httpd_UrlDelete( p_media->p_rtsp_url );
 407         free( p_media->psz_rtsp_path );
 408         free( p_media );
 409         return NULL;
 410     }
 411     if( asprintf( &p_media->psz_rtsp_control_v6,
 412                "a=control:rtsp://[%%s]:%d%s/trackID=%%d\r\n",
 413               p_sys->i_port, p_media->psz_rtsp_path ) < 0 )
 414     {
 415         httpd_UrlDelete( p_media->p_rtsp_url );
 416         free( p_media->psz_rtsp_path );
 417         free( p_media );
 418         return NULL;
 419     }
 420
 421     httpd_UrlCatch( p_media->p_rtsp_url, HTTPD_MSG_SETUP,
 422                     RtspCallback, (void*)p_media );
 423     httpd_UrlCatch( p_media->p_rtsp_url, HTTPD_MSG_DESCRIBE,
 424                     RtspCallback, (void*)p_media );
 425     httpd_UrlCatch( p_media->p_rtsp_url, HTTPD_MSG_PLAY,
 426                     RtspCallback, (void*)p_media );
 427     httpd_UrlCatch( p_media->p_rtsp_url, HTTPD_MSG_PAUSE,
 428                     RtspCallback, (void*)p_media );
 429     httpd_UrlCatch( p_media->p_rtsp_url, HTTPD_MSG_GETPARAMETER,
 430                     RtspCallback, (void*)p_media );
 431     httpd_UrlCatch( p_media->p_rtsp_url, HTTPD_MSG_TEARDOWN,
 432                     RtspCallback, (void*)p_media );
 433
 434     p_media->p_vod = p_vod;
 435
 436     vlc_mutex_lock( &p_sys->lock_media );
 437     TAB_APPEND( p_sys->i_media, p_sys->media, p_media );
 438     vlc_mutex_unlock( &p_sys->lock_media );
 439
 440     vlc_mutex_init( &p_media->lock );
 441     p_media->psz_session_name = strdup("");
 442     p_media->psz_session_description = strdup("");
 443     p_media->psz_session_url = strdup("");
 444     p_media->psz_session_email = strdup("");
 445
 446     p_media->i_port_audio = 1234;
 447     p_media->i_port_video = 1236;
 448     p_media->i_port       = 1238;
 449     p_media->i_payload_type = 96;
 450
 451     p_media->i_sdp_id = mdate();
 452     p_media->i_sdp_version = 1;
 453     p_media->i_length = input_item_GetDuration( p_item );
 454
 455     vlc_mutex_lock( &p_item->lock );
 456     msg_Dbg( p_vod, "media has %i declared ES", p_item->i_es );
 457     for( i = 0; i < p_item->i_es; i++ )
 458     {
 459         MediaAddES( p_vod, p_media, p_item->es[i] );
 460     }
 461     vlc_mutex_unlock( &p_item->lock );
 462
 463     return p_media;
 464 }
 465
 466 static void MediaDel( vod_t *p_vod, vod_media_t *p_media )
 467 {
 468     vod_sys_t *p_sys = p_vod->p_sys;
 469
 470     msg_Dbg( p_vod, "deleting media: %s", p_media->psz_rtsp_path );
 471
 472     vlc_mutex_lock( &p_sys->lock_media );
 473     TAB_REMOVE( p_sys->i_media, p_sys->media, p_media );
 474     vlc_mutex_unlock( &p_sys->lock_media );
 475
 476     while( p_media->i_rtsp > 0 )
 477         RtspClientDel( p_media, p_media->rtsp[0] );
 478     TAB_CLEAN( p_media->i_rtsp, p_media->rtsp );
 479
 480     httpd_UrlDelete( p_media->p_rtsp_url );
 481     free( p_media->psz_rtsp_path );
 482     free( p_media->psz_rtsp_control_v6 );
 483     free( p_media->psz_rtsp_control_v4 );
 484
 485     while( p_media->i_es )
 486         MediaDelES( p_vod, p_media, &p_media->es[0]->fmt );
 487     TAB_CLEAN( p_media->i_es, p_media->es );
 488
 489     vlc_mutex_destroy( &p_media->lock );
 490
 491     free( p_media->psz_session_name );
 492     free( p_media->psz_session_description );
 493     free( p_media->psz_session_url );
 494     free( p_media->psz_session_email );
 495     free( p_media->psz_mux );
 496     free( p_media );
 497 }
 498
 499 static int MediaAddES( vod_t *p_vod, vod_media_t *p_media, es_format_t *p_fmt )
 500 {
 501     media_es_t *p_es = malloc( sizeof(media_es_t) );
 502     char *psz_urlc;
 503
 504     if( !p_es ) return VLC_ENOMEM;
 505     memset( p_es, 0, sizeof(media_es_t) );
 506
 507     free( p_media->psz_mux );
 508     p_media->psz_mux = NULL;
 509
 510     /* TODO: update SDP, etc... */
 511     if( asprintf( &psz_urlc, "%s/trackID=%d",
 512               p_media->psz_rtsp_path, p_media->i_es ) < 0 )
 513     {
 514         free( p_es );
 515         return VLC_ENOMEM;
 516     }
 517     msg_Dbg( p_vod, "  - ES %4.4s (%s)", (char *)&p_fmt->i_codec, psz_urlc );
 518
 519     switch( p_fmt->i_codec )
 520     {
 521         case VLC_FOURCC( 's', '1', '6', 'b' ):
 522             if( p_fmt->audio.i_channels == 1 && p_fmt->audio.i_rate == 44100 )
 523             {
 524                 p_es->i_payload_type = 11;
 525             }
 526             else if( p_fmt->audio.i_channels == 2 &&
 527                      p_fmt->audio.i_rate == 44100 )
 528             {
 529                 p_es->i_payload_type = 10;
 530             }
 531             else
 532             {
 533                 p_es->i_payload_type = p_media->i_payload_type++;
 534             }
 535             if( asprintf( &p_es->psz_rtpmap, "L16/%d/%d", p_fmt->audio.i_rate,
 536                           p_fmt->audio.i_channels ) == -1 )
 537                 p_es->psz_rtpmap = NULL;
 538             break;
 539         case VLC_FOURCC( 'u', '8', ' ', ' ' ):
 540             p_es->i_payload_type = p_media->i_payload_type++;
 541             if( asprintf( &p_es->psz_rtpmap, "L8/%d/%d", p_fmt->audio.i_rate,
 542                           p_fmt->audio.i_channels ) == -1 )
 543                 p_es->psz_rtpmap = NULL;
 544             break;
 545         case VLC_FOURCC( 'm', 'p', 'g', 'a' ):
 546         case VLC_FOURCC( 'm', 'p', '3', ' ' ):
 547             p_es->i_payload_type = 14;
 548             p_es->psz_rtpmap = strdup( "MPA/90000" );
 549             break;
 550         case VLC_FOURCC( 'm', 'p', 'g', 'v' ):
 551             p_es->i_payload_type = 32;
 552             p_es->psz_rtpmap = strdup( "MPV/90000" );
 553             break;
 554         case VLC_FOURCC( 'a', '5', '2', ' ' ):
 555             p_es->i_payload_type = p_media->i_payload_type++;
 556             if( asprintf( &p_es->psz_rtpmap, "ac3/%d", p_fmt->audio.i_rate ) == -1 )
 557                 p_es->psz_rtpmap = NULL;
 558             break;
 559         case VLC_FOURCC( 'H', '2', '6', '3' ):
 560             p_es->i_payload_type = p_media->i_payload_type++;
 561             p_es->psz_rtpmap = strdup( "H263-1998/90000" );
 562             break;
 563         case VLC_FOURCC( 'h', '2', '6', '4' ):
 564             p_es->i_payload_type = p_media->i_payload_type++;
 565             p_es->psz_rtpmap = strdup( "H264/90000" );
 566             p_es->psz_fmtp = NULL;
 567             /* FIXME AAAAAAAAAAAARRRRRRRRGGGG copied from stream_out/rtp.c */
 568             if( p_fmt->i_extra > 0 )
 569             {
 570                 uint8_t *p_buffer = p_fmt->p_extra;
 571                 int     i_buffer = p_fmt->i_extra;
 572                 char    *p_64_sps = NULL;
 573                 char    *p_64_pps = NULL;
 574                 char    hexa[6+1];
 575
 576                 while( i_buffer > 4 &&
 577                        p_buffer[0] == 0 && p_buffer[1] == 0 &&
 578                        p_buffer[2] == 0 && p_buffer[3] == 1 )
 579                 {
 580                     const int i_nal_type = p_buffer[4]&0x1f;
 581                     int i_offset;
 582                     int i_size      = 0;
 583
 584                     i_size = i_buffer;
 585                     for( i_offset = 4; i_offset+3 < i_buffer ; i_offset++)
 586                     {
 587                         if( p_buffer[i_offset] == 0 && p_buffer[i_offset+1] == 0 && p_buffer[i_offset+2] == 0 && p_buffer[i_offset+3] == 1 )
 588                         {
 589                             /* we found another startcode */
 590                             i_size = i_offset;
 591                             break;
 592                         }
 593                     }
 594                     if( i_nal_type == 7 )
 595                     {
 596                         p_64_sps = vlc_b64_encode_binary( &p_buffer[4], i_size - 4 );
 597                         sprintf_hexa( hexa, &p_buffer[5], 3 );
 598                     }
 599                     else if( i_nal_type == 8 )
 600                     {
 601                         p_64_pps = vlc_b64_encode_binary( &p_buffer[4], i_size - 4 );
 602                     }
 603                     i_buffer -= i_size;
 604                     p_buffer += i_size;
 605                 }
 606                 /* */
 607                 if( p_64_sps && p_64_pps )
 608                 {
 609                     if( asprintf( &p_es->psz_fmtp,
 610                                   "packetization-mode=1;profile-level-id=%s;"
 611                                   "sprop-parameter-sets=%s,%s;", hexa, p_64_sps,
 612                                   p_64_pps ) < 0 )
 613                     {
 614                         free( p_64_sps );
 615                         free( p_64_pps );
 616                         free( psz_urlc );
 617                         free( p_es );
 618                         return VLC_ENOMEM;
 619                     }
 620                 }
 621                 free( p_64_sps );
 622                 free( p_64_pps );
 623             }
 624             if( !p_es->psz_fmtp )
 625                 p_es->psz_fmtp = strdup( "packetization-mode=1" );
 626             break;
 627         case VLC_FOURCC( 'm', 'p', '4', 'v' ):
 628             p_es->i_payload_type = p_media->i_payload_type++;
 629             p_es->psz_rtpmap = strdup( "MP4V-ES/90000" );
 630             if( p_fmt->i_extra > 0 )
 631             {
 632                 char *p_hexa = malloc( 2 * p_fmt->i_extra + 1 );
 633                 sprintf_hexa( p_hexa, p_fmt->p_extra, p_fmt->i_extra );
 634                 if( asprintf( &p_es->psz_fmtp,
 635                               "profile-level-id=3; config=%s;", p_hexa ) == -1 )
 636                     p_es->psz_fmtp = NULL;
 637                 free( p_hexa );
 638             }
 639             break;
 640         case VLC_FOURCC( 'm', 'p', '4', 'a' ):
 641             p_es->i_payload_type = p_media->i_payload_type++;
 642             if( asprintf( &p_es->psz_rtpmap, "mpeg4-generic/%d", p_fmt->audio.i_rate ) == -1 )
 643                 p_es->psz_rtpmap = NULL;
 644             if( p_fmt->i_extra > 0 )
 645             {
 646                 char *p_hexa = malloc( 2 * p_fmt->i_extra + 1 );
 647                 sprintf_hexa( p_hexa, p_fmt->p_extra, p_fmt->i_extra );
 648                 if( asprintf( &p_es->psz_fmtp,
 649                               "streamtype=5; profile-level-id=15; mode=AAC-hbr; "
 650                               "config=%s; SizeLength=13;IndexLength=3; "
 651                               "IndexDeltaLength=3; Profile=1;", p_hexa ) == -1 )
 652                     p_es->psz_fmtp = NULL;
 653                 free( p_hexa );
 654             }
 655             break;
 656         case VLC_FOURCC( 'm', 'p', '2', 't' ):
 657             p_media->psz_mux = strdup("ts");
 658             p_es->i_payload_type = 33;
 659             p_es->psz_rtpmap = strdup( "MP2T/90000" );
 660             break;
 661         case VLC_FOURCC( 'm', 'p', '2', 'p' ):
 662             p_media->psz_mux = strdup("ps");
 663             p_es->i_payload_type = p_media->i_payload_type++;
 664             p_es->psz_rtpmap = strdup( "MP2P/90000" );
 665             break;
 666         case VLC_FOURCC( 's', 'a', 'm', 'r' ):
 667             p_es->i_payload_type = p_media->i_payload_type++;
 668             p_es->psz_rtpmap = strdup( p_fmt->audio.i_channels == 2 ?
 669                                     "AMR/8000/2" : "AMR/8000" );
 670             p_es->psz_fmtp = strdup( "octet-align=1" );
 671             break;
 672         case VLC_FOURCC( 's', 'a', 'w', 'b' ):
 673             p_es->i_payload_type = p_media->i_payload_type++;
 674             p_es->psz_rtpmap = strdup( p_fmt->audio.i_channels == 2 ?
 675                                     "AMR-WB/16000/2" : "AMR-WB/16000" );
 676             p_es->psz_fmtp = strdup( "octet-align=1" );
 677             break;
 678
 679         default:
 680             msg_Err( p_vod, "cannot add this stream (unsupported "
 681                     "codec: %4.4s)", (char*)&p_fmt->i_codec );
 682             free( psz_urlc );
 683             free( p_es );
 684             return VLC_EGENERIC;
 685     }
 686
 687     p_es->p_rtsp_url =
 688         httpd_UrlNewUnique( p_vod->p_sys->p_rtsp_host, psz_urlc, NULL, NULL,
 689                             NULL );
 690
 691     if( !p_es->p_rtsp_url )
 692     {
 693         msg_Err( p_vod, "cannot create RTSP url (%s)", psz_urlc );
 694         free( psz_urlc );
 695         free( p_es );
 696         return VLC_EGENERIC;
 697     }
 698     free( psz_urlc );
 699
 700     httpd_UrlCatch( p_es->p_rtsp_url, HTTPD_MSG_SETUP,
 701                     RtspCallbackES, (void*)p_es );
 702     httpd_UrlCatch( p_es->p_rtsp_url, HTTPD_MSG_TEARDOWN,
 703                     RtspCallbackES, (void*)p_es );
 704     httpd_UrlCatch( p_es->p_rtsp_url, HTTPD_MSG_PLAY,
 705                     RtspCallbackES, (void*)p_es );
 706     httpd_UrlCatch( p_es->p_rtsp_url, HTTPD_MSG_PAUSE,
 707                     RtspCallbackES, (void*)p_es );
 708
 709     es_format_Copy( &p_es->fmt, p_fmt );
 710     p_es->p_vod = p_vod;
 711     p_es->p_media = p_media;
 712
 713 #if 0
 714     /* Choose the port */
 715     if( p_fmt->i_cat == AUDIO_ES && p_media->i_port_audio > 0 )
 716     {
 717         p_es->i_port = p_media->i_port_audio;
 718         p_media->i_port_audio = 0;
 719     }
 720     else if( p_fmt->i_cat == VIDEO_ES && p_media->i_port_video > 0 )
 721     {
 722         p_es->i_port = p_media->i_port_video;
 723         p_media->i_port_video = 0;
 724     }
 725     while( !p_es->i_port )
 726     {
 727         if( p_media->i_port != p_media->i_port_audio &&
 728             p_media->i_port != p_media->i_port_video )
 729         {
 730             p_es->i_port = p_media->i_port;
 731             p_media->i_port += 2;
 732             break;
 733         }
 734         p_media->i_port += 2;
 735     }
 736 #else
 737
 738     p_es->i_port = 0;
 739 #endif
 740
 741     vlc_mutex_lock( &p_media->lock );
 742     TAB_APPEND( p_media->i_es, p_media->es, p_es );
 743     vlc_mutex_unlock( &p_media->lock );
 744
 745     p_media->i_sdp_version++;
 746
 747     return VLC_SUCCESS;
 748 }
 749
 750 static void MediaDelES( vod_t *p_vod, vod_media_t *p_media, es_format_t *p_fmt)
 751 {
 752     media_es_t *p_es = NULL;
 753     int i;
 754
 755     /* Find the ES */
 756     for( i = 0; i < p_media->i_es; i++ )
 757     {
 758         if( p_media->es[i]->fmt.i_cat == p_fmt->i_cat &&
 759             p_media->es[i]->fmt.i_codec == p_fmt->i_codec &&
 760             p_media->es[i]->fmt.i_id == p_fmt->i_id )
 761         {
 762             p_es = p_media->es[i];
 763         }
 764     }
 765     if( !p_es ) return;
 766
 767     msg_Dbg( p_vod, "  - Removing ES %4.4s", (char *)&p_fmt->i_codec );
 768
 769     vlc_mutex_lock( &p_media->lock );
 770     TAB_REMOVE( p_media->i_es, p_media->es, p_es );
 771     vlc_mutex_unlock( &p_media->lock );
 772
 773     free( p_es->psz_rtpmap );
 774     free( p_es->psz_fmtp );
 775     p_media->i_sdp_version++;
 776
 777     if( p_es->p_rtsp_url ) httpd_UrlDelete( p_es->p_rtsp_url );
 778     es_format_Clean( &p_es->fmt );
 779     free( p_es );
 780 }
 781
 782 /* */
 783 typedef struct
 784 {
 785     int i_type;
 786     int i_media_id;
 787     //vod_media_t *p_media;
 788     char *psz_session;
 789     char *psz_arg;
 790     double f_arg;
 791 } rtsp_cmd_t;
 792
 793 static void CommandPush( vod_t *p_vod, rtsp_cmd_type_t i_type, vod_media_t *p_media, const char *psz_session,
 794                          double f_arg, const char *psz_arg )
 795 {
 796     rtsp_cmd_t cmd;
 797     block_t *p_cmd;
 798
 799     memset( &cmd, 0, sizeof(cmd) );
 800     cmd.i_type = i_type;
 801     if( p_media )
 802         cmd.i_media_id = p_media->id;
 803     if( psz_session )
 804         cmd.psz_session = strdup(psz_session);
 805     cmd.f_arg = f_arg;
 806     if( psz_arg )
 807         cmd.psz_arg = strdup(psz_arg);
 808
 809     p_cmd = block_New( p_vod, sizeof(rtsp_cmd_t) );
 810     memcpy( p_cmd->p_buffer, &cmd, sizeof(cmd) );
 811
 812     block_FifoPut( p_vod->p_sys->p_fifo_cmd, p_cmd );
 813 }
 814
 815 static void* CommandThread( vlc_object_t *p_this )
 816 {
 817     vod_t *p_vod = (vod_t*)p_this;
 818     vod_sys_t *p_sys = p_vod->p_sys;
 819
 820     while( vlc_object_alive (p_vod) )
 821     {
 822         block_t *p_block_cmd = block_FifoGet( p_sys->p_fifo_cmd );
 823         rtsp_cmd_t cmd;
 824         vod_media_t *p_media = NULL;
 825         int i;
 826
 827         if( !p_block_cmd )
 828             break;
 829
 830         memcpy( &cmd, p_block_cmd->p_buffer, sizeof(cmd) );
 831         block_Release( p_block_cmd );
 832
 833         if( cmd.i_type == RTSP_CMD_TYPE_NONE )
 834             break;
 835
 836         /* */
 837         vlc_mutex_lock( &p_sys->lock_media );
 838         for( i = 0; i < p_sys->i_media; i++ )
 839         {
 840             if( p_sys->media[i]->id == cmd.i_media_id )
 841                 break;
 842         }
 843         if( i >= p_sys->i_media )
 844             goto next;
 845         p_media = p_sys->media[i];
 846
 847         switch( cmd.i_type )
 848         {
 849         case RTSP_CMD_TYPE_PLAY:
 850             vod_MediaControl( p_vod, p_media, cmd.psz_session,
 851                               VOD_MEDIA_PLAY, cmd.psz_arg );
 852             break;
 853         case RTSP_CMD_TYPE_PAUSE:
 854             vod_MediaControl( p_vod, p_media, cmd.psz_session,
 855                               VOD_MEDIA_PAUSE );
 856             break;
 857
 858         case RTSP_CMD_TYPE_STOP:
 859             vod_MediaControl( p_vod, p_media, cmd.psz_session, VOD_MEDIA_STOP );
 860             break;
 861
 862         case RTSP_CMD_TYPE_SEEK:
 863             vod_MediaControl( p_vod, p_media, cmd.psz_session,
 864                               VOD_MEDIA_SEEK, cmd.f_arg );
 865             break;
 866
 867         case RTSP_CMD_TYPE_REWIND:
 868             vod_MediaControl( p_vod, p_media, cmd.psz_session,
 869                               VOD_MEDIA_REWIND, cmd.f_arg );
 870             break;
 871
 872         case RTSP_CMD_TYPE_FORWARD:
 873             vod_MediaControl( p_vod, p_media, cmd.psz_session,
 874                               VOD_MEDIA_FORWARD, cmd.f_arg );
 875             break;
 876
 877         default:
 878             break;
 879         }
 880
 881     next:
 882         vlc_mutex_unlock( &p_sys->lock_media );
 883         free( cmd.psz_session );
 884         free( cmd.psz_arg );
 885     }
 886     return NULL;
 887 }
 888
 889 /****************************************************************************
 890  * RTSP server implementation
 891  ****************************************************************************/
 892 static rtsp_client_t *RtspClientNew( vod_media_t *p_media, char *psz_session )
 893 {
 894     rtsp_client_t *p_rtsp = malloc( sizeof(rtsp_client_t) );
 895
 896     if( !p_rtsp ) return NULL;
 897     memset( p_rtsp, 0, sizeof(rtsp_client_t) );
 898     p_rtsp->es = 0;
 899
 900     p_rtsp->psz_session = psz_session;
 901     TAB_APPEND( p_media->i_rtsp, p_media->rtsp, p_rtsp );
 902
 903     p_media->p_vod->p_sys->i_connections++;
 904     msg_Dbg( p_media->p_vod, "new session: %s, connections: %d",
 905              psz_session, p_media->p_vod->p_sys->i_throttle_users );
 906
 907     return p_rtsp;
 908 }
 909
 910 static rtsp_client_t *RtspClientGet( vod_media_t *p_media, const char *psz_session )
 911 {
 912     int i;
 913
 914     for( i = 0; psz_session && i < p_media->i_rtsp; i++ )
 915     {
 916         if( !strcmp( p_media->rtsp[i]->psz_session, psz_session ) )
 917             return p_media->rtsp[i];
 918     }
 919
 920     return NULL;
 921 }
 922
 923 static void RtspClientDel( vod_media_t *p_media, rtsp_client_t *p_rtsp )
 924 {
 925     p_media->p_vod->p_sys->i_connections--;
 926     msg_Dbg( p_media->p_vod, "closing session: %s, connections: %d",
 927              p_rtsp->psz_session, p_media->p_vod->p_sys->i_throttle_users );
 928
 929     while( p_rtsp->i_es-- )
 930     {
 931         free( p_rtsp->es[p_rtsp->i_es]->psz_ip );
 932         free( p_rtsp->es[p_rtsp->i_es] );
 933         if( !p_rtsp->i_es ) free( p_rtsp->es );
 934     }
 935
 936     TAB_REMOVE( p_media->i_rtsp, p_media->rtsp, p_rtsp );
 937
 938     free( p_rtsp->psz_session );
 939     free( p_rtsp );
 940 }
 941
 942
 943 static float ParseNPT (const char *str)
 944 {
 945      locale_t loc = newlocale (LC_NUMERIC_MASK, "C", NULL);
 946      locale_t oldloc = uselocale (loc);
 947      unsigned hour, min;
 948      float sec;
 949
 950      if (sscanf (str, "%u:%u:%f", &hour, &min, &sec) == 3)
 951          sec += ((hour * 60) + min) * 60;
 952      else
 953      if (sscanf (str, "%f", &sec) != 1)
 954          sec = 0.;
 955
 956      if (loc != (locale_t)0)
 957      {
 958          uselocale (oldloc);
 959          freelocale (loc);
 960      }
 961      return sec;
 962 }
 963
 964
 965 static int RtspCallback( httpd_callback_sys_t *p_args, httpd_client_t *cl,
 966                          httpd_message_t *answer, const httpd_message_t *query )
 967 {
 968     vod_media_t *p_media = (vod_media_t*)p_args;
 969     vod_t *p_vod = p_media->p_vod;
 970     const char *psz_transport = NULL;
 971     const char *psz_playnow = NULL; /* support option: x-playNow */
 972     const char *psz_session = NULL;
 973     const char *psz_cseq = NULL;
 974     rtsp_client_t *p_rtsp;
 975     int i_port = 0;
 976     int i_cseq = 0;
 977
 978     if( answer == NULL || query == NULL ) return VLC_SUCCESS;
 979
 980     msg_Dbg( p_vod, "RtspCallback query: type=%d", query->i_type );
 981
 982     answer->i_proto   = HTTPD_PROTO_RTSP;
 983     answer->i_version = query->i_version;
 984     answer->i_type    = HTTPD_MSG_ANSWER;
 985     answer->i_body    = 0;
 986     answer->p_body    = NULL;
 987
 988     switch( query->i_type )
 989     {
 990         case HTTPD_MSG_SETUP:
 991         {
 992             psz_playnow = httpd_MsgGet( query, "x-playNow" );
 993             psz_transport = httpd_MsgGet( query, "Transport" );
 994             if( psz_transport == NULL )
 995             {
 996                 answer->i_status = 400;
 997                 break;
 998             }
 999             msg_Dbg( p_vod, "HTTPD_MSG_SETUP: transport=%s", psz_transport );
1000
1001             if( strstr( psz_transport, "unicast" ) &&
1002                 strstr( psz_transport, "client_port=" ) )
1003             {
1004                 rtsp_client_t *p_rtsp = NULL;
1005                 char ip[NI_MAXNUMERICHOST];
1006                 i_port = atoi( strstr( psz_transport, "client_port=" ) +
1007                                 strlen("client_port=") );
1008
1009                 if( strstr( psz_transport, "MP2T/H2221/UDP" ) ||
1010                     strstr( psz_transport, "RAW/RAW/UDP" ) )
1011                 {
1012                     free( p_media->psz_mux );
1013                     p_media->psz_mux = NULL;
1014                     p_media->psz_mux = strdup( p_vod->p_sys->psz_raw_mux );
1015                     p_media->b_raw = true;
1016                 }
1017
1018                 if( httpd_ClientIP( cl, ip ) == NULL )
1019                 {
1020                     answer->i_status = 500;
1021                     answer->i_body = 0;
1022                     answer->p_body = NULL;
1023                     break;
1024                 }
1025
1026                 msg_Dbg( p_vod, "HTTPD_MSG_SETUP: unicast ip=%s port=%d",
1027                          ip, i_port );
1028
1029                 psz_session = httpd_MsgGet( query, "Session" );
1030                 if( !psz_session || !*psz_session )
1031                 {
1032                     char *psz_new;
1033                     if( ( p_vod->p_sys->i_throttle_users > 0 ) &&
1034                         ( p_vod->p_sys->i_connections >= p_vod->p_sys->i_throttle_users ) )
1035                     {
1036                         answer->i_status = 503;
1037                         answer->i_body = 0;
1038                         answer->p_body = NULL;
1039                         break;
1040                     }
1041                     if( asprintf( &psz_new, "%d", rand() ) < 0 )
1042                         return VLC_ENOMEM;
1043                     psz_session = psz_new;
1044
1045                     p_rtsp = RtspClientNew( p_media, psz_new );
1046                     if( !p_rtsp )
1047                     {
1048                         answer->i_status = 454;
1049                         answer->i_body = 0;
1050                         answer->p_body = NULL;
1051                         break;
1052                     }
1053                 }
1054                 else
1055                 {
1056                     p_rtsp = RtspClientGet( p_media, psz_session );
1057                     if( !p_rtsp )
1058                     {
1059                         answer->i_status = 454;
1060                         answer->i_body = 0;
1061                         answer->p_body = NULL;
1062                         break;
1063                     }
1064                 }
1065
1066                 answer->i_status = 200;
1067                 answer->i_body = 0;
1068                 answer->p_body = NULL;
1069
1070                 if( p_media->b_raw )
1071                 {
1072                     if( strstr( psz_transport, "MP2T/H2221/UDP" ) )
1073                     {
1074                         httpd_MsgAdd( answer, "Transport",
1075                                       "MP2T/H2221/UDP;unicast;client_port=%d-%d",
1076                                       i_port, i_port + 1 );
1077                     }
1078                     else if( strstr( psz_transport, "RAW/RAW/UDP" ) )
1079                     {
1080                         httpd_MsgAdd( answer, "Transport",
1081                                       "RAW/RAW/UDP;unicast;client_port=%d-%d",
1082                                       i_port, i_port + 1 );
1083                     }
1084                 }
1085                 else
1086                     httpd_MsgAdd( answer, "Transport",
1087                                   "RTP/AVP/UDP;unicast;client_port=%d-%d",
1088                                   i_port, i_port + 1 );
1089             }
1090             else /* TODO  strstr( psz_transport, "interleaved" ) ) */
1091             {
1092                 answer->i_status = 461;
1093                 answer->i_body = 0;
1094                 answer->p_body = NULL;
1095             }
1096
1097             /* Intentional fall-through on x-playNow option in RTSP request */
1098             if( !psz_playnow )
1099                 break;
1100         }
1101
1102         case HTTPD_MSG_PLAY:
1103         {
1104             char *psz_output, ip[NI_MAXNUMERICHOST];
1105             int i, i_port_audio = 0, i_port_video = 0;
1106
1107             /* for now only multicast so easy */
1108             if( !psz_playnow )
1109             {
1110                 answer->i_status = 200;
1111                 answer->i_body = 0;
1112                 answer->p_body = NULL;
1113             }
1114
1115             if( !psz_session )
1116                 psz_session = httpd_MsgGet( query, "Session" );
1117             msg_Dbg( p_vod, "HTTPD_MSG_PLAY for session: %s", psz_session );
1118
1119             p_rtsp = RtspClientGet( p_media, psz_session );
1120             if( !p_rtsp )
1121             {
1122                 answer->i_status = 500;
1123                 answer->i_body = 0;
1124                 answer->p_body = NULL;
1125                 break;
1126             }
1127
1128             if( p_rtsp->b_playing )
1129             {
1130                 const char *psz_position = httpd_MsgGet( query, "Range" );
1131                 const char *psz_scale = httpd_MsgGet( query, "Scale" );
1132                 if( psz_position )
1133                     psz_position = strstr( psz_position, "npt=" );
1134                 if( psz_position && !psz_scale )
1135                 {
1136                     double f_pos = ParseNPT (psz_position + 4);
1137                     msg_Dbg( p_vod, "seeking request: %s", psz_position );
1138                     f_pos /= ((double)(p_media->i_length))/1000 /1000 / 100;
1139                     CommandPush( p_vod, RTSP_CMD_TYPE_SEEK, p_media,
1140                                  psz_session, f_pos, NULL );
1141                     break;
1142                 }
1143                 if( psz_scale )
1144                 {
1145                     double f_scale = 0.0;
1146                     char *end;
1147
1148                     f_scale = us_strtod( psz_scale, &end );
1149                     if( end > psz_scale )
1150                     {
1151                         f_scale = (f_scale * 30.0);
1152                         if( psz_scale[0] == '-' ) /* rewind */
1153                         {
1154                             msg_Dbg( p_vod, "rewind request: %s", psz_scale );
1155                             CommandPush( p_vod, RTSP_CMD_TYPE_REWIND, p_media,
1156                                          psz_session, f_scale, NULL );
1157                         }
1158                         else if(psz_scale[0] != '1' ) /* fast-forward */
1159                         {
1160                             msg_Dbg( p_vod, "fastforward request: %s",
1161                                      psz_scale );
1162                             CommandPush( p_vod, RTSP_CMD_TYPE_FORWARD, p_media,
1163                                          psz_session, f_scale, NULL );
1164                         }
1165
1166                         if( p_rtsp->b_paused == true )
1167                         {
1168                             p_rtsp->b_paused = false;
1169                             CommandPush( p_vod, RTSP_CMD_TYPE_PAUSE, p_media,
1170                                          psz_session, 0, NULL );
1171                         }
1172                     }
1173                     break;
1174                 }
1175             }
1176
1177             if( p_rtsp->b_playing && p_rtsp->b_paused )
1178             {
1179                 CommandPush( p_vod, RTSP_CMD_TYPE_PAUSE, p_media,
1180                              psz_session, 0, NULL );
1181                 p_rtsp->b_paused = false;
1182                 break;
1183             }
1184             else if( p_rtsp->b_playing ) break;
1185
1186             if( httpd_ClientIP( cl, ip ) == NULL ) break;
1187
1188             p_rtsp->b_playing = true;
1189
1190             /* FIXME for != 1 video and 1 audio */
1191             for( i = 0; i < p_rtsp->i_es; i++ )
1192             {
1193                 if( p_rtsp->es[i]->p_media_es->fmt.i_cat == AUDIO_ES )
1194                     i_port_audio = p_rtsp->es[i]->i_port;
1195                 if( p_rtsp->es[i]->p_media_es->fmt.i_cat == VIDEO_ES )
1196                     i_port_video = p_rtsp->es[i]->i_port;
1197             }
1198
1199             if( p_media->psz_mux )
1200             {
1201                 if( p_media->b_raw )
1202                 {
1203                     if( asprintf( &psz_output,
1204                               "std{access=udp,dst=%s:%i,mux=%s}",
1205                               ip, i_port, p_media->psz_mux ) < 0 )
1206                         return VLC_ENOMEM;
1207                 }
1208                 else
1209                 {
1210                     if( asprintf( &psz_output,
1211                               "rtp{dst=%s,port=%i,mux=%s}",
1212                               ip, i_port_video, p_media->psz_mux ) < 0 )
1213                         return VLC_ENOMEM;
1214                 }
1215             }
1216             else
1217             {
1218                 if( asprintf( &psz_output,
1219                               "rtp{dst=%s,port-video=%i,port-audio=%i}",
1220                               ip, i_port_video, i_port_audio ) < 0 )
1221                     return VLC_ENOMEM;
1222             }
1223
1224             CommandPush( p_vod, RTSP_CMD_TYPE_PLAY, p_media, psz_session,
1225                          0, psz_output );
1226             free( psz_output );
1227             break;
1228         }
1229
1230         case HTTPD_MSG_DESCRIBE:
1231         {
1232             char *psz_sdp =
1233                 SDPGenerate( p_media, cl );
1234
1235             if( psz_sdp != NULL )
1236             {
1237                 answer->i_status = 200;
1238                 httpd_MsgAdd( answer, "Content-type",  "%s",
1239                               "application/sdp" );
1240
1241                 answer->p_body = (uint8_t *)psz_sdp;
1242                 answer->i_body = strlen( psz_sdp );
1243             }
1244             else
1245             {
1246                 answer->i_status = 500;
1247                 answer->p_body = NULL;
1248                 answer->i_body = 0;
1249             }
1250             break;
1251         }
1252
1253         case HTTPD_MSG_PAUSE:
1254             psz_session = httpd_MsgGet( query, "Session" );
1255             msg_Dbg( p_vod, "HTTPD_MSG_PAUSE for session: %s", psz_session );
1256
1257             p_rtsp = RtspClientGet( p_media, psz_session );
1258             if( !p_rtsp ) break;
1259
1260             CommandPush( p_vod, RTSP_CMD_TYPE_PAUSE, p_media, psz_session,
1261                          0, NULL );
1262             p_rtsp->b_paused = true;
1263
1264             answer->i_status = 200;
1265             answer->i_body = 0;
1266             answer->p_body = NULL;
1267             break;
1268
1269         case HTTPD_MSG_TEARDOWN:
1270             /* for now only multicast so easy again */
1271             answer->i_status = 200;
1272             answer->i_body = 0;
1273             answer->p_body = NULL;
1274
1275             psz_session = httpd_MsgGet( query, "Session" );
1276             msg_Dbg( p_vod, "HTTPD_MSG_TEARDOWN for session: %s", psz_session);
1277
1278             p_rtsp = RtspClientGet( p_media, psz_session );
1279             if( !p_rtsp ) break;
1280
1281             CommandPush( p_vod, RTSP_CMD_TYPE_STOP, p_media, psz_session,
1282                          0, NULL );
1283             RtspClientDel( p_media, p_rtsp );
1284             break;
1285
1286         case HTTPD_MSG_GETPARAMETER:
1287             answer->i_status = 200;
1288             answer->i_body = 0;
1289             answer->p_body = NULL;
1290             break;
1291
1292         default:
1293             return VLC_EGENERIC;
1294     }
1295
1296     httpd_MsgAdd( answer, "Server", "VLC Server" );
1297     httpd_MsgAdd( answer, "Content-Length", "%d", answer->i_body );
1298     psz_cseq = httpd_MsgGet( query, "Cseq" );
1299     psz_cseq ? i_cseq = atoi( psz_cseq ) : 0;
1300     httpd_MsgAdd( answer, "CSeq", "%d", i_cseq );
1301     httpd_MsgAdd( answer, "Cache-Control", "%s", "no-cache" );
1302
1303     if( psz_session )
1304     {
1305          if( p_media->p_vod->p_sys->i_session_timeout >= 0 )
1306              httpd_MsgAdd( answer, "Session", "%s;timeout=%i", psz_session,
1307                p_media->p_vod->p_sys->i_session_timeout );
1308          else
1309               httpd_MsgAdd( answer, "Session", "%s", psz_session );
1310     }
1311
1312     return VLC_SUCCESS;
1313 }
1314
1315 static int RtspCallbackES( httpd_callback_sys_t *p_args, httpd_client_t *cl,
1316                            httpd_message_t *answer,
1317                            const httpd_message_t *query )
1318 {
1319     media_es_t *p_es = (media_es_t*)p_args;
1320     vod_media_t *p_media = p_es->p_media;
1321     vod_t *p_vod = p_media->p_vod;
1322     rtsp_client_t *p_rtsp = NULL;
1323     const char *psz_transport = NULL;
1324     const char *psz_playnow = NULL; /* support option: x-playNow */
1325     const char *psz_session = NULL;
1326     const char *psz_position = NULL;
1327     const char *psz_cseq = NULL;
1328     int i_cseq = 0;
1329     int i;
1330
1331     if( answer == NULL || query == NULL ) return VLC_SUCCESS;
1332
1333     msg_Dbg( p_vod, "RtspCallback query: type=%d", query->i_type );
1334
1335     answer->i_proto   = HTTPD_PROTO_RTSP;
1336     answer->i_version = query->i_version;
1337     answer->i_type    = HTTPD_MSG_ANSWER;
1338     answer->i_body    = 0;
1339     answer->p_body      = NULL;
1340
1341     switch( query->i_type )
1342     {
1343         case HTTPD_MSG_SETUP:
1344             psz_playnow = httpd_MsgGet( query, "x-playNow" );
1345             psz_transport = httpd_MsgGet( query, "Transport" );
1346
1347             msg_Dbg( p_vod, "HTTPD_MSG_SETUP: transport=%s", psz_transport );
1348
1349             if( strstr( psz_transport, "unicast" ) &&
1350                 strstr( psz_transport, "client_port=" ) )
1351             {
1352                 rtsp_client_t *p_rtsp = NULL;
1353                 rtsp_client_es_t *p_rtsp_es = NULL;
1354                 char ip[NI_MAXNUMERICHOST];
1355                 int i_port = atoi( strstr( psz_transport, "client_port=" ) +
1356                                    strlen("client_port=") );
1357
1358                 if( httpd_ClientIP( cl, ip ) == NULL )
1359                 {
1360                     answer->i_status = 500;
1361                     answer->i_body = 0;
1362                     answer->p_body = NULL;
1363                     break;
1364                 }
1365
1366                 msg_Dbg( p_vod, "HTTPD_MSG_SETUP: unicast ip=%s port=%d",
1367                         ip, i_port );
1368
1369                 psz_session = httpd_MsgGet( query, "Session" );
1370                 if( !psz_session || !*psz_session )
1371                 {
1372                     char *psz_new;
1373                     if( ( p_vod->p_sys->i_throttle_users > 0 ) &&
1374                         ( p_vod->p_sys->i_connections >= p_vod->p_sys->i_throttle_users ) )
1375                     {
1376                         answer->i_status = 503;
1377                         answer->i_body = 0;
1378                         answer->p_body = NULL;
1379                         break;
1380                     }
1381                     if( asprintf( &psz_new, "%d", rand() ) < 0 )
1382                         return VLC_ENOMEM;
1383                     psz_session = psz_new;
1384
1385                     p_rtsp = RtspClientNew( p_media, psz_new );
1386                 }
1387                 else
1388                 {
1389                     p_rtsp = RtspClientGet( p_media, psz_session );
1390                     if( !p_rtsp )
1391                     {
1392                         answer->i_status = 454;
1393                         answer->i_body = 0;
1394                         answer->p_body = NULL;
1395                         break;
1396                     }
1397                 }
1398
1399                 p_rtsp_es = malloc( sizeof(rtsp_client_es_t) );
1400                 if( !p_rtsp_es )
1401                 {
1402                     answer->i_status = 500;
1403                     answer->i_body = 0;
1404                     answer->p_body = NULL;
1405                     break;
1406                 }
1407                 p_rtsp_es->i_port = i_port;
1408                 p_rtsp_es->psz_ip = strdup( ip );
1409                 p_rtsp_es->p_media_es = p_es;
1410                 TAB_APPEND( p_rtsp->i_es, p_rtsp->es, p_rtsp_es );
1411
1412                 answer->i_status = 200;
1413                 answer->i_body = 0;
1414                 answer->p_body = NULL;
1415
1416                 if( p_media->b_raw )
1417                 {
1418                     if( strstr( psz_transport, "MP2T/H2221/UDP" ) )
1419                     {
1420                         httpd_MsgAdd( answer, "Transport",
1421                                      "MP2T/H2221/UDP;client_port=%d-%d",
1422                                      p_rtsp_es->i_port, p_rtsp_es->i_port + 1 );
1423                     }
1424                     else if( strstr( psz_transport, "RAW/RAW/UDP" ) )
1425                     {
1426                         httpd_MsgAdd( answer, "Transport",
1427                                      "RAW/RAW/UDP;client_port=%d-%d",
1428                                      p_rtsp_es->i_port, p_rtsp_es->i_port + 1 );
1429                     }
1430                 }
1431                 else
1432                 {
1433                     httpd_MsgAdd( answer, "Transport",
1434                                   "RTP/AVP/UDP;client_port=%d-%d",
1435                                   p_rtsp_es->i_port, p_rtsp_es->i_port + 1 );
1436                 }
1437             }
1438             else /* TODO  strstr( psz_transport, "interleaved" ) ) */
1439             {
1440                 answer->i_status = 461;
1441                 answer->i_body = 0;
1442                 answer->p_body = NULL;
1443             }
1444
1445             /* Intentional fall-through on x-playNow option in RTSP request */
1446             if( !psz_playnow )
1447                 break;
1448
1449         case HTTPD_MSG_PLAY:
1450             /* This is kind of a kludge. Should we only support Aggregate
1451              * Operations ? */
1452             psz_session = httpd_MsgGet( query, "Session" );
1453             msg_Dbg( p_vod, "HTTPD_MSG_PLAY for session: %s", psz_session );
1454
1455             p_rtsp = RtspClientGet( p_media, psz_session );
1456
1457             psz_position = httpd_MsgGet( query, "Range" );
1458             if( psz_position ) psz_position = strstr( psz_position, "npt=" );
1459             if( psz_position )
1460             {
1461                 double f_pos = ParseNPT (psz_position + 4);
1462                 msg_Dbg( p_vod, "seeking request: %s", psz_position );
1463                 f_pos /= ((double)(p_media->i_length))/1000 /1000 / 100;
1464                 CommandPush( p_vod, RTSP_CMD_TYPE_SEEK, p_media,
1465                              psz_session, f_pos, NULL );
1466             }
1467
1468             if( !psz_playnow )
1469             {
1470                 answer->i_status = 200;
1471                 answer->i_body = 0;
1472                 answer->p_body = NULL;
1473             }
1474             break;
1475
1476         case HTTPD_MSG_TEARDOWN:
1477             answer->i_status = 200;
1478             answer->i_body = 0;
1479             answer->p_body = NULL;
1480
1481             psz_session = httpd_MsgGet( query, "Session" );
1482             msg_Dbg( p_vod, "HTTPD_MSG_TEARDOWN for session: %s", psz_session);
1483
1484             p_rtsp = RtspClientGet( p_media, psz_session );
1485             if( !p_rtsp ) break;
1486
1487             for( i = 0; i < p_rtsp->i_es; i++ )
1488             {
1489                 if( p_rtsp->es[i]->p_media_es == p_es )
1490                 {
1491                     free( p_rtsp->es[i]->psz_ip );
1492                     TAB_REMOVE( p_rtsp->i_es, p_rtsp->es, p_rtsp->es[i] );
1493                     break;
1494                 }
1495             }
1496
1497             if( !p_rtsp->i_es )
1498             {
1499                 CommandPush( p_vod, RTSP_CMD_TYPE_STOP, p_media, psz_session,
1500                              0, NULL );
1501                 RtspClientDel( p_media, p_rtsp );
1502             }
1503             break;
1504
1505         case HTTPD_MSG_PAUSE:
1506             /* This is kind of a kludge. Should we only support Aggregate
1507              * Operations ? */
1508             psz_session = httpd_MsgGet( query, "Session" );
1509             msg_Dbg( p_vod, "HTTPD_MSG_PAUSE for session: %s", psz_session );
1510
1511             p_rtsp = RtspClientGet( p_media, psz_session );
1512             if( !p_rtsp ) break;
1513
1514             CommandPush( p_vod, RTSP_CMD_TYPE_PAUSE, p_media, psz_session,
1515                          0, NULL );
1516             p_rtsp->b_paused = true;
1517
1518             answer->i_status = 200;
1519             answer->i_body = 0;
1520             answer->p_body = NULL;
1521             break;
1522
1523         default:
1524             return VLC_EGENERIC;
1525             break;
1526     }
1527
1528     httpd_MsgAdd( answer, "Server", "VLC Server" );
1529     httpd_MsgAdd( answer, "Content-Length", "%d", answer->i_body );
1530     psz_cseq = httpd_MsgGet( query, "Cseq" );
1531     if (psz_cseq)
1532         i_cseq = atoi( psz_cseq );
1533     else
1534         i_cseq = 0;
1535     httpd_MsgAdd( answer, "Cseq", "%d", i_cseq );
1536     httpd_MsgAdd( answer, "Cache-Control", "%s", "no-cache" );
1537
1538     if( psz_session )
1539         httpd_MsgAdd( answer, "Session", "%s"/*;timeout=5*/, psz_session );
1540
1541     return VLC_SUCCESS;
1542 }
1543
1544 /*****************************************************************************
1545  * SDPGenerate: TODO
1546  * FIXME: need to be moved to a common place ?
1547  *****************************************************************************/
1548 static char *SDPGenerate( const vod_media_t *p_media, httpd_client_t *cl )
1549 {
1550     int i, i_size;
1551     char *p, *psz_sdp, ip[NI_MAXNUMERICHOST], ipv;
1552     const char *psz_control;
1553
1554     if( httpd_ServerIP( cl, ip ) == NULL )
1555         return NULL;
1556
1557     p = strchr( ip, '%' );
1558     if( p != NULL )
1559         *p = '\0'; /* remove scope if present */
1560
1561     ipv = ( strchr( ip, ':' ) != NULL ) ? '6' : '4';
1562
1563     /* Calculate size */
1564     i_size = sizeof( "v=0\r\n" ) +
1565         sizeof( "o=- * * IN IP4 \r\n" ) + 10 + NI_MAXNUMERICHOST +
1566         sizeof( "s=*\r\n" ) + strlen( p_media->psz_session_name ) +
1567         sizeof( "i=*\r\n" ) + strlen( p_media->psz_session_description ) +
1568         sizeof( "u=*\r\n" ) + strlen( p_media->psz_session_url ) +
1569         sizeof( "e=*\r\n" ) + strlen( p_media->psz_session_email ) +
1570         sizeof( "c=IN IP4 0.0.0.0\r\n" ) + 20 + 10 +
1571         sizeof( "t=0 0\r\n" ) + /* FIXME */
1572         sizeof( "a=tool:"PACKAGE_STRING"\r\n" ) +
1573         sizeof( "a=range:npt=0-1000000000.000\r\n" );
1574
1575     psz_control = (ipv == '6') ? p_media->psz_rtsp_control_v6
1576                                : p_media->psz_rtsp_control_v4;
1577     for( i = 0; i < p_media->i_es; i++ )
1578     {
1579         media_es_t *p_es = p_media->es[i];
1580
1581         i_size += sizeof( "m=**d*o * RTP/AVP *\r\n" ) + 19;
1582         if( p_es->psz_rtpmap )
1583         {
1584             i_size += sizeof( "a=rtpmap:* *\r\n" ) +
1585                 strlen( p_es->psz_rtpmap ) + 9;
1586         }
1587         if( p_es->psz_fmtp )
1588         {
1589             i_size += sizeof( "a=fmtp:* *\r\n" ) +
1590                 strlen( p_es->psz_fmtp ) + 9;
1591         }
1592     }
1593     i_size += (strlen( psz_control ) + strlen( ip ) + 9) * p_media->i_es;
1594
1595     p = psz_sdp = malloc( i_size );
1596     p += sprintf( p, "v=0\r\n" );
1597     p += sprintf( p, "o=- %"PRId64" %d IN IP%c %s\r\n",
1598                   p_media->i_sdp_id, p_media->i_sdp_version, ipv, ip );
1599     if( *p_media->psz_session_name )
1600         p += sprintf( p, "s=%s\r\n", p_media->psz_session_name );
1601     if( *p_media->psz_session_description )
1602         p += sprintf( p, "i=%s\r\n", p_media->psz_session_description );
1603     if( *p_media->psz_session_url )
1604         p += sprintf( p, "u=%s\r\n", p_media->psz_session_url );
1605     if( *p_media->psz_session_email )
1606         p += sprintf( p, "e=%s\r\n", p_media->psz_session_email );
1607
1608     p += sprintf( p, "c=IN IP%c %s\r\n", ipv, ipv == '6' ? "::" : "0.0.0.0" );
1609     p += sprintf( p, "t=0 0\r\n" ); /* FIXME */
1610     p += sprintf( p, "a=tool:"PACKAGE_STRING"\r\n" );
1611
1612     if( p_media->i_length > 0 )
1613     {
1614         lldiv_t d = lldiv( p_media->i_length / 1000, 1000 );
1615         p += sprintf( p, "a=range:npt=0-%lld.%03u\r\n", d.quot,
1616                       (unsigned)d.rem );
1617     }
1618
1619     for( i = 0; i < p_media->i_es; i++ )
1620     {
1621         media_es_t *p_es = p_media->es[i];
1622
1623         if( p_es->fmt.i_cat == AUDIO_ES )
1624         {
1625             p += sprintf( p, "m=audio %d RTP/AVP %d\r\n",
1626                           p_es->i_port, p_es->i_payload_type );
1627         }
1628         else if( p_es->fmt.i_cat == VIDEO_ES )
1629         {
1630             p += sprintf( p, "m=video %d RTP/AVP %d\r\n",
1631                           p_es->i_port, p_es->i_payload_type );
1632         }
1633         else
1634         {
1635             continue;
1636         }
1637
1638         if( p_es->psz_rtpmap )
1639         {
1640             p += sprintf( p, "a=rtpmap:%d %s\r\n", p_es->i_payload_type,
1641                           p_es->psz_rtpmap );
1642         }
1643         if( p_es->psz_fmtp )
1644         {
1645             p += sprintf( p, "a=fmtp:%d %s\r\n", p_es->i_payload_type,
1646                           p_es->psz_fmtp );
1647         }
1648
1649         p += sprintf( p, psz_control, ip, i );
1650     }
1651
1652     return psz_sdp;
1653 }