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

   1 /*****************************************************************************
   2  * rtsp.c: rtsp VoD server module
   3  *****************************************************************************
   4  * Copyright (C) 2003-2004 VideoLAN
   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., 59 Temple Place - Suite 330, Boston, MA  02111, USA.
  23  *****************************************************************************/
  24
  25 /*****************************************************************************
  26  * Preamble
  27  *****************************************************************************/
  28 #include <stdlib.h>
  29
  30 #include <errno.h>
  31
  32 #include <vlc/vlc.h>
  33 #include <vlc/input.h>
  34 #include <vlc/sout.h>
  35
  36 #include "vlc_httpd.h"
  37 #include "vlc_vod.h"
  38 #include "network.h"
  39
  40 /*****************************************************************************
  41  * Module descriptor
  42  *****************************************************************************/
  43 static int  Open ( vlc_object_t * );
  44 static void Close( vlc_object_t * );
  45
  46 #define HOST_TEXT N_( "Host address" )
  47 #define HOST_LONGTEXT N_( \
  48     "You can set the address, port and path the rtsp interface will bind to." )
  49
  50 vlc_module_begin();
  51     set_description( _("RTSP VoD server") );
  52     set_category( CAT_SOUT );
  53     set_subcategory( SUBCAT_SOUT_VOD );
  54     set_capability( "vod server", 1 );
  55     set_callbacks( Open, Close );
  56     add_shortcut( "rtsp" );
  57     add_string ( "rtsp-host", NULL, NULL, HOST_TEXT, HOST_LONGTEXT, VLC_TRUE );
  58 vlc_module_end();
  59
  60 /*****************************************************************************
  61  * Exported prototypes
  62  *****************************************************************************/
  63
  64 typedef struct media_es_t media_es_t;
  65
  66 typedef struct
  67 {
  68     media_es_t *p_media_es;
  69     char *psz_ip;
  70     int i_port;
  71
  72 } rtsp_client_es_t;
  73
  74 typedef struct
  75 {
  76     char *psz_session;
  77     int64_t i_last; /* for timeout */
  78
  79     vlc_bool_t b_playing; /* is it in "play" state */
  80     vlc_bool_t b_paused; /* is it in "pause" state */
  81
  82     int i_es;
  83     rtsp_client_es_t **es;
  84
  85 } rtsp_client_t;
  86
  87 struct media_es_t
  88 {
  89     /* VoD server */
  90     vod_t *p_vod;
  91
  92     /* RTSP server */
  93     httpd_url_t *p_rtsp_url;
  94
  95     vod_media_t *p_media;
  96
  97     es_format_t fmt;
  98     int         i_port;
  99     uint8_t     i_payload_type;
 100     char        *psz_rtpmap;
 101     char        *psz_fmtp;
 102
 103 };
 104
 105 struct vod_media_t
 106 {
 107     /* VoD server */
 108     vod_t *p_vod;
 109
 110     /* RTSP server */
 111     httpd_url_t  *p_rtsp_url;
 112     char         *psz_rtsp_control;
 113     char         *psz_rtsp_path;
 114
 115     char *psz_destination;
 116     int  i_port;
 117     int  i_port_audio;
 118     int  i_port_video;
 119     int  i_ttl;
 120     int  i_payload_type;
 121
 122     int64_t i_sdp_id;
 123     int     i_sdp_version;
 124
 125     vlc_bool_t b_multicast;
 126
 127     vlc_mutex_t lock;
 128
 129     /* ES list */
 130     int        i_es;
 131     media_es_t **es;
 132     char       *psz_mux;
 133
 134     /* RTSP client */
 135     int           i_rtsp;
 136     rtsp_client_t **rtsp;
 137
 138     /* Infos */
 139     char *psz_session_name;
 140     char *psz_session_description;
 141     char *psz_session_url;
 142     char *psz_session_email;
 143     mtime_t i_length;
 144 };
 145
 146 struct vod_sys_t
 147 {
 148     /* RTSP server */
 149     httpd_host_t *p_rtsp_host;
 150     char *psz_host;
 151     char *psz_path;
 152     int i_port;
 153
 154     /* List of media */
 155     int i_media;
 156     vod_media_t **media;
 157 };
 158
 159 static vod_media_t *MediaNew( vod_t *, char *, input_item_t * );
 160 static void         MediaDel( vod_t *, vod_media_t * );
 161 static int          MediaAddES( vod_t *, vod_media_t *, es_format_t * );
 162 static void         MediaDelES( vod_t *, vod_media_t *, es_format_t * );
 163
 164 static rtsp_client_t *RtspClientNew( vod_media_t *, char * );
 165 static rtsp_client_t *RtspClientGet( vod_media_t *, char * );
 166 static void           RtspClientDel( vod_media_t *, rtsp_client_t * );
 167
 168 static int RtspCallback( httpd_callback_sys_t *, httpd_client_t *,
 169                          httpd_message_t *, httpd_message_t * );
 170 static int RtspCallbackES( httpd_callback_sys_t *, httpd_client_t *,
 171                            httpd_message_t *, httpd_message_t * );
 172
 173 static char *SDPGenerate( vod_media_t *, char * );
 174
 175 static void sprintf_hexa( char *s, uint8_t *p_data, int i_data )
 176 {
 177     static const char hex[16] = "0123456789abcdef";
 178     int i;
 179
 180     for( i = 0; i < i_data; i++ )
 181     {
 182         s[2*i+0] = hex[(p_data[i]>>4)&0xf];
 183         s[2*i+1] = hex[(p_data[i]   )&0xf];
 184     }
 185     s[2*i_data] = '\0';
 186 }
 187
 188 /*****************************************************************************
 189  * Open: Starts the RTSP server module
 190  *****************************************************************************/
 191 static int Open( vlc_object_t *p_this )
 192 {
 193     vod_t *p_vod = (vod_t *)p_this;
 194     vod_sys_t *p_sys = 0;
 195     char *psz_url = 0;
 196     vlc_url_t url;
 197
 198     psz_url = config_GetPsz( p_vod, "rtsp-host" );
 199     vlc_UrlParse( &url, psz_url, 0 );
 200     if( psz_url ) free( psz_url );
 201
 202     if( !url.psz_host || !*url.psz_host )
 203     {
 204         if( url.psz_host ) free( url.psz_host );
 205         url.psz_host = strdup( "localhost" );
 206     }
 207     if( url.i_port <= 0 ) url.i_port = 554;
 208
 209     p_vod->p_sys = p_sys = malloc( sizeof( vod_sys_t ) );
 210     if( !p_sys ) goto error;
 211     p_sys->p_rtsp_host = 0;
 212
 213     p_sys->p_rtsp_host =
 214         httpd_HostNew( VLC_OBJECT(p_vod), url.psz_host, url.i_port );
 215     if( !p_sys->p_rtsp_host )
 216     {
 217         msg_Err( p_vod, "cannot create http server (%s:%i)",
 218                  url.psz_host, url.i_port );
 219         goto error;
 220     }
 221
 222     p_sys->psz_host = strdup( url.psz_host );
 223     p_sys->psz_path = strdup( url.psz_path ? url.psz_path : "/" );
 224     p_sys->i_port = url.i_port;
 225
 226     vlc_UrlClean( &url );
 227     p_sys->media = 0;
 228     p_sys->i_media = 0;
 229
 230     p_vod->pf_media_new = MediaNew;
 231     p_vod->pf_media_del = MediaDel;
 232     p_vod->pf_media_add_es = MediaAddES;
 233     p_vod->pf_media_del_es = MediaDelES;
 234
 235     return VLC_SUCCESS;
 236
 237  error:
 238
 239     if( p_sys && p_sys->p_rtsp_host ) httpd_HostDelete( p_sys->p_rtsp_host );
 240     if( p_sys ) free( p_sys );
 241     vlc_UrlClean( &url );
 242     return VLC_EGENERIC;
 243 }
 244
 245 /*****************************************************************************
 246  * Close:
 247  *****************************************************************************/
 248 static void Close( vlc_object_t * p_this )
 249 {
 250     vod_t *p_vod = (vod_t *)p_this;
 251     vod_sys_t *p_sys = p_vod->p_sys;
 252
 253     httpd_HostDelete( p_sys->p_rtsp_host );
 254
 255     /* TODO delete medias */
 256
 257     free( p_sys );
 258 }
 259
 260 /*****************************************************************************
 261  * Media handling
 262  *****************************************************************************/
 263 static vod_media_t *MediaNew( vod_t *p_vod, char *psz_name,
 264                               input_item_t *p_item )
 265 {
 266     vod_sys_t *p_sys = p_vod->p_sys;
 267     vod_media_t *p_media = malloc( sizeof(vod_media_t) );
 268     int i;
 269
 270     memset( p_media, 0, sizeof(vod_media_t) );
 271     p_media->es = 0;
 272     p_media->psz_mux = 0;
 273     p_media->rtsp = 0;
 274
 275     asprintf( &p_media->psz_rtsp_path, "%s%s", p_sys->psz_path, psz_name );
 276     p_media->p_rtsp_url =
 277         httpd_UrlNewUnique( p_sys->p_rtsp_host, p_media->psz_rtsp_path, 0, 0 );
 278
 279     if( !p_media->p_rtsp_url )
 280     {
 281         msg_Err( p_vod, "cannot create http url (%s)", p_media->psz_rtsp_path);
 282         free( p_media->psz_rtsp_path );
 283         free( p_media );
 284         return 0;
 285     }
 286
 287     msg_Dbg( p_vod, "created rtsp url: %s", p_media->psz_rtsp_path );
 288
 289     asprintf( &p_media->psz_rtsp_control, "rtsp://%s:%d%s",
 290               p_sys->psz_host, p_sys->i_port, p_media->psz_rtsp_path );
 291
 292     httpd_UrlCatch( p_media->p_rtsp_url, HTTPD_MSG_DESCRIBE,
 293                     RtspCallback, (void*)p_media );
 294     httpd_UrlCatch( p_media->p_rtsp_url, HTTPD_MSG_PLAY,
 295                     RtspCallback, (void*)p_media );
 296     httpd_UrlCatch( p_media->p_rtsp_url, HTTPD_MSG_PAUSE,
 297                     RtspCallback, (void*)p_media );
 298     httpd_UrlCatch( p_media->p_rtsp_url, HTTPD_MSG_TEARDOWN,
 299                     RtspCallback, (void*)p_media );
 300
 301     p_media->p_vod = p_vod;
 302
 303     TAB_APPEND( p_sys->i_media, p_sys->media, p_media );
 304
 305     vlc_mutex_init( p_vod, &p_media->lock );
 306     p_media->psz_session_name = strdup("");
 307     p_media->psz_session_description = strdup("");
 308     p_media->psz_session_url = strdup("");
 309     p_media->psz_session_email = strdup("");
 310
 311     p_media->i_port_audio = 1234;
 312     p_media->i_port_video = 1236;
 313     p_media->i_port       = 1238;
 314     p_media->i_payload_type = 96;
 315
 316     p_media->i_sdp_id = mdate();
 317     p_media->i_sdp_version = 1;
 318     p_media->i_length = p_item->i_duration;
 319
 320     vlc_mutex_lock( &p_item->lock );
 321     msg_Dbg( p_vod, "media has %i declared ES", p_item->i_es );
 322     for( i = 0; i < p_item->i_es; i++ )
 323     {
 324         MediaAddES( p_vod, p_media, p_item->es[i] );
 325     }
 326     vlc_mutex_unlock( &p_item->lock );
 327
 328     return p_media;
 329 }
 330
 331 static void MediaDel( vod_t *p_vod, vod_media_t *p_media )
 332 {
 333     vod_sys_t *p_sys = p_vod->p_sys;
 334
 335     msg_Dbg( p_vod, "deleting media: %s", p_media->psz_rtsp_path );
 336
 337     while( p_media->i_rtsp > 0 ) RtspClientDel( p_media, p_media->rtsp[0] );
 338     httpd_UrlDelete( p_media->p_rtsp_url );
 339     if( p_media->psz_rtsp_path ) free( p_media->psz_rtsp_path );
 340     if( p_media->psz_rtsp_control ) free( p_media->psz_rtsp_control );
 341
 342     TAB_REMOVE( p_sys->i_media, p_sys->media, p_media );
 343
 344     while( p_media->i_es ) MediaDelES( p_vod, p_media, &p_media->es[0]->fmt );
 345
 346     vlc_mutex_destroy( &p_media->lock );
 347     free( p_media );
 348 }
 349
 350 static int MediaAddES( vod_t *p_vod, vod_media_t *p_media, es_format_t *p_fmt )
 351 {
 352     media_es_t *p_es = malloc( sizeof(media_es_t) );
 353     char *psz_urlc;
 354
 355     memset( p_es, 0, sizeof(media_es_t) );
 356     p_media->psz_mux = NULL;
 357
 358     /* TODO: update SDP, etc... */
 359     asprintf( &psz_urlc, "%s/trackid=%d",
 360               p_media->psz_rtsp_path, p_media->i_es );
 361     msg_Dbg( p_vod, "  - ES %4.4s (%s)", (char *)&p_fmt->i_codec, psz_urlc );
 362
 363     switch( p_fmt->i_codec )
 364     {
 365     case VLC_FOURCC( 's', '1', '6', 'b' ):
 366         if( p_fmt->audio.i_channels == 1 && p_fmt->audio.i_rate == 44100 )
 367         {
 368             p_es->i_payload_type = 11;
 369         }
 370         else if( p_fmt->audio.i_channels == 2 && p_fmt->audio.i_rate == 44100 )
 371         {
 372             p_es->i_payload_type = 10;
 373         }
 374         else
 375         {
 376             p_es->i_payload_type = p_media->i_payload_type++;
 377         }
 378
 379         p_es->psz_rtpmap = malloc( strlen( "L16/*/*" ) + 20+1 );
 380         sprintf( p_es->psz_rtpmap, "L16/%d/%d", p_fmt->audio.i_rate,
 381                  p_fmt->audio.i_channels );
 382         break;
 383     case VLC_FOURCC( 'u', '8', ' ', ' ' ):
 384         p_es->i_payload_type = p_media->i_payload_type++;
 385         p_es->psz_rtpmap = malloc( strlen( "L8/*/*" ) + 20+1 );
 386         sprintf( p_es->psz_rtpmap, "L8/%d/%d", p_fmt->audio.i_rate,
 387                  p_fmt->audio.i_channels );
 388         break;
 389     case VLC_FOURCC( 'm', 'p', 'g', 'a' ):
 390         p_es->i_payload_type = 14;
 391         p_es->psz_rtpmap = strdup( "MPA/90000" );
 392         break;
 393     case VLC_FOURCC( 'm', 'p', 'g', 'v' ):
 394         p_es->i_payload_type = 32;
 395         p_es->psz_rtpmap = strdup( "MPV/90000" );
 396         break;
 397     case VLC_FOURCC( 'a', '5', '2', ' ' ):
 398         p_es->i_payload_type = p_media->i_payload_type++;
 399         p_es->psz_rtpmap = strdup( "ac3/90000" );
 400         break;
 401     case VLC_FOURCC( 'H', '2', '6', '3' ):
 402         p_es->i_payload_type = p_media->i_payload_type++;
 403         p_es->psz_rtpmap = strdup( "H263-1998/90000" );
 404         break;
 405     case VLC_FOURCC( 'm', 'p', '4', 'v' ):
 406         p_es->i_payload_type = p_media->i_payload_type++;
 407         p_es->psz_rtpmap = strdup( "MP4V-ES/90000" );
 408         if( p_fmt->i_extra > 0 )
 409         {
 410             char *p_hexa = malloc( 2 * p_fmt->i_extra + 1 );
 411             p_es->psz_fmtp = malloc( 100 + 2 * p_fmt->i_extra );
 412             sprintf_hexa( p_hexa, p_fmt->p_extra, p_fmt->i_extra );
 413             sprintf( p_es->psz_fmtp,
 414                      "profile-level-id=3; config=%s;", p_hexa );
 415             free( p_hexa );
 416         }
 417         break;
 418     case VLC_FOURCC( 'm', 'p', '4', 'a' ):
 419         p_es->i_payload_type = p_media->i_payload_type++;
 420         p_es->psz_rtpmap = malloc( strlen( "mpeg4-generic/" ) + 12 );
 421         sprintf( p_es->psz_rtpmap, "mpeg4-generic/%d", p_fmt->audio.i_rate );
 422         if( p_fmt->i_extra > 0 )
 423         {
 424             char *p_hexa = malloc( 2 * p_fmt->i_extra + 1 );
 425             p_es->psz_fmtp = malloc( 200 + 2 * p_fmt->i_extra );
 426             sprintf_hexa( p_hexa, p_fmt->p_extra, p_fmt->i_extra );
 427             sprintf( p_es->psz_fmtp,
 428                      "streamtype=5; profile-level-id=15; mode=AAC-hbr; "
 429                      "config=%s; SizeLength=13;IndexLength=3; "
 430                      "IndexDeltaLength=3; Profile=1;", p_hexa );
 431             free( p_hexa );
 432         }
 433         break;
 434     case VLC_FOURCC( 'm', 'p', '2', 't' ):
 435         p_media->psz_mux = "ts";
 436         p_es->i_payload_type = 33;
 437         p_es->psz_rtpmap = strdup( "MP2T/90000" );
 438         break;
 439     case VLC_FOURCC( 'm', 'p', '2', 'p' ):
 440         p_media->psz_mux = "ps";
 441         p_es->i_payload_type = p_media->i_payload_type++;
 442         p_es->psz_rtpmap = strdup( "MP2P/90000" );
 443         break;
 444
 445     default:
 446         msg_Err( p_vod, "cannot add this stream (unsupported "
 447                  "codec: %4.4s)", (char*)&p_fmt->i_codec );
 448         free( p_es );
 449         return VLC_EGENERIC;
 450     }
 451
 452     p_es->p_rtsp_url =
 453         httpd_UrlNewUnique( p_vod->p_sys->p_rtsp_host, psz_urlc, 0, 0 );
 454
 455     if( !p_es->p_rtsp_url )
 456     {
 457         msg_Err( p_vod, "cannot create http url (%s)", psz_urlc );
 458         free( psz_urlc );
 459         free( p_es );
 460         return VLC_EGENERIC;
 461     }
 462     free( psz_urlc );
 463
 464     httpd_UrlCatch( p_es->p_rtsp_url, HTTPD_MSG_SETUP,
 465                     RtspCallbackES, (void*)p_es );
 466     httpd_UrlCatch( p_es->p_rtsp_url, HTTPD_MSG_TEARDOWN,
 467                     RtspCallbackES, (void*)p_es );
 468     httpd_UrlCatch( p_es->p_rtsp_url, HTTPD_MSG_PLAY,
 469                     RtspCallbackES, (void*)p_es );
 470     httpd_UrlCatch( p_es->p_rtsp_url, HTTPD_MSG_PAUSE,
 471                     RtspCallbackES, (void*)p_es );
 472
 473     es_format_Copy( &p_es->fmt, p_fmt );
 474     p_es->p_vod = p_vod;
 475     p_es->p_media = p_media;
 476
 477 #if 0
 478     /* Choose the port */
 479     if( p_fmt->i_cat == AUDIO_ES && p_media->i_port_audio > 0 )
 480     {
 481         p_es->i_port = p_media->i_port_audio;
 482         p_media->i_port_audio = 0;
 483     }
 484     else if( p_fmt->i_cat == VIDEO_ES && p_media->i_port_video > 0 )
 485     {
 486         p_es->i_port = p_media->i_port_video;
 487         p_media->i_port_video = 0;
 488     }
 489     while( !p_es->i_port )
 490     {
 491         if( p_media->i_port != p_media->i_port_audio &&
 492             p_media->i_port != p_media->i_port_video )
 493         {
 494             p_es->i_port = p_media->i_port;
 495             p_media->i_port += 2;
 496             break;
 497         }
 498         p_media->i_port += 2;
 499     }
 500 #else
 501
 502     p_es->i_port = 0;
 503 #endif
 504
 505     vlc_mutex_lock( &p_media->lock );
 506     TAB_APPEND( p_media->i_es, p_media->es, p_es );
 507     vlc_mutex_unlock( &p_media->lock );
 508
 509     p_media->i_sdp_version++;
 510
 511     return VLC_SUCCESS;
 512 }
 513
 514 static void MediaDelES( vod_t *p_vod, vod_media_t *p_media, es_format_t *p_fmt)
 515 {
 516     media_es_t *p_es = 0;
 517     int i;
 518
 519     /* Find the ES */
 520     for( i = 0; i < p_media->i_es; i++ )
 521     {
 522         if( p_media->es[i]->fmt.i_cat == p_fmt->i_cat &&
 523             p_media->es[i]->fmt.i_codec == p_fmt->i_codec &&
 524             p_media->es[i]->fmt.i_id == p_fmt->i_id )
 525         {
 526             p_es = p_media->es[i];
 527         }
 528     }
 529     if( !p_es ) return;
 530
 531     msg_Dbg( p_vod, "  - Removing ES %4.4s", (char *)&p_fmt->i_codec );
 532
 533     vlc_mutex_lock( &p_media->lock );
 534     TAB_REMOVE( p_media->i_es, p_media->es, p_es );
 535     vlc_mutex_unlock( &p_media->lock );
 536
 537     if( p_es->psz_rtpmap ) free( p_es->psz_rtpmap );
 538     if( p_es->psz_fmtp ) free( p_es->psz_fmtp );
 539     p_media->i_sdp_version++;
 540
 541     if( p_es->p_rtsp_url ) httpd_UrlDelete( p_es->p_rtsp_url );
 542     es_format_Clean( &p_es->fmt );
 543 }
 544
 545 /****************************************************************************
 546  * RTSP server implementation
 547  ****************************************************************************/
 548 static rtsp_client_t *RtspClientNew( vod_media_t *p_media, char *psz_session )
 549 {
 550     rtsp_client_t *p_rtsp = malloc( sizeof(rtsp_client_t) );
 551     memset( p_rtsp, 0, sizeof(rtsp_client_t) );
 552     p_rtsp->es = 0;
 553
 554     p_rtsp->psz_session = psz_session;
 555     TAB_APPEND( p_media->i_rtsp, p_media->rtsp, p_rtsp );
 556
 557     msg_Dbg( p_media->p_vod, "new session: %s", psz_session );
 558
 559     return p_rtsp;
 560 }
 561
 562 static rtsp_client_t *RtspClientGet( vod_media_t *p_media, char *psz_session )
 563 {
 564     int i;
 565
 566     for( i = 0; psz_session && i < p_media->i_rtsp; i++ )
 567     {
 568         if( !strcmp( p_media->rtsp[i]->psz_session, psz_session ) )
 569         {
 570             return p_media->rtsp[i];
 571         }
 572     }
 573
 574     return NULL;
 575 }
 576
 577 static void RtspClientDel( vod_media_t *p_media, rtsp_client_t *p_rtsp )
 578 {
 579     msg_Dbg( p_media->p_vod, "closing session: %s", p_rtsp->psz_session );
 580
 581     while( p_rtsp->i_es-- )
 582     {
 583         if( p_rtsp->es[p_rtsp->i_es]->psz_ip )
 584             free( p_rtsp->es[p_rtsp->i_es]->psz_ip );
 585         free( p_rtsp->es[p_rtsp->i_es] );
 586         if( !p_rtsp->i_es ) free( p_rtsp->es );
 587     }
 588
 589     TAB_REMOVE( p_media->i_rtsp, p_media->rtsp, p_rtsp );
 590
 591     free( p_rtsp->psz_session );
 592     free( p_rtsp );
 593 }
 594
 595 static int RtspCallback( httpd_callback_sys_t *p_args, httpd_client_t *cl,
 596                          httpd_message_t *answer, httpd_message_t *query )
 597 {
 598     vod_media_t *p_media = (vod_media_t*)p_args;
 599     vod_t *p_vod = p_media->p_vod;
 600     char *psz_destination = p_media->psz_destination;
 601     char *psz_session = NULL;
 602     rtsp_client_t *p_rtsp;
 603
 604     if( answer == NULL || query == NULL ) return VLC_SUCCESS;
 605
 606     fprintf( stderr, "RtspCallback query: type=%d\n", query->i_type );
 607
 608     answer->i_proto   = HTTPD_PROTO_RTSP;
 609     answer->i_version = query->i_version;
 610     answer->i_type    = HTTPD_MSG_ANSWER;
 611
 612     switch( query->i_type )
 613     {
 614         case HTTPD_MSG_DESCRIBE:
 615         {
 616             char *psz_sdp =
 617                 SDPGenerate( p_media, psz_destination ?
 618                              psz_destination : "0.0.0.0" );
 619
 620             answer->i_status = 200;
 621             answer->psz_status = strdup( "OK" );
 622             httpd_MsgAdd( answer, "Content-type",  "%s", "application/sdp" );
 623
 624             answer->p_body = psz_sdp;
 625             answer->i_body = strlen( psz_sdp );
 626             break;
 627         }
 628
 629         case HTTPD_MSG_PLAY:
 630         {
 631             char *psz_output, *ip;
 632             int i, i_port_audio = 0, i_port_video = 0;
 633
 634             /* for now only multicast so easy */
 635             answer->i_status = 200;
 636             answer->psz_status = strdup( "OK" );
 637             answer->i_body = 0;
 638             answer->p_body = NULL;
 639
 640             psz_session = httpd_MsgGet( query, "Session" );
 641             msg_Dbg( p_vod, "HTTPD_MSG_PLAY for session: %s", psz_session );
 642
 643             p_rtsp = RtspClientGet( p_media, psz_session );
 644             if( !p_rtsp ) break;
 645
 646             if( p_rtsp->b_playing && p_rtsp->b_paused )
 647             {
 648                 vod_MediaControl( p_vod, p_media, psz_session,
 649                                   VOD_MEDIA_PAUSE );
 650                 p_rtsp->b_paused = VLC_FALSE;
 651                 break;
 652             }
 653             else if( p_rtsp->b_playing ) break;
 654
 655             if( !(ip = httpd_ClientIP( cl )) ) break;
 656
 657             p_rtsp->b_playing = VLC_TRUE;
 658
 659             /* FIXME for != 1 video and 1 audio */
 660             for( i = 0; i < p_rtsp->i_es; i++ )
 661             {
 662                 if( p_rtsp->es[i]->p_media_es->fmt.i_cat == AUDIO_ES )
 663                     i_port_audio = p_rtsp->es[i]->i_port;
 664                 if( p_rtsp->es[i]->p_media_es->fmt.i_cat == VIDEO_ES )
 665                     i_port_video = p_rtsp->es[i]->i_port;
 666             }
 667
 668             if( p_media->psz_mux )
 669             {
 670                 asprintf( &psz_output, "rtp{dst=%s,port=%i,mux=%s}",
 671                           ip, i_port_video, p_media->psz_mux );
 672             }
 673             else
 674             {
 675                 asprintf( &psz_output, "rtp{dst=%s,port-video=%i,"
 676                           "port-audio=%i}", ip, i_port_video, i_port_audio );
 677             }
 678
 679             vod_MediaControl( p_vod, p_media, psz_session, VOD_MEDIA_PLAY,
 680                               psz_output );
 681             free( psz_output );
 682             free( ip );
 683             break;
 684         }
 685
 686         case HTTPD_MSG_PAUSE:
 687             psz_session = httpd_MsgGet( query, "Session" );
 688             msg_Dbg( p_vod, "HTTPD_MSG_PAUSE for session: %s", psz_session );
 689
 690             p_rtsp = RtspClientGet( p_media, psz_session );
 691             if( !p_rtsp ) break;
 692
 693             vod_MediaControl( p_vod, p_media, psz_session, VOD_MEDIA_PAUSE );
 694             p_rtsp->b_paused = VLC_TRUE;
 695
 696             answer->i_status = 200;
 697             answer->psz_status = strdup( "OK" );
 698             answer->i_body = 0;
 699             answer->p_body = NULL;
 700             break;
 701
 702         case HTTPD_MSG_TEARDOWN:
 703             /* for now only multicast so easy again */
 704             answer->i_status = 200;
 705             answer->psz_status = strdup( "OK" );
 706             answer->i_body = 0;
 707             answer->p_body = NULL;
 708
 709             psz_session = httpd_MsgGet( query, "Session" );
 710             msg_Dbg( p_vod, "HTTPD_MSG_TEARDOWN for session: %s", psz_session);
 711
 712             p_rtsp = RtspClientGet( p_media, psz_session );
 713             if( !p_rtsp ) break;
 714
 715             vod_MediaControl( p_vod, p_media, psz_session, VOD_MEDIA_STOP );
 716             RtspClientDel( p_media, p_rtsp );
 717             break;
 718
 719         default:
 720             return VLC_EGENERIC;
 721     }
 722
 723     httpd_MsgAdd( answer, "Server", "VLC Server" );
 724     httpd_MsgAdd( answer, "Content-Length", "%d", answer->i_body );
 725     httpd_MsgAdd( answer, "Cseq", "%d",
 726                   atoi( httpd_MsgGet( query, "Cseq" ) ) );
 727     httpd_MsgAdd( answer, "Cache-Control", "%s", "no-cache" );
 728
 729     if( psz_session )
 730     {
 731         httpd_MsgAdd( answer, "Session", "%s;timeout=5", psz_session );
 732     }
 733
 734     return VLC_SUCCESS;
 735 }
 736
 737 static int RtspCallbackES( httpd_callback_sys_t *p_args, httpd_client_t *cl,
 738                            httpd_message_t *answer, httpd_message_t *query )
 739 {
 740     media_es_t *p_es = (media_es_t*)p_args;
 741     vod_media_t *p_media = p_es->p_media;
 742     vod_t *p_vod = p_media->p_vod;
 743     rtsp_client_t *p_rtsp = NULL;
 744     char *psz_session = NULL;
 745     char *psz_transport = NULL;
 746     char *psz_position = NULL;
 747     int i;
 748
 749     if( answer == NULL || query == NULL ) return VLC_SUCCESS;
 750
 751     fprintf( stderr, "RtspCallback query: type=%d\n", query->i_type );
 752
 753     answer->i_proto   = HTTPD_PROTO_RTSP;
 754     answer->i_version = query->i_version;
 755     answer->i_type    = HTTPD_MSG_ANSWER;
 756
 757     switch( query->i_type )
 758     {
 759     case HTTPD_MSG_SETUP:
 760         psz_transport = httpd_MsgGet( query, "Transport" );
 761         fprintf( stderr, "HTTPD_MSG_SETUP: transport=%s\n", psz_transport );
 762
 763         if( strstr( psz_transport, "multicast" ) && p_media->psz_destination )
 764         {
 765             fprintf( stderr, "HTTPD_MSG_SETUP: multicast\n" );
 766             answer->i_status = 200;
 767             answer->psz_status = strdup( "OK" );
 768             answer->i_body = 0;
 769             answer->p_body = NULL;
 770
 771             psz_session = httpd_MsgGet( query, "Session" );
 772             if( !psz_session || !*psz_session )
 773             {
 774                 asprintf( &psz_session, "%d", rand() );
 775             }
 776
 777             httpd_MsgAdd( answer, "Transport",
 778                           "RTP/AVP/UDP;destination=%s;port=%d-%d;ttl=%d",
 779                           p_media->psz_destination, p_media->i_port,
 780                           p_media->i_port+1, p_media->i_ttl );
 781         }
 782         else if( strstr( psz_transport, "unicast" ) &&
 783                  strstr( psz_transport, "client_port=" ) )
 784         {
 785             rtsp_client_t *p_rtsp;
 786             rtsp_client_es_t *p_rtsp_es;
 787             char *ip = httpd_ClientIP( cl );
 788             int i_port = atoi( strstr( psz_transport, "client_port=" ) +
 789                                strlen("client_port=") );
 790
 791             if( !ip )
 792             {
 793                 answer->i_status = 400;
 794                 answer->psz_status = strdup( "Internal server error" );
 795                 answer->i_body = 0;
 796                 answer->p_body = NULL;
 797                 break;
 798             }
 799
 800             fprintf( stderr, "HTTPD_MSG_SETUP: unicast ip=%s port=%d\n",
 801                      ip, i_port );
 802
 803             psz_session = httpd_MsgGet( query, "Session" );
 804             if( !psz_session || !*psz_session )
 805             {
 806                 asprintf( &psz_session, "%d", rand() );
 807                 p_rtsp = RtspClientNew( p_media, psz_session );
 808             }
 809             else
 810             {
 811                 p_rtsp = RtspClientGet( p_media, psz_session );
 812                 if( !p_rtsp )
 813                 {
 814                     /* FIXME right error code */
 815                     answer->i_status = 400;
 816                     answer->psz_status = strdup( "Unknown session id" );
 817                     answer->i_body = 0;
 818                     answer->p_body = NULL;
 819                     free( ip );
 820                     break;
 821                 }
 822             }
 823
 824             p_rtsp_es = malloc( sizeof(rtsp_client_es_t) );
 825             p_rtsp_es->i_port = i_port;
 826             p_rtsp_es->psz_ip = strdup( ip );
 827             p_rtsp_es->p_media_es = p_es;
 828             TAB_APPEND( p_rtsp->i_es, p_rtsp->es, p_rtsp_es );
 829
 830             answer->i_status = 200;
 831             answer->psz_status = strdup( "OK" );
 832             answer->i_body = 0;
 833             answer->p_body = NULL;
 834
 835             httpd_MsgAdd( answer, "Transport", "RTP/AVP/UDP;client_port=%d-%d",
 836                           i_port, i_port + 1 );
 837         }
 838         else /* TODO  strstr( psz_transport, "interleaved" ) ) */
 839         {
 840             answer->i_status = 400;
 841             answer->psz_status = strdup( "Bad Request" );
 842             answer->i_body = 0;
 843             answer->p_body = NULL;
 844         }
 845         break;
 846
 847         case HTTPD_MSG_TEARDOWN:
 848             answer->i_status = 200;
 849             answer->psz_status = strdup( "OK" );
 850             answer->i_body = 0;
 851             answer->p_body = NULL;
 852
 853             psz_session = httpd_MsgGet( query, "Session" );
 854             msg_Dbg( p_vod, "HTTPD_MSG_TEARDOWN for session: %s", psz_session);
 855
 856             p_rtsp = RtspClientGet( p_media, psz_session );
 857             if( !p_rtsp ) break;
 858
 859             for( i = 0; i < p_rtsp->i_es; i++ )
 860             {
 861                 if( p_rtsp->es[i]->p_media_es == p_es )
 862                 {
 863                     if( p_rtsp->es[i]->psz_ip ) free( p_rtsp->es[i]->psz_ip );
 864                     TAB_REMOVE( p_rtsp->i_es, p_rtsp->es, p_rtsp->es[i] );
 865                     break;
 866                 }
 867             }
 868
 869             if( !p_rtsp->i_es )
 870             {
 871                 vod_MediaControl( p_vod, p_media, psz_session,
 872                                   VOD_MEDIA_STOP );
 873                 RtspClientDel( p_media, p_rtsp );
 874             }
 875             break;
 876
 877         case HTTPD_MSG_PLAY:
 878             psz_session = httpd_MsgGet( query, "Session" );
 879             msg_Dbg( p_vod, "HTTPD_MSG_PLAY for session: %s", psz_session );
 880
 881             p_rtsp = RtspClientGet( p_media, psz_session );
 882
 883             psz_position = httpd_MsgGet( query, "Range" );
 884             if( psz_position ) psz_position = strstr( psz_position, "npt=" );
 885             if( psz_position )
 886             {
 887                 float f_pos;
 888
 889                 msg_Dbg( p_vod, "seeking request: %s", psz_position );
 890
 891                 psz_position += 4;
 892                 if( sscanf( psz_position, "%f", &f_pos ) == 1 )
 893                 {
 894                     f_pos /= ((float)(p_media->i_length/1000))/1000 / 100;
 895                     vod_MediaControl( p_vod, p_media, psz_session,
 896                                       VOD_MEDIA_SEEK, (double)f_pos );
 897                 }
 898             }
 899
 900             answer->i_status = 460;
 901             answer->psz_status = strdup( "Only Aggregate Operation Allowed" );
 902             answer->i_body = 0;
 903             answer->p_body = NULL;
 904             break;
 905
 906         case HTTPD_MSG_PAUSE:
 907             psz_session = httpd_MsgGet( query, "Session" );
 908             msg_Dbg( p_vod, "HTTPD_MSG_PAUSE for session: %s", psz_session );
 909
 910             answer->i_status = 460;
 911             answer->psz_status = strdup( "Only Aggregate Operation Allowed" );
 912             answer->i_body = 0;
 913             answer->p_body = NULL;
 914             break;
 915
 916         default:
 917             return VLC_EGENERIC;
 918             break;
 919     }
 920
 921     httpd_MsgAdd( answer, "Server", "VLC Server" );
 922     httpd_MsgAdd( answer, "Content-Length", "%d", answer->i_body );
 923     httpd_MsgAdd( answer, "Cseq", "%d",
 924                   atoi( httpd_MsgGet( query, "Cseq" ) ) );
 925     httpd_MsgAdd( answer, "Cache-Control", "%s", "no-cache" );
 926
 927     if( psz_session )
 928     {
 929         httpd_MsgAdd( answer, "Session", "%s"/*;timeout=5*/, psz_session );
 930     }
 931
 932     return VLC_SUCCESS;
 933 }
 934
 935 /*****************************************************************************
 936  * SDPGenerate: TODO
 937  * FIXME: need to be moved to a common place ?
 938  *****************************************************************************/
 939 static char *SDPGenerate( vod_media_t *p_media, char *psz_destination )
 940 {
 941     int i, i_size;
 942     char *p, *psz_sdp;
 943
 944     /* Calculate size */
 945     i_size = strlen( "v=0\r\n" ) +
 946         strlen( "o=- * * IN IP4 127.0.0.1\r\n" ) + 10 + 10 +
 947         strlen( "s=*\r\n" ) + strlen( p_media->psz_session_name ) +
 948         strlen( "i=*\r\n" ) + strlen( p_media->psz_session_description ) +
 949         strlen( "u=*\r\n" ) + strlen( p_media->psz_session_url ) +
 950         strlen( "e=*\r\n" ) + strlen( p_media->psz_session_email ) +
 951         strlen( "t=0 0\r\n" ) + /* FIXME */
 952         strlen( "a=tool:"PACKAGE_STRING"\r\n" ) +
 953         strlen( "c=IN IP4 */*\r\n" ) + 20 + 10 +
 954         strlen( psz_destination ? psz_destination : "0.0.0.0" ) +
 955         strlen( "a=range:npt=0-1000000000.000\r\n" );
 956
 957     for( i = 0; i < p_media->i_es; i++ )
 958     {
 959         media_es_t *p_es = p_media->es[i];
 960
 961         i_size += strlen( "m=**d*o * RTP/AVP *\r\n" ) + 10 + 10;
 962         if( p_es->psz_rtpmap )
 963         {
 964             i_size += strlen( "a=rtpmap:* *\r\n" ) +
 965                 strlen( p_es->psz_rtpmap ) + 10;
 966         }
 967         if( p_es->psz_fmtp )
 968         {
 969             i_size += strlen( "a=fmtp:* *\r\n" ) +
 970                 strlen( p_es->psz_fmtp ) + 10;
 971         }
 972
 973         i_size += strlen( "a=control:*/trackid=*\r\n" ) +
 974             strlen( p_media->psz_rtsp_control ) + 10;
 975     }
 976
 977     p = psz_sdp = malloc( i_size );
 978     p += sprintf( p, "v=0\r\n" );
 979     p += sprintf( p, "o=- "I64Fd" %d IN IP4 127.0.0.1\r\n",
 980                   p_media->i_sdp_id, p_media->i_sdp_version );
 981     if( *p_media->psz_session_name )
 982         p += sprintf( p, "s=%s\r\n", p_media->psz_session_name );
 983     if( *p_media->psz_session_description )
 984         p += sprintf( p, "i=%s\r\n", p_media->psz_session_description );
 985     if( *p_media->psz_session_url )
 986         p += sprintf( p, "u=%s\r\n", p_media->psz_session_url );
 987     if( *p_media->psz_session_email )
 988         p += sprintf( p, "e=%s\r\n", p_media->psz_session_email );
 989
 990     p += sprintf( p, "t=0 0\r\n" ); /* FIXME */
 991     p += sprintf( p, "a=tool:"PACKAGE_STRING"\r\n" );
 992
 993     p += sprintf( p, "c=IN IP4 %s/%d\r\n", psz_destination ?
 994                   psz_destination : "0.0.0.0", p_media->i_ttl );
 995
 996     if( p_media->i_length > 0 )
 997     p += sprintf( p, "a=range:npt=0-%.3f\r\n",
 998                   ((float)(p_media->i_length/1000))/1000 );
 999
1000     for( i = 0; i < p_media->i_es; i++ )
1001     {
1002         media_es_t *p_es = p_media->es[i];
1003
1004         if( p_es->fmt.i_cat == AUDIO_ES )
1005         {
1006             p += sprintf( p, "m=audio %d RTP/AVP %d\r\n",
1007                           p_es->i_port, p_es->i_payload_type );
1008         }
1009         else if( p_es->fmt.i_cat == VIDEO_ES )
1010         {
1011             p += sprintf( p, "m=video %d RTP/AVP %d\r\n",
1012                           p_es->i_port, p_es->i_payload_type );
1013         }
1014         else
1015         {
1016             continue;
1017         }
1018
1019         if( p_es->psz_rtpmap )
1020         {
1021             p += sprintf( p, "a=rtpmap:%d %s\r\n", p_es->i_payload_type,
1022                           p_es->psz_rtpmap );
1023         }
1024         if( p_es->psz_fmtp )
1025         {
1026             p += sprintf( p, "a=fmtp:%d %s\r\n", p_es->i_payload_type,
1027                           p_es->psz_fmtp );
1028         }
1029
1030         p += sprintf( p, "a=control:%s/trackid=%d\r\n",
1031                       p_media->psz_rtsp_control, i );
1032     }
1033
1034     fprintf( stderr, psz_sdp );
1035     return psz_sdp;
1036 }