git.sesse.net Git - vlc/blob - modules/demux/live555.cpp

   1 /*****************************************************************************
   2  * live555.cpp : LIVE555 Streaming Media support.
   3  *****************************************************************************
   4  * Copyright (C) 2003-2007 the VideoLAN team
   5  * $Id$
   6  *
   7  * Authors: Laurent Aimar <fenrir@via.ecp.fr>
   8  *          Derk-Jan Hartman <hartman at videolan. org>
   9  *          Derk-Jan Hartman <djhartman at m2x .dot. nl> for M2X
  10  *
  11  * This program is free software; you can redistribute it and/or modify
  12  * it under the terms of the GNU General Public License as published by
  13  * the Free Software Foundation; either version 2 of the License, or
  14  * (at your option) any later version.
  15  *
  16  * This program is distributed in the hope that it will be useful,
  17  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  18  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  19  * GNU General Public License for more details.
  20  *
  21  * You should have received a copy of the GNU General Public License
  22  * along with this program; if not, write to the Free Software
  23  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
  24  *****************************************************************************/
  25
  26 /*****************************************************************************
  27  * Preamble
  28  *****************************************************************************/
  29
  30 /* For inttypes.h
  31  * Note: config.h may include inttypes.h, so make sure we define this option
  32  * early enough. */
  33 #define __STDC_CONSTANT_MACROS 1
  34
  35 #ifdef HAVE_CONFIG_H
  36 # include "config.h"
  37 #endif
  38
  39 #include <inttypes.h>
  40
  41 #include <vlc_common.h>
  42 #include <vlc_plugin.h>
  43
  44 #include <vlc_demux.h>
  45 #include <vlc_interface.h>
  46 #include <vlc_network.h>
  47 #include <vlc_url.h>
  48
  49 #include <iostream>
  50 #include <limits.h>
  51
  52
  53 #if defined( WIN32 )
  54 #   include <winsock2.h>
  55 #endif
  56
  57 #include "UsageEnvironment.hh"
  58 #include "BasicUsageEnvironment.hh"
  59 #include "GroupsockHelper.hh"
  60 #include "liveMedia.hh"
  61
  62 extern "C" {
  63 #include "../access/mms/asf.h"  /* Who said ugly ? */
  64 }
  65
  66 using namespace std;
  67
  68 /*****************************************************************************
  69  * Module descriptor
  70  *****************************************************************************/
  71 static int  Open ( vlc_object_t * );
  72 static void Close( vlc_object_t * );
  73
  74 #define CACHING_TEXT N_("Caching value (ms)")
  75 #define CACHING_LONGTEXT N_( \
  76     "Allows you to modify the default caching value for RTSP streams. This " \
  77     "value should be set in millisecond units." )
  78
  79 #define KASENNA_TEXT N_( "Kasenna RTSP dialect")
  80 #define KASENNA_LONGTEXT N_( "Kasenna servers use an old and unstandard " \
  81     "dialect of RTSP. When you set this parameter, VLC will try this dialect "\
  82     "for communication. In this mode you cannot connect to normal RTSP servers." )
  83
  84 #define USER_TEXT N_("RTSP user name")
  85 #define USER_LONGTEXT N_("Allows you to modify the user name that will " \
  86     "be used for authenticating the connection.")
  87 #define PASS_TEXT N_("RTSP password")
  88 #define PASS_LONGTEXT N_("Allows you to modify the password that will be " \
  89     "used for the connection.")
  90
  91 vlc_module_begin();
  92     set_description( N_("RTP/RTSP/SDP demuxer (using Live555)" ) );
  93     set_capability( "demux", 50 );
  94     set_shortname( "RTP/RTSP");
  95     set_callbacks( Open, Close );
  96     add_shortcut( "live" );
  97     add_shortcut( "livedotcom" );
  98     set_category( CAT_INPUT );
  99     set_subcategory( SUBCAT_INPUT_DEMUX );
 100
 101     add_submodule();
 102         set_description( N_("RTSP/RTP access and demux") );
 103         add_shortcut( "rtsp" );
 104         add_shortcut( "sdp" );
 105         set_capability( "access_demux", 0 );
 106         set_callbacks( Open, Close );
 107         add_bool( "rtsp-tcp", 0, NULL,
 108                   N_("Use RTP over RTSP (TCP)"),
 109                   N_("Use RTP over RTSP (TCP)"), true );
 110         add_integer( "rtp-client-port", -1, NULL,
 111                   N_("Client port"),
 112                   N_("Port to use for the RTP source of the session"), true );
 113         add_bool( "rtsp-http", 0, NULL,
 114                   N_("Tunnel RTSP and RTP over HTTP"),
 115                   N_("Tunnel RTSP and RTP over HTTP"), true );
 116         add_integer( "rtsp-http-port", 80, NULL,
 117                   N_("HTTP tunnel port"),
 118                   N_("Port to use for tunneling the RTSP/RTP over HTTP."),
 119                   true );
 120         add_integer("rtsp-caching", 4 * DEFAULT_PTS_DELAY / 1000, NULL,
 121                     CACHING_TEXT, CACHING_LONGTEXT, true );
 122         add_bool(   "rtsp-kasenna", false, NULL, KASENNA_TEXT,
 123                     KASENNA_LONGTEXT, true );
 124         add_string( "rtsp-user", NULL, NULL, USER_TEXT,
 125                     USER_LONGTEXT, true );
 126         add_string( "rtsp-pwd", NULL, NULL, PASS_TEXT,
 127                     PASS_LONGTEXT, true );
 128 vlc_module_end();
 129
 130
 131 /*****************************************************************************
 132  * Local prototypes
 133  *****************************************************************************/
 134
 135 typedef struct
 136 {
 137     demux_t         *p_demux;
 138     MediaSubsession *sub;
 139
 140     es_format_t     fmt;
 141     es_out_id_t     *p_es;
 142
 143     bool      b_muxed;
 144     bool      b_quicktime;
 145     bool      b_asf;
 146     stream_t        *p_out_muxed;    /* for muxed stream */
 147
 148     uint8_t         *p_buffer;
 149     unsigned int    i_buffer;
 150
 151     bool      b_rtcp_sync;
 152     char            waiting;
 153     int64_t         i_pts;
 154     u_int32_t       i_start_seq;
 155
 156 } live_track_t;
 157
 158 struct timeout_thread_t
 159 {
 160     VLC_COMMON_MEMBERS
 161
 162     int64_t      i_remain;
 163     bool   b_handle_keep_alive;
 164     demux_sys_t  *p_sys;
 165 };
 166
 167 struct demux_sys_t
 168 {
 169     char            *p_sdp;    /* XXX mallocated */
 170     char            *psz_path; /* URL-encoded path */
 171     vlc_url_t       url;
 172
 173     MediaSession     *ms;
 174     TaskScheduler    *scheduler;
 175     UsageEnvironment *env ;
 176     RTSPClient       *rtsp;
 177
 178     /* */
 179     int              i_track;
 180     live_track_t     **track;   /* XXX mallocated */
 181
 182     /* Weird formats */
 183     asf_header_t     asfh;
 184     stream_t         *p_out_asf;
 185     bool       b_real;
 186
 187     /* */
 188     int64_t          i_pcr; /* The clock */
 189     int64_t          i_npt; /* The current time in the stream */
 190     int64_t          i_npt_length;
 191     int64_t          i_npt_start;
 192
 193     /* timeout thread information */
 194     int              i_timeout;     /* session timeout value in seconds */
 195     bool       b_timeout_call;/* mark to send an RTSP call to prevent server timeout */
 196     timeout_thread_t *p_timeout;    /* the actual thread that makes sure we don't timeout */
 197
 198     /* */
 199     bool       b_multicast;   /* true if one of the tracks is multicasted */
 200     bool       b_no_data;     /* true if we never receive any data */
 201     int              i_no_data_ti;  /* consecutive number of TaskInterrupt */
 202
 203     char             event;
 204 };
 205
 206 static int Demux  ( demux_t * );
 207 static int Control( demux_t *, int, va_list );
 208
 209 static int Connect      ( demux_t * );
 210 static int SessionsSetup( demux_t * );
 211 static int Play         ( demux_t *);
 212 static int ParseASF     ( demux_t * );
 213 static int RollOverTcp  ( demux_t * );
 214
 215 static void StreamRead  ( void *, unsigned int, unsigned int,
 216                           struct timeval, unsigned int );
 217 static void StreamClose ( void * );
 218 static void TaskInterrupt( void * );
 219
 220 static void TimeoutPrevention( timeout_thread_t * );
 221
 222 static unsigned char* parseH264ConfigStr( char const* configStr,
 223                                           unsigned int& configSize );
 224
 225 /*****************************************************************************
 226  * DemuxOpen:
 227  *****************************************************************************/
 228 static int  Open ( vlc_object_t *p_this )
 229 {
 230     demux_t     *p_demux = (demux_t*)p_this;
 231     demux_sys_t *p_sys = NULL;
 232
 233     MediaSubsessionIterator *iter   = NULL;
 234     MediaSubsession         *sub    = NULL;
 235     int i, i_return;
 236     int i_error = VLC_EGENERIC;
 237
 238     if( p_demux->s )
 239     {
 240         /* See if it looks like a SDP
 241            v, o, s fields are mandatory and in this order */
 242         const uint8_t *p_peek;
 243         if( stream_Peek( p_demux->s, &p_peek, 7 ) < 7 ) return VLC_EGENERIC;
 244
 245         if( memcmp( p_peek, "v=0\r\n", 5 ) &&
 246             memcmp( p_peek, "v=0\n", 4 ) &&
 247             ( p_peek[0] < 'a' || p_peek[0] > 'z' || p_peek[1] != '=' ) )
 248         {
 249             return VLC_EGENERIC;
 250         }
 251     }
 252     else
 253     {
 254         var_Create( p_demux, "rtsp-caching", VLC_VAR_INTEGER|VLC_VAR_DOINHERIT );
 255     }
 256
 257     p_demux->pf_demux  = Demux;
 258     p_demux->pf_control= Control;
 259     p_demux->p_sys     = p_sys = (demux_sys_t*)malloc( sizeof( demux_sys_t ) );
 260     if( !p_sys ) return VLC_ENOMEM;
 261
 262     p_sys->p_sdp = NULL;
 263     p_sys->scheduler = NULL;
 264     p_sys->env = NULL;
 265     p_sys->ms = NULL;
 266     p_sys->rtsp = NULL;
 267     p_sys->i_track = 0;
 268     p_sys->track   = NULL;
 269     p_sys->i_pcr = 0;
 270     p_sys->i_npt = 0;
 271     p_sys->i_npt_start = 0;
 272     p_sys->i_npt_length = 0;
 273     p_sys->p_out_asf = NULL;
 274     p_sys->b_no_data = true;
 275     p_sys->i_no_data_ti = 0;
 276     p_sys->p_timeout = NULL;
 277     p_sys->i_timeout = 0;
 278     p_sys->b_timeout_call = false;
 279     p_sys->b_multicast = false;
 280     p_sys->b_real = false;
 281     p_sys->psz_path = strdup( p_demux->psz_path );
 282
 283     /* parse URL for rtsp://[user:[passwd]@]serverip:port/options */
 284     vlc_UrlParse( &p_sys->url, p_sys->psz_path, 0 );
 285
 286     if( ( p_sys->scheduler = BasicTaskScheduler::createNew() ) == NULL )
 287     {
 288         msg_Err( p_demux, "BasicTaskScheduler::createNew failed" );
 289         goto error;
 290     }
 291     if( !( p_sys->env = BasicUsageEnvironment::createNew(*p_sys->scheduler) ) )
 292     {
 293         msg_Err( p_demux, "BasicUsageEnvironment::createNew failed" );
 294         goto error;
 295     }
 296
 297     if( strcasecmp( p_demux->psz_access, "sdp" ) )
 298     {
 299         char *p = p_sys->psz_path;
 300         while( (p = strchr( p, ' ' )) != NULL ) *p = '+';
 301     }
 302
 303     if( p_demux->s != NULL )
 304     {
 305         /* Gather the complete sdp file */
 306         int     i_sdp       = 0;
 307         int     i_sdp_max   = 1000;
 308         uint8_t *p_sdp      = (uint8_t*) malloc( i_sdp_max );
 309
 310         if( !p_sdp )
 311         {
 312             i_error = VLC_ENOMEM;
 313             goto error;
 314         }
 315
 316         for( ;; )
 317         {
 318             int i_read = stream_Read( p_demux->s, &p_sdp[i_sdp],
 319                                       i_sdp_max - i_sdp - 1 );
 320
 321             if( p_demux->b_die || p_demux->b_error )
 322             {
 323                 free( p_sdp );
 324                 goto error;
 325             }
 326
 327             if( i_read < 0 )
 328             {
 329                 msg_Err( p_demux, "failed to read SDP" );
 330                 free( p_sdp );
 331                 goto error;
 332             }
 333
 334             i_sdp += i_read;
 335
 336             if( i_read < i_sdp_max - i_sdp - 1 )
 337             {
 338                 p_sdp[i_sdp] = '\0';
 339                 break;
 340             }
 341
 342             i_sdp_max += 1000;
 343             p_sdp = (uint8_t*)realloc( p_sdp, i_sdp_max );
 344         }
 345         p_sys->p_sdp = (char*)p_sdp;
 346     }
 347     else if( ( p_demux->s == NULL ) &&
 348              !strcasecmp( p_demux->psz_access, "sdp" ) )
 349     {
 350         /* sdp:// link from SAP */
 351         p_sys->p_sdp = strdup( p_sys->psz_path );
 352     }
 353     else if( ( i_return = Connect( p_demux ) ) != VLC_SUCCESS )
 354     {
 355         msg_Err( p_demux, "Failed to connect with rtsp://%s", p_sys->psz_path );
 356         goto error;
 357     }
 358
 359     if( p_sys->p_sdp == NULL )
 360     {
 361         msg_Err( p_demux, "Failed to retrieve the RTSP Session Description" );
 362         i_error = VLC_ENOMEM;
 363         goto error;
 364     }
 365
 366     if( ( i_return = SessionsSetup( p_demux ) ) != VLC_SUCCESS )
 367     {
 368         msg_Err( p_demux, "Nothing to play for rtsp://%s", p_sys->psz_path );
 369         goto error;
 370     }
 371
 372     if( p_sys->b_real ) goto error;
 373
 374     if( ( i_return = Play( p_demux ) ) != VLC_SUCCESS )
 375         goto error;
 376
 377     if( p_sys->p_out_asf && ParseASF( p_demux ) )
 378     {
 379         msg_Err( p_demux, "cannot find a usable asf header" );
 380         /* TODO Clean tracks */
 381         goto error;
 382     }
 383
 384     if( p_sys->i_track <= 0 )
 385         goto error;
 386
 387     return VLC_SUCCESS;
 388
 389 error:
 390     for( i = 0; i < p_sys->i_track; i++ )
 391     {
 392         live_track_t *tk = p_sys->track[i];
 393
 394         if( tk->b_muxed ) stream_DemuxDelete( tk->p_out_muxed );
 395         es_format_Clean( &tk->fmt );
 396         free( tk->p_buffer );
 397         free( tk );
 398     }
 399
 400     if( p_sys->i_track ) free( p_sys->track );
 401     if( p_sys->p_out_asf ) stream_DemuxDelete( p_sys->p_out_asf );
 402     if( p_sys->rtsp && p_sys->ms ) p_sys->rtsp->teardownMediaSession( *p_sys->ms );
 403     if( p_sys->ms ) Medium::close( p_sys->ms );
 404     if( p_sys->rtsp ) RTSPClient::close( p_sys->rtsp );
 405     if( p_sys->env ) p_sys->env->reclaim();
 406     if( p_sys->p_timeout )
 407     {
 408         vlc_object_kill( p_sys->p_timeout );
 409         vlc_thread_join( p_sys->p_timeout );
 410         vlc_object_detach( p_sys->p_timeout );
 411         vlc_object_release( p_sys->p_timeout );
 412     }
 413     delete p_sys->scheduler;
 414     free( p_sys->p_sdp );
 415     free( p_sys->psz_path );
 416
 417     vlc_UrlClean( &p_sys->url );
 418
 419     free( p_sys );
 420     return i_error;
 421 }
 422
 423 /*****************************************************************************
 424  * DemuxClose:
 425  *****************************************************************************/
 426 static void Close( vlc_object_t *p_this )
 427 {
 428     demux_t *p_demux = (demux_t*)p_this;
 429     demux_sys_t *p_sys = p_demux->p_sys;
 430     int i;
 431
 432     for( i = 0; i < p_sys->i_track; i++ )
 433     {
 434         live_track_t *tk = p_sys->track[i];
 435
 436         if( tk->b_muxed ) stream_DemuxDelete( tk->p_out_muxed );
 437         es_format_Clean( &tk->fmt );
 438         free( tk->p_buffer );
 439         free( tk );
 440     }
 441
 442     if( p_sys->i_track ) free( p_sys->track );
 443     if( p_sys->p_out_asf ) stream_DemuxDelete( p_sys->p_out_asf );
 444     if( p_sys->rtsp && p_sys->ms ) p_sys->rtsp->teardownMediaSession( *p_sys->ms );
 445     if( p_sys->ms ) Medium::close( p_sys->ms );
 446     if( p_sys->rtsp ) RTSPClient::close( p_sys->rtsp );
 447     if( p_sys->env ) p_sys->env->reclaim();
 448     if( p_sys->p_timeout )
 449     {
 450         vlc_object_kill( p_sys->p_timeout );
 451         vlc_thread_join( p_sys->p_timeout );
 452         vlc_object_detach( p_sys->p_timeout );
 453         vlc_object_release( p_sys->p_timeout );
 454     }
 455     delete p_sys->scheduler;
 456     free( p_sys->p_sdp );
 457     free( p_sys->psz_path );
 458
 459     vlc_UrlClean( &p_sys->url );
 460
 461     free( p_sys );
 462 }
 463
 464 /*****************************************************************************
 465  * Connect: connects to the RTSP server to setup the session DESCRIBE
 466  *****************************************************************************/
 467 static int Connect( demux_t *p_demux )
 468 {
 469     demux_sys_t *p_sys = p_demux->p_sys;
 470     Authenticator authenticator;
 471     bool b_firstpass  = true;
 472     char *psz_user    = NULL;
 473     char *psz_pwd     = NULL;
 474     char *psz_url     = NULL;
 475     char *psz_options = NULL;
 476     char *p_sdp       = NULL;
 477     int  i_http_port  = 0;
 478     int  i_ret        = VLC_SUCCESS;
 479     int i_lefttries;
 480
 481     if( p_sys->url.i_port == 0 ) p_sys->url.i_port = 554;
 482     if( p_sys->url.psz_username || p_sys->url.psz_password )
 483     {
 484         int err;
 485         err = asprintf( &psz_url, "rtsp://%s:%d%s", p_sys->url.psz_host,
 486                         p_sys->url.i_port, p_sys->url.psz_path );
 487         if( err == -1 ) return VLC_ENOMEM;
 488
 489         psz_user = strdup( p_sys->url.psz_username );
 490         psz_pwd  = strdup( p_sys->url.psz_password );
 491     }
 492     else
 493     {
 494         int err;
 495         err = asprintf( &psz_url, "rtsp://%s", p_sys->psz_path );
 496         if( err == -1 ) return VLC_ENOMEM;
 497
 498         psz_user = var_CreateGetString( p_demux, "rtsp-user" );
 499         psz_pwd  = var_CreateGetString( p_demux, "rtsp-pwd" );
 500     }
 501
 502     i_lefttries = 3;
 503 createnew:
 504     i_lefttries--;
 505     if( p_demux->b_die || p_demux->b_error )
 506     {
 507         free( psz_user );
 508         free( psz_pwd );
 509         free( psz_url );
 510         return VLC_EGENERIC;
 511     }
 512
 513     if( var_CreateGetBool( p_demux, "rtsp-http" ) )
 514         i_http_port = var_CreateGetInteger( p_demux, "rtsp-http-port" );
 515
 516     if( ( p_sys->rtsp = RTSPClient::createNew( *p_sys->env,
 517           var_CreateGetInteger( p_demux, "verbose" ) > 1,
 518           "VLC media player", i_http_port ) ) == NULL )
 519     {
 520         msg_Err( p_demux, "RTSPClient::createNew failed (%s)",
 521                  p_sys->env->getResultMsg() );
 522         free( psz_user );
 523         free( psz_pwd );
 524         free( psz_url );
 525         return VLC_EGENERIC;
 526     }
 527
 528     /* Kasenna enables KeepAlive by analysing the User-Agent string.
 529      * Appending _KA to the string should be enough to enable this feature,
 530      * however, there is a bug where the _KA doesn't get parsed from the
 531      * default User-Agent as created by VLC/Live555 code. This is probably due
 532      * to spaces in the string or the string being too long. Here we override
 533      * the default string with a more compact version.
 534      */
 535     if( var_CreateGetBool( p_demux, "rtsp-kasenna" ))
 536     {
 537 #if LIVEMEDIA_LIBRARY_VERSION_INT > 1130457500
 538         p_sys->rtsp->setUserAgentString( "VLC_MEDIA_PLAYER_KA" );
 539 #endif
 540     }
 541
 542 describe:
 543     authenticator.setUsernameAndPassword( (const char*)psz_user,
 544                                           (const char*)psz_pwd );
 545
 546     psz_options = p_sys->rtsp->sendOptionsCmd( psz_url, psz_user, psz_pwd,
 547                                                &authenticator );
 548     delete [] psz_options;
 549
 550     p_sdp = p_sys->rtsp->describeURL( psz_url, &authenticator,
 551                          var_GetBool( p_demux, "rtsp-kasenna" ) );
 552     if( p_sdp == NULL )
 553     {
 554         /* failure occurred */
 555         int i_code = 0;
 556         const char *psz_error = p_sys->env->getResultMsg();
 557
 558         if( var_GetBool( p_demux, "rtsp-http" ) )
 559             sscanf( psz_error, "%*s %*s HTTP GET %*s HTTP/%*u.%*u %3u %*s",
 560                     &i_code );
 561         else
 562         {
 563             const char *psz_tmp = strstr( psz_error, "RTSP" );
 564             if( psz_tmp )
 565                 sscanf( psz_tmp, "RTSP/%*s%3u", &i_code );
 566             else
 567                 i_code = 0;
 568         }
 569         msg_Dbg( p_demux, "DESCRIBE failed with %d: %s", i_code, psz_error );
 570
 571         if( b_firstpass )
 572         {   /* describeURL always returns an "RTSP/1.0 401 Unauthorized" the
 573              * first time. This is a workaround to avoid asking for a
 574              * user/passwd the first time the code passess here. */
 575             i_code = 0;
 576             b_firstpass = false;
 577         }
 578
 579         if( i_code == 401 )
 580         {
 581             int i_result;
 582             msg_Dbg( p_demux, "authentication failed" );
 583
 584             free( psz_user );
 585             free( psz_pwd );
 586             psz_user = psz_pwd = NULL;
 587
 588             i_result = intf_UserLoginPassword( p_demux, _("RTSP authentication"),
 589                            _("Please enter a valid login name and a password."),
 590                                                    &psz_user, &psz_pwd );
 591             if( i_result == DIALOG_OK_YES )
 592             {
 593                 msg_Dbg( p_demux, "retrying with user=%s, pwd=%s",
 594                          psz_user, psz_pwd );
 595                 goto describe;
 596             }
 597         }
 598         else if( (i_code != 0) && !var_GetBool( p_demux, "rtsp-http" ) )
 599         {
 600             /* Perhaps a firewall is being annoying. Try HTTP tunneling mode */
 601             vlc_value_t val;
 602             val.b_bool = true;
 603             msg_Dbg( p_demux, "we will now try HTTP tunneling mode" );
 604             var_Set( p_demux, "rtsp-http", val );
 605             if( p_sys->rtsp ) RTSPClient::close( p_sys->rtsp );
 606             p_sys->rtsp = NULL;
 607             goto createnew;
 608         }
 609         else
 610         {
 611             msg_Dbg( p_demux, "connection timeout, retrying" );
 612             if( p_sys->rtsp ) RTSPClient::close( p_sys->rtsp );
 613             p_sys->rtsp = NULL;
 614             if( i_lefttries > 0 )
 615                 goto createnew;
 616         }
 617         i_ret = VLC_EGENERIC;
 618     }
 619
 620     /* malloc-ated copy */
 621     free( psz_url );
 622     free( psz_user );
 623     free( psz_pwd );
 624
 625     free( p_sys->p_sdp );
 626     p_sys->p_sdp = NULL;
 627     if( p_sdp ) p_sys->p_sdp = strdup( (char*)p_sdp );
 628     delete[] p_sdp;
 629
 630     return i_ret;
 631 }
 632
 633 /*****************************************************************************
 634  * SessionsSetup: prepares the subsessions and does the SETUP
 635  *****************************************************************************/
 636 static int SessionsSetup( demux_t *p_demux )
 637 {
 638     demux_sys_t             *p_sys  = p_demux->p_sys;
 639     MediaSubsessionIterator *iter   = NULL;
 640     MediaSubsession         *sub    = NULL;
 641
 642     bool     b_rtsp_tcp = false;
 643     int            i_client_port;
 644     int            i_return = VLC_SUCCESS;
 645     unsigned int   i_buffer = 0;
 646     unsigned const thresh = 200000; /* RTP reorder threshold .2 second (default .1) */
 647
 648     b_rtsp_tcp    = var_CreateGetBool( p_demux, "rtsp-tcp" ) ||
 649                     var_GetBool( p_demux, "rtsp-http" );
 650     i_client_port = var_CreateGetInteger( p_demux, "rtp-client-port" );
 651
 652     /* Create the session from the SDP */
 653     if( !( p_sys->ms = MediaSession::createNew( *p_sys->env, p_sys->p_sdp ) ) )
 654     {
 655         msg_Err( p_demux, "Could not create the RTSP Session: %s",
 656             p_sys->env->getResultMsg() );
 657         return VLC_EGENERIC;
 658     }
 659
 660     /* Initialise each media subsession */
 661     iter = new MediaSubsessionIterator( *p_sys->ms );
 662     while( ( sub = iter->next() ) != NULL )
 663     {
 664         Boolean bInit;
 665         live_track_t *tk;
 666
 667         if( p_demux->b_die || p_demux->b_error )
 668         {
 669             delete iter;
 670             return VLC_EGENERIC;
 671         }
 672
 673         /* Value taken from mplayer */
 674         if( !strcmp( sub->mediumName(), "audio" ) )
 675             i_buffer = 100000;
 676         else if( !strcmp( sub->mediumName(), "video" ) )
 677             i_buffer = 2000000;
 678         else continue;
 679
 680         if( i_client_port != -1 )
 681         {
 682             sub->setClientPortNum( i_client_port );
 683             i_client_port += 2;
 684         }
 685
 686         if( strcasestr( sub->codecName(), "REAL" ) )
 687         {
 688             msg_Info( p_demux, "real codec detected, using real-RTSP instead" );
 689             p_sys->b_real = true; /* This is a problem, we'll handle it later */
 690             continue;
 691         }
 692
 693         if( !strcmp( sub->codecName(), "X-ASF-PF" ) )
 694             bInit = sub->initiate( 4 ); /* Constant ? */
 695         else
 696             bInit = sub->initiate();
 697
 698         if( !bInit )
 699         {
 700             msg_Warn( p_demux, "RTP subsession '%s/%s' failed (%s)",
 701                       sub->mediumName(), sub->codecName(),
 702                       p_sys->env->getResultMsg() );
 703         }
 704         else
 705         {
 706             if( sub->rtpSource() != NULL )
 707             {
 708                 int fd = sub->rtpSource()->RTPgs()->socketNum();
 709
 710                 /* Increase the buffer size */
 711                 if( i_buffer > 0 )
 712                     increaseReceiveBufferTo( *p_sys->env, fd, i_buffer );
 713
 714                 /* Increase the RTP reorder timebuffer just a bit */
 715                 sub->rtpSource()->setPacketReorderingThresholdTime(thresh);
 716             }
 717             msg_Dbg( p_demux, "RTP subsession '%s/%s'", sub->mediumName(),
 718                      sub->codecName() );
 719
 720             /* Issue the SETUP */
 721             if( p_sys->rtsp )
 722             {
 723                 if( !( p_sys->rtsp->setupMediaSubsession( *sub, False,
 724                                                    b_rtsp_tcp ? True : False ) ) )
 725                 {
 726                     /* if we get an unsupported transport error, toggle TCP use and try again */
 727                     if( !strstr(p_sys->env->getResultMsg(), "461 Unsupported Transport")
 728                      || !( p_sys->rtsp->setupMediaSubsession( *sub, False,
 729                                                    b_rtsp_tcp ? False : True ) ) )
 730                     {
 731                         msg_Err( p_demux, "SETUP of'%s/%s' failed %s", sub->mediumName(),
 732                                  sub->codecName(), p_sys->env->getResultMsg() );
 733                         continue;
 734                     }
 735                 }
 736             }
 737
 738             /* Check if we will receive data from this subsession for this track */
 739             if( sub->readSource() == NULL ) continue;
 740             if( !p_sys->b_multicast )
 741             {
 742                 /* Check, because we need diff. rollover behaviour for multicast */
 743                 p_sys->b_multicast = IsMulticastAddress( sub->connectionEndpointAddress() );
 744             }
 745
 746             tk = (live_track_t*)malloc( sizeof( live_track_t ) );
 747             if( !tk )
 748             {
 749                 delete iter;
 750                 return VLC_ENOMEM;
 751             }
 752             tk->p_demux     = p_demux;
 753             tk->sub         = sub;
 754             tk->p_es        = NULL;
 755             tk->b_quicktime = false;
 756             tk->b_asf       = false;
 757             tk->b_muxed     = false;
 758             tk->p_out_muxed = NULL;
 759             tk->waiting     = 0;
 760             tk->b_rtcp_sync = false;
 761             tk->i_pts       = 0;
 762             tk->i_buffer    = 65536;
 763             tk->p_buffer    = (uint8_t *)malloc( 65536 );
 764             if( !tk->p_buffer )
 765             {
 766                 delete iter;
 767                 return VLC_ENOMEM;
 768             }
 769
 770             /* Value taken from mplayer */
 771             if( !strcmp( sub->mediumName(), "audio" ) )
 772             {
 773                 es_format_Init( &tk->fmt, AUDIO_ES, VLC_FOURCC('u','n','d','f') );
 774                 tk->fmt.audio.i_channels = sub->numChannels();
 775                 tk->fmt.audio.i_rate = sub->rtpTimestampFrequency();
 776
 777                 if( !strcmp( sub->codecName(), "MPA" ) ||
 778                     !strcmp( sub->codecName(), "MPA-ROBUST" ) ||
 779                     !strcmp( sub->codecName(), "X-MP3-DRAFT-00" ) )
 780                 {
 781                     tk->fmt.i_codec = VLC_FOURCC( 'm', 'p', 'g', 'a' );
 782                     tk->fmt.audio.i_rate = 0;
 783                 }
 784                 else if( !strcmp( sub->codecName(), "AC3" ) )
 785                 {
 786                     tk->fmt.i_codec = VLC_FOURCC( 'a', '5', '2', ' ' );
 787                     tk->fmt.audio.i_rate = 0;
 788                 }
 789                 else if( !strcmp( sub->codecName(), "L16" ) )
 790                 {
 791                     tk->fmt.i_codec = VLC_FOURCC( 't', 'w', 'o', 's' );
 792                     tk->fmt.audio.i_bitspersample = 16;
 793                 }
 794                 else if( !strcmp( sub->codecName(), "L8" ) )
 795                 {
 796                     tk->fmt.i_codec = VLC_FOURCC( 'a', 'r', 'a', 'w' );
 797                     tk->fmt.audio.i_bitspersample = 8;
 798                 }
 799                 else if( !strcmp( sub->codecName(), "PCMU" ) )
 800                 {
 801                     tk->fmt.i_codec = VLC_FOURCC( 'u', 'l', 'a', 'w' );
 802                 }
 803                 else if( !strcmp( sub->codecName(), "PCMA" ) )
 804                 {
 805                     tk->fmt.i_codec = VLC_FOURCC( 'a', 'l', 'a', 'w' );
 806                 }
 807                 else if( !strncmp( sub->codecName(), "G726", 4 ) )
 808                 {
 809                     tk->fmt.i_codec = VLC_FOURCC( 'g', '7', '2', '6' );
 810                     tk->fmt.audio.i_rate = 8000;
 811                     tk->fmt.audio.i_channels = 1;
 812                     if( !strcmp( sub->codecName()+5, "40" ) )
 813                         tk->fmt.i_bitrate = 40000;
 814                     else if( !strcmp( sub->codecName()+5, "32" ) )
 815                         tk->fmt.i_bitrate = 32000;
 816                     else if( !strcmp( sub->codecName()+5, "24" ) )
 817                         tk->fmt.i_bitrate = 24000;
 818                     else if( !strcmp( sub->codecName()+5, "16" ) )
 819                         tk->fmt.i_bitrate = 16000;
 820                 }
 821                 else if( !strcmp( sub->codecName(), "AMR" ) )
 822                 {
 823                     tk->fmt.i_codec = VLC_FOURCC( 's', 'a', 'm', 'r' );
 824                 }
 825                 else if( !strcmp( sub->codecName(), "AMR-WB" ) )
 826                 {
 827                     tk->fmt.i_codec = VLC_FOURCC( 's', 'a', 'w', 'b' );
 828                 }
 829                 else if( !strcmp( sub->codecName(), "MP4A-LATM" ) )
 830                 {
 831                     unsigned int i_extra;
 832                     uint8_t      *p_extra;
 833
 834                     tk->fmt.i_codec = VLC_FOURCC( 'm', 'p', '4', 'a' );
 835
 836                     if( ( p_extra = parseStreamMuxConfigStr( sub->fmtp_config(),
 837                                                              i_extra ) ) )
 838                     {
 839                         tk->fmt.i_extra = i_extra;
 840                         tk->fmt.p_extra = malloc( i_extra );
 841                         memcpy( tk->fmt.p_extra, p_extra, i_extra );
 842                         delete[] p_extra;
 843                     }
 844                     /* Because the "faad" decoder does not handle the LATM data length field
 845                        at the start of each returned LATM frame, tell the RTP source to omit it. */
 846                     ((MPEG4LATMAudioRTPSource*)sub->rtpSource())->omitLATMDataLengthField();
 847                 }
 848                 else if( !strcmp( sub->codecName(), "MPEG4-GENERIC" ) )
 849                 {
 850                     unsigned int i_extra;
 851                     uint8_t      *p_extra;
 852
 853                     tk->fmt.i_codec = VLC_FOURCC( 'm', 'p', '4', 'a' );
 854
 855                     if( ( p_extra = parseGeneralConfigStr( sub->fmtp_config(),
 856                                                            i_extra ) ) )
 857                     {
 858                         tk->fmt.i_extra = i_extra;
 859                         tk->fmt.p_extra = malloc( i_extra );
 860                         memcpy( tk->fmt.p_extra, p_extra, i_extra );
 861                         delete[] p_extra;
 862                     }
 863                 }
 864                 else if( !strcmp( sub->codecName(), "X-ASF-PF" ) )
 865                 {
 866                     tk->b_asf = true;
 867                     if( p_sys->p_out_asf == NULL )
 868                         p_sys->p_out_asf = stream_DemuxNew( p_demux, "asf",
 869                                                             p_demux->out );
 870                 }
 871                 else if( !strcmp( sub->codecName(), "X-QT" ) ||
 872                          !strcmp( sub->codecName(), "X-QUICKTIME" ) )
 873                 {
 874                     tk->b_quicktime = true;
 875                 }
 876                 else if( !strcmp( sub->codecName(), "SPEEX" ) )
 877                 {
 878                     tk->fmt.i_codec = VLC_FOURCC( 's', 'p', 'x', 'r' );
 879                     if ( sub->rtpTimestampFrequency() )
 880                         tk->fmt.audio.i_rate = sub->rtpTimestampFrequency();
 881                     else
 882                     {
 883                         msg_Warn( p_demux,"Using 8kHz as default sample rate." );
 884                         tk->fmt.audio.i_rate = 8000;
 885                     }
 886                 }
 887             }
 888             else if( !strcmp( sub->mediumName(), "video" ) )
 889             {
 890                 es_format_Init( &tk->fmt, VIDEO_ES, VLC_FOURCC('u','n','d','f') );
 891                 if( !strcmp( sub->codecName(), "MPV" ) )
 892                 {
 893                     tk->fmt.i_codec = VLC_FOURCC( 'm', 'p', 'g', 'v' );
 894                 }
 895                 else if( !strcmp( sub->codecName(), "H263" ) ||
 896                          !strcmp( sub->codecName(), "H263-1998" ) ||
 897                          !strcmp( sub->codecName(), "H263-2000" ) )
 898                 {
 899                     tk->fmt.i_codec = VLC_FOURCC( 'H', '2', '6', '3' );
 900                 }
 901                 else if( !strcmp( sub->codecName(), "H261" ) )
 902                 {
 903                     tk->fmt.i_codec = VLC_FOURCC( 'H', '2', '6', '1' );
 904                 }
 905                 else if( !strcmp( sub->codecName(), "H264" ) )
 906                 {
 907                     unsigned int i_extra = 0;
 908                     uint8_t      *p_extra = NULL;
 909
 910                     tk->fmt.i_codec = VLC_FOURCC( 'h', '2', '6', '4' );
 911                     tk->fmt.b_packetized = false;
 912
 913                     if((p_extra=parseH264ConfigStr( sub->fmtp_spropparametersets(),
 914                                                     i_extra ) ) )
 915                     {
 916                         tk->fmt.i_extra = i_extra;
 917                         tk->fmt.p_extra = malloc( i_extra );
 918                         memcpy( tk->fmt.p_extra, p_extra, i_extra );
 919
 920                         delete[] p_extra;
 921                     }
 922                 }
 923                 else if( !strcmp( sub->codecName(), "JPEG" ) )
 924                 {
 925                     tk->fmt.i_codec = VLC_FOURCC( 'M', 'J', 'P', 'G' );
 926                 }
 927                 else if( !strcmp( sub->codecName(), "MP4V-ES" ) )
 928                 {
 929                     unsigned int i_extra;
 930                     uint8_t      *p_extra;
 931
 932                     tk->fmt.i_codec = VLC_FOURCC( 'm', 'p', '4', 'v' );
 933
 934                     if( ( p_extra = parseGeneralConfigStr( sub->fmtp_config(),
 935                                                            i_extra ) ) )
 936                     {
 937                         tk->fmt.i_extra = i_extra;
 938                         tk->fmt.p_extra = malloc( i_extra );
 939                         memcpy( tk->fmt.p_extra, p_extra, i_extra );
 940                         delete[] p_extra;
 941                     }
 942                 }
 943                 else if( !strcmp( sub->codecName(), "X-QT" ) ||
 944                          !strcmp( sub->codecName(), "X-QUICKTIME" ) ||
 945                          !strcmp( sub->codecName(), "X-QDM" ) ||
 946                          !strcmp( sub->codecName(), "X-SV3V-ES" )  ||
 947                          !strcmp( sub->codecName(), "X-SORENSONVIDEO" ) )
 948                 {
 949                     tk->b_quicktime = true;
 950                 }
 951                 else if( !strcmp( sub->codecName(), "MP2T" ) )
 952                 {
 953                     tk->b_muxed = true;
 954                     tk->p_out_muxed = stream_DemuxNew( p_demux, "ts", p_demux->out );
 955                 }
 956                 else if( !strcmp( sub->codecName(), "MP2P" ) ||
 957                          !strcmp( sub->codecName(), "MP1S" ) )
 958                 {
 959                     tk->b_muxed = true;
 960                     tk->p_out_muxed = stream_DemuxNew( p_demux, "ps",
 961                                                        p_demux->out );
 962                 }
 963                 else if( !strcmp( sub->codecName(), "X-ASF-PF" ) )
 964                 {
 965                     tk->b_asf = true;
 966                     if( p_sys->p_out_asf == NULL )
 967                         p_sys->p_out_asf = stream_DemuxNew( p_demux, "asf",
 968                                                             p_demux->out );;
 969                 }
 970             }
 971
 972             if( !tk->b_quicktime && !tk->b_muxed && !tk->b_asf )
 973             {
 974                 tk->p_es = es_out_Add( p_demux->out, &tk->fmt );
 975             }
 976
 977             if( sub->rtcpInstance() != NULL )
 978             {
 979                 sub->rtcpInstance()->setByeHandler( StreamClose, tk );
 980             }
 981
 982             if( tk->p_es || tk->b_quicktime || tk->b_muxed || tk->b_asf )
 983             {
 984                 /* Append */
 985                 p_sys->track = (live_track_t**)realloc( p_sys->track, sizeof( live_track_t ) * ( p_sys->i_track + 1 ) );
 986                 p_sys->track[p_sys->i_track++] = tk;
 987             }
 988             else
 989             {
 990                 /* BUG ??? */
 991                 msg_Err( p_demux, "unusable RTSP track. this should not happen" );
 992                 es_format_Clean( &tk->fmt );
 993                 free( tk );
 994             }
 995         }
 996     }
 997     delete iter;
 998     if( p_sys->i_track <= 0 ) i_return = VLC_EGENERIC;
 999
1000 #if (LIVEMEDIA_LIBRARY_VERSION_INT >= 1199404800)
1001     /* Retrieve the starttime if possible */
1002     p_sys->i_npt_start = (int64_t)( p_sys->ms->playStartTime() * (double)1000000.0 );
1003 #else
1004     p_sys->i_npt_start = (int64_t) -1;
1005 #endif
1006     if( p_sys->i_npt_start < 0 )
1007         p_sys->i_npt_start = -1;
1008
1009 #if (LIVEMEDIA_LIBRARY_VERSION_INT >= 1199404800)
1010     /* Retrieve the duration if possible */
1011     p_sys->i_npt_length = (int64_t)( p_sys->ms->playEndTime() * (double)1000000.0 );
1012 #else
1013     p_sys->i_npt_length = (int64_t) -1;
1014 #endif
1015     if( p_sys->i_npt_length < 0 )
1016         p_sys->i_npt_length = -1;
1017
1018     msg_Dbg( p_demux, "setup start: %lld stop:%lld", p_sys->i_npt_start, p_sys->i_npt_length );
1019     return i_return;
1020 }
1021
1022 /*****************************************************************************
1023  * Play: starts the actual playback of the stream
1024  *****************************************************************************/
1025 static int Play( demux_t *p_demux )
1026 {
1027     demux_sys_t *p_sys = p_demux->p_sys;
1028     int i;
1029
1030     if( p_sys->rtsp )
1031     {
1032         /* The PLAY */
1033         if( !p_sys->rtsp->playMediaSession( *p_sys->ms, p_sys->i_npt_start, -1, 1 ) )
1034         {
1035             msg_Err( p_demux, "RTSP PLAY failed %s", p_sys->env->getResultMsg() );
1036             return VLC_EGENERIC;
1037         }
1038
1039         /* Retrieve the timeout value and set up a timeout prevention thread */
1040         p_sys->i_timeout = p_sys->rtsp->sessionTimeoutParameter();
1041         if( p_sys->i_timeout <= 0 )
1042             p_sys->i_timeout = 60; /* default value from RFC2326 */
1043
1044         /* start timeout-thread only on x-asf streams (wms), it has rtcp support but doesn't
1045          * seem to use it for liveness/keep-alive, get_parameter seems to work for it. get_parameter
1046          * doesn't work with dss 5.5.4 & 5.5.5, they seems to work with rtcp */
1047         if( !p_sys->p_timeout && p_sys->p_out_asf )
1048         {
1049             msg_Dbg( p_demux, "We have a timeout of %d seconds",  p_sys->i_timeout );
1050             p_sys->p_timeout = (timeout_thread_t *)vlc_object_create( p_demux, sizeof(timeout_thread_t) );
1051             p_sys->p_timeout->p_sys = p_demux->p_sys; /* lol, object recursion :D */
1052             if( vlc_thread_create( p_sys->p_timeout, "liveMedia-timeout", TimeoutPrevention,
1053                                    VLC_THREAD_PRIORITY_LOW, true ) )
1054             {
1055                 msg_Err( p_demux, "cannot spawn liveMedia timeout thread" );
1056                 vlc_object_release( p_sys->p_timeout );
1057             }
1058             msg_Dbg( p_demux, "spawned timeout thread" );
1059             vlc_object_attach( p_sys->p_timeout, p_demux );
1060         }
1061     }
1062     p_sys->i_pcr = 0;
1063
1064 #if (LIVEMEDIA_LIBRARY_VERSION_INT >= 1195257600)
1065     for( i = 0; i < p_sys->i_track; i++ )
1066     {
1067         if( !p_sys->track[i]->b_rtcp_sync )
1068             p_sys->track[i]->i_pts = (int64_t) ( p_sys->track[i]->sub->rtpInfo.timestamp * (double)1000000.0 );
1069         p_sys->track[i]->i_start_seq = (int)p_sys->track[i]->sub->rtpInfo.seqNum;
1070         msg_Info( p_demux, "set startseq: %u", p_sys->track[i]->i_start_seq );
1071     }
1072 #endif
1073
1074 #if (LIVEMEDIA_LIBRARY_VERSION_INT >= 1199404800)
1075     /* Retrieve the starttime if possible */
1076     p_sys->i_npt_start = (int64_t)( p_sys->ms->playStartTime() * (double)1000000.0 );
1077 #else
1078     p_sys->i_npt_start = -1;
1079 #endif
1080     if( p_sys->i_npt_start < 0 )
1081     {
1082         p_sys->i_npt_start = -1;
1083         p_sys->i_npt = 0;
1084     }
1085     else
1086         p_sys->i_npt = p_sys->i_npt_start;
1087
1088 #if (LIVEMEDIA_LIBRARY_VERSION_INT >= 1199404800)
1089     /* Retrieve the duration if possible */
1090     p_sys->i_npt_length = (int64_t)( p_sys->ms->playEndTime() * (double)1000000.0 );
1091 #else
1092     p_sys->i_npt_length = -1;
1093 #endif
1094     if( p_sys->i_npt_length < 0 )
1095         p_sys->i_npt_length = -1;
1096
1097     msg_Dbg( p_demux, "play start: %lld stop:%lld", p_sys->i_npt_start, p_sys->i_npt_length );
1098     return VLC_SUCCESS;
1099 }
1100
1101
1102 /*****************************************************************************
1103  * Demux:
1104  *****************************************************************************/
1105 static int Demux( demux_t *p_demux )
1106 {
1107     demux_sys_t    *p_sys = p_demux->p_sys;
1108     TaskToken      task;
1109
1110     bool      b_send_pcr = true;
1111     int64_t         i_pcr = 0;
1112     int             i;
1113
1114     /* Check if we need to send the server a Keep-A-Live signal */
1115     if( p_sys->b_timeout_call && p_sys->rtsp && p_sys->ms )
1116     {
1117         char *psz_bye = NULL;
1118         p_sys->rtsp->getMediaSessionParameter( *p_sys->ms, NULL, psz_bye );
1119         p_sys->b_timeout_call = false;
1120     }
1121
1122
1123     for( i = 0; i < p_sys->i_track; i++ )
1124     {
1125         live_track_t *tk = p_sys->track[i];
1126
1127         if( tk->b_asf || tk->b_muxed )
1128             b_send_pcr = false;
1129 #if 0
1130         if( i_pcr == 0 )
1131         {
1132             i_pcr = tk->i_pts;
1133         }
1134         else if( tk->i_pts != 0 && i_pcr > tk->i_pts )
1135         {
1136             i_pcr = tk->i_pts ;
1137         }
1138 #endif
1139     }
1140     if( p_sys->i_pcr > 0 )
1141     {
1142         if( b_send_pcr )
1143             es_out_Control( p_demux->out, ES_OUT_SET_PCR, p_sys->i_pcr );
1144     }
1145
1146     /* First warn we want to read data */
1147     p_sys->event = 0;
1148     for( i = 0; i < p_sys->i_track; i++ )
1149     {
1150         live_track_t *tk = p_sys->track[i];
1151
1152         if( tk->waiting == 0 )
1153         {
1154             tk->waiting = 1;
1155             tk->sub->readSource()->getNextFrame( tk->p_buffer, tk->i_buffer,
1156                                           StreamRead, tk, StreamClose, tk );
1157         }
1158     }
1159     /* Create a task that will be called if we wait more than 300ms */
1160     task = p_sys->scheduler->scheduleDelayedTask( 300000, TaskInterrupt, p_demux );
1161
1162     /* Do the read */
1163     p_sys->scheduler->doEventLoop( &p_sys->event );
1164
1165     /* remove the task */
1166     p_sys->scheduler->unscheduleDelayedTask( task );
1167
1168     /* Check for gap in pts value */
1169     for( i = 0; i < p_sys->i_track; i++ )
1170     {
1171         live_track_t *tk = p_sys->track[i];
1172
1173         if( !tk->b_muxed && !tk->b_rtcp_sync &&
1174             tk->sub->rtpSource() && tk->sub->rtpSource()->hasBeenSynchronizedUsingRTCP() )
1175         {
1176             msg_Dbg( p_demux, "tk->rtpSource->hasBeenSynchronizedUsingRTCP()" );
1177
1178             es_out_Control( p_demux->out, ES_OUT_RESET_PCR );
1179             tk->b_rtcp_sync = true;
1180             /* reset PCR */
1181             tk->i_pts = 0;
1182             p_sys->i_pcr = 0;
1183             i_pcr = 0;
1184         }
1185     }
1186
1187     if( p_sys->b_multicast && p_sys->b_no_data &&
1188         ( p_sys->i_no_data_ti > 120 ) )
1189     {
1190         /* FIXME Make this configurable
1191         msg_Err( p_demux, "no multicast data received in 36s, aborting" );
1192         return 0;
1193         */
1194     }
1195     else if( !p_sys->b_multicast && p_sys->b_no_data &&
1196              ( p_sys->i_no_data_ti > 34 ) )
1197     {
1198         bool b_rtsp_tcp = var_GetBool( p_demux, "rtsp-tcp" ) ||
1199                                 var_GetBool( p_demux, "rtsp-http" );
1200
1201         if( !b_rtsp_tcp && p_sys->rtsp && p_sys->ms )
1202         {
1203             msg_Warn( p_demux, "no data received in 10s. Switching to TCP" );
1204             if( RollOverTcp( p_demux ) )
1205             {
1206                 msg_Err( p_demux, "TCP rollover failed, aborting" );
1207                 return 0;
1208             }
1209         }
1210         else if( p_sys->i_no_data_ti > 34 )
1211         {
1212             msg_Err( p_demux, "no data received in 10s, aborting" );
1213             return 0;
1214         }
1215     }
1216     else if( !p_sys->b_multicast && p_sys->b_no_data && p_sys->i_no_data_ti > 34 )
1217     {
1218         /* EOF ? */
1219         msg_Warn( p_demux, "no data received in 10s, eof ?" );
1220         return 0;
1221     }
1222     return p_demux->b_error ? 0 : 1;
1223 }
1224
1225 /*****************************************************************************
1226  * Control:
1227  *****************************************************************************/
1228 static int Control( demux_t *p_demux, int i_query, va_list args )
1229 {
1230     demux_sys_t *p_sys = p_demux->p_sys;
1231     int64_t *pi64;
1232     double  *pf, f;
1233     bool *pb, *pb2, b_bool;
1234     int *pi_int;
1235
1236     switch( i_query )
1237     {
1238         case DEMUX_GET_TIME:
1239             pi64 = (int64_t*)va_arg( args, int64_t * );
1240             if( p_sys->i_npt > 0 )
1241             {
1242                 *pi64 = p_sys->i_npt;
1243                 return VLC_SUCCESS;
1244             }
1245             return VLC_EGENERIC;
1246
1247         case DEMUX_GET_LENGTH:
1248             pi64 = (int64_t*)va_arg( args, int64_t * );
1249             if( p_sys->i_npt_length > 0 )
1250             {
1251                 *pi64 = p_sys->i_npt_length;
1252                 return VLC_SUCCESS;
1253             }
1254             return VLC_EGENERIC;
1255
1256         case DEMUX_GET_POSITION:
1257             pf = (double*)va_arg( args, double* );
1258             if( p_sys->i_npt_length > 0 && p_sys->i_npt > 0 )
1259             {
1260                 *pf = (double)p_sys->i_npt / (double)p_sys->i_npt_length;
1261                 return VLC_SUCCESS;
1262             }
1263             return VLC_EGENERIC;
1264
1265         case DEMUX_SET_POSITION:
1266         {
1267             float time;
1268
1269             f = (double)va_arg( args, double );
1270             if( p_sys->rtsp && p_sys->i_npt_length > 0 )
1271             {
1272                 int i;
1273                 time = f * (double)p_sys->i_npt_length / (double)1000000.0;   /* in second */
1274                 if( !p_sys->rtsp->playMediaSession( *p_sys->ms, time, -1, 1 ) )
1275                 {
1276                     msg_Err( p_demux, "PLAY failed %s",
1277                         p_sys->env->getResultMsg() );
1278                     return VLC_EGENERIC;
1279                 }
1280                 es_out_Control( p_demux->out, ES_OUT_RESET_PCR );
1281                 p_sys->i_pcr = 0;
1282 #if (LIVEMEDIA_LIBRARY_VERSION_INT >= 1195257600)
1283                 /* Retrieve RTP-Info values */
1284                 for( i = 0; i < p_sys->i_track; i++ )
1285                 {
1286                     //if( !p_sys->track[i]->b_rtcp_sync )
1287                     p_sys->track[i]->b_rtcp_sync = false;
1288                     p_sys->track[i]->i_pts = (int64_t) ( p_sys->track[i]->sub->rtpInfo.timestamp * (double)1000000.0 );
1289                     p_sys->track[i]->i_start_seq = p_sys->track[i]->sub->rtpInfo.seqNum;
1290                     msg_Info( p_demux, "set pos startseq: %u", p_sys->track[i]->i_start_seq );
1291                 }
1292 #endif
1293 #if (LIVEMEDIA_LIBRARY_VERSION_INT >= 1199404800)
1294                 /* Retrieve the starttime if possible */
1295                 p_sys->i_npt_start = (int64_t)( p_sys->ms->playStartTime() * (double)1000000.0 );
1296 #else
1297                 p_sys->i_npt_start = -1;
1298 #endif
1299                 if( p_sys->i_npt_start < 0 )
1300                 {
1301                     p_sys->i_npt_start = -1;
1302                     p_sys->i_npt = 0;
1303                 }
1304                 else
1305                     p_sys->i_npt = p_sys->i_npt_start;
1306
1307 #if (LIVEMEDIA_LIBRARY_VERSION_INT >= 1199404800)
1308                 /* Retrieve the duration if possible */
1309                 p_sys->i_npt_length = (int64_t)( p_sys->ms->playEndTime() * (double)1000000.0 );
1310 #else
1311                 p_sys->i_npt_length = -1;
1312 #endif
1313                 if( p_sys->i_npt_length < 0 )
1314                     p_sys->i_npt_length = -1;
1315
1316                 msg_Dbg( p_demux, "seek start: %lld stop:%lld", p_sys->i_npt_start, p_sys->i_npt_length );
1317                 return VLC_SUCCESS;
1318             }
1319             return VLC_EGENERIC;
1320         }
1321         /* Special for access_demux */
1322         case DEMUX_CAN_PAUSE:
1323         case DEMUX_CAN_SEEK:
1324             pb = (bool*)va_arg( args, bool * );
1325             if( p_sys->rtsp && p_sys->i_npt_length )
1326                 /* Not always true, but will be handled in SET_PAUSE_STATE */
1327                 *pb = true;
1328             else
1329                 *pb = false;
1330             return VLC_SUCCESS;
1331
1332         case DEMUX_CAN_CONTROL_PACE:
1333             pb = (bool*)va_arg( args, bool * );
1334
1335 #if 1       /* Disable for now until we have a clock synchro algo
1336              * which works with something else than MPEG over UDP */
1337             *pb = false;
1338 #else
1339             *pb = true;
1340 #endif
1341             return VLC_SUCCESS;
1342
1343 #if 0
1344         case DEMUX_CAN_CONTROL_RATE:
1345             pb = (bool*)va_arg( args, bool * );
1346             pb2 = (bool*)va_arg( args, bool * );
1347
1348             *pb = p_sys->rtsp != NULL && p_sys->i_npt_length > 0 && !var_GetBool( p_demux, "rtsp-kasenna" );
1349             *pb2 = false;
1350             return VLC_SUCCESS;
1351
1352         case DEMUX_SET_RATE:
1353         {
1354             double f_scale;
1355
1356             if( !p_sys->rtsp || p_sys->i_npt_length <= 0 || var_GetBool( p_demux, "rtsp-kasenna" ) )
1357                 return VLC_EGENERIC;
1358
1359             /* TODO we might want to ensure that the new rate is different from
1360              * old rate after playMediaSession...
1361              * I have no idea how the server map the requested rate to the
1362              * ones it supports.
1363              * ex:
1364              *  current is x2 we request x1.5 if the server return x2 we will
1365              *  never succeed to return to x1.
1366              *  In this case we should retry with a lower rate until we have
1367              *  one (even x1).
1368              */
1369
1370             pi_int = (int*)va_arg( args, int * );
1371             f_scale = (double)INPUT_RATE_DEFAULT / (*p_int);
1372
1373             /* Passing -1 for the start and end time will mean liveMedia won't
1374             * create a Range: section for the RTSP message. The server should
1375             * pick up from the current position */
1376             if( !p_sys->rtsp->playMediaSession( *p_sys->ms, -1, -1, f_scale ) )
1377             {
1378                 msg_Err( p_demux, "PLAY with Scale %0.2f failed %s", f_scale,
1379                         p_sys->env->getResultMsg() );
1380                 return VLC_EGENERIC;
1381             }
1382             /* ReSync the stream */
1383             p_sys->i_npt_start = 0;
1384             p_sys->i_pcr = 0;
1385             es_out_Control( p_demux->out, ES_OUT_RESET_PCR );
1386
1387             *pi_int = (int)( INPUT_RATE_DEFAULT / p_sys->ms->scale() + 0.5 );
1388             return VLC_SUCCESS;
1389         }
1390 #endif
1391
1392         case DEMUX_SET_PAUSE_STATE:
1393         {
1394             double d_npt = (double) p_sys->i_npt / INT64_C(1000000);
1395             int i;
1396
1397             b_bool = (bool)va_arg( args, int );
1398             if( p_sys->rtsp == NULL )
1399                 return VLC_EGENERIC;
1400
1401             /* FIXME */
1402             if( ( b_bool && !p_sys->rtsp->pauseMediaSession( *p_sys->ms ) ) ||
1403                     ( !b_bool && !p_sys->rtsp->playMediaSession( *p_sys->ms,
1404                       d_npt > 0 ? d_npt : -1 ) ) )
1405             {
1406                     msg_Err( p_demux, "PLAY or PAUSE failed %s", p_sys->env->getResultMsg() );
1407                     return VLC_EGENERIC;
1408             }
1409
1410             /* When we Pause, we'll need the TimeoutPrevention thread to
1411              * handle sending the "Keep Alive" message to the server.
1412              * Unfortunately Live555 isn't thread safe and so can't
1413              * do this normally while the main Demux thread is handling
1414              * a live stream. We end up with the Timeout thread blocking
1415              * waiting for a response from the server. So when we PAUSE
1416              * we set a flag that the TimeoutPrevention function will check
1417              * and if it's set, it will trigger the GET_PARAMETER message */
1418             if( b_bool && p_sys->p_timeout != NULL )
1419                 p_sys->p_timeout->b_handle_keep_alive = true;
1420             else if( !b_bool && p_sys->p_timeout != NULL )
1421                 p_sys->p_timeout->b_handle_keep_alive = false;
1422
1423             es_out_Control( p_demux->out, ES_OUT_RESET_PCR );
1424             p_sys->i_pcr = 0;
1425 #if (LIVEMEDIA_LIBRARY_VERSION_INT >= 1195257600)
1426             for( i = 0; i < p_sys->i_track; i++ )
1427             {
1428                 if( !p_sys->track[i]->b_rtcp_sync )
1429                     p_sys->track[i]->i_pts = 0; // (int64_t) ( p_sys->track[i]->sub->rtpInfo.timestamp * (double)1000000.0 );
1430                 p_sys->track[i]->i_start_seq = p_sys->track[i]->sub->rtpInfo.seqNum;
1431                 msg_Info( p_demux, "set pause startseq: %u", p_sys->track[i]->i_start_seq );
1432             }
1433 #endif
1434 #if (LIVEMEDIA_LIBRARY_VERSION_INT >= 1199404800)
1435             /* Retrieve the starttime if possible */
1436             p_sys->i_npt_start = (int64_t)( p_sys->ms->playStartTime() * (double)1000000.0 );
1437 #else
1438             p_sys->i_npt_start = -1;
1439 #endif
1440             if( p_sys->i_npt_start < 0 )
1441             {
1442                 p_sys->i_npt_start = -1;
1443                 p_sys->i_npt = 0;
1444             }
1445             else
1446                 p_sys->i_npt = p_sys->i_npt_start;
1447
1448 #if (LIVEMEDIA_LIBRARY_VERSION_INT >= 1199404800)
1449             /* Retrieve the duration if possible */
1450             p_sys->i_npt_length = (int64_t)( p_sys->ms->playEndTime() * (double)1000000.0 );
1451 #else
1452             p_sys->i_npt_length = -1;
1453 #endif
1454             if( p_sys->i_npt_length < 0 )
1455                 p_sys->i_npt_length = -1;
1456
1457             msg_Dbg( p_demux, "pause start: %lld stop:%lld", p_sys->i_npt_start, p_sys->i_npt_length );
1458             return VLC_SUCCESS;
1459         }
1460         case DEMUX_GET_TITLE_INFO:
1461         case DEMUX_SET_TITLE:
1462         case DEMUX_SET_SEEKPOINT:
1463             return VLC_EGENERIC;
1464
1465         case DEMUX_GET_PTS_DELAY:
1466             pi64 = (int64_t*)va_arg( args, int64_t * );
1467             *pi64 = (int64_t)var_GetInteger( p_demux, "rtsp-caching" ) * 1000;
1468             return VLC_SUCCESS;
1469
1470         default:
1471             return VLC_EGENERIC;
1472     }
1473 }
1474
1475 /*****************************************************************************
1476  * RollOverTcp: reopen the rtsp into TCP mode
1477  * XXX: ugly, a lot of code are duplicated from Open()
1478  * This should REALLY be fixed
1479  *****************************************************************************/
1480 static int RollOverTcp( demux_t *p_demux )
1481 {
1482     demux_sys_t *p_sys = p_demux->p_sys;
1483     int i, i_return;
1484
1485     var_SetBool( p_demux, "rtsp-tcp", true );
1486
1487     /* We close the old RTSP session */
1488     for( i = 0; i < p_sys->i_track; i++ )
1489     {
1490         live_track_t *tk = p_sys->track[i];
1491
1492         if( tk->b_muxed ) stream_DemuxDelete( tk->p_out_muxed );
1493         es_format_Clean( &tk->fmt );
1494         free( tk->p_buffer );
1495         free( tk );
1496     }
1497     if( p_sys->i_track ) free( p_sys->track );
1498     if( p_sys->p_out_asf ) stream_DemuxDelete( p_sys->p_out_asf );
1499
1500     p_sys->rtsp->teardownMediaSession( *p_sys->ms );
1501     Medium::close( p_sys->ms );
1502     RTSPClient::close( p_sys->rtsp );
1503
1504     p_sys->ms = NULL;
1505     p_sys->rtsp = NULL;
1506     p_sys->track = NULL;
1507     p_sys->i_track = 0;
1508
1509     /* Reopen rtsp client */
1510     if( ( p_demux->s != NULL ) &&
1511         ( i_return = Connect( p_demux ) ) != VLC_SUCCESS )
1512     {
1513         msg_Err( p_demux, "Failed to connect with rtsp://%s",
1514                  p_sys->psz_path );
1515         goto error;
1516     }
1517
1518     if( p_sys->p_sdp == NULL )
1519     {
1520         msg_Err( p_demux, "Failed to retrieve the RTSP Session Description" );
1521         goto error;
1522     }
1523
1524     if( ( i_return = SessionsSetup( p_demux ) ) != VLC_SUCCESS )
1525     {
1526         msg_Err( p_demux, "Nothing to play for rtsp://%s", p_sys->psz_path );
1527         goto error;
1528     }
1529
1530     if( ( i_return = Play( p_demux ) ) != VLC_SUCCESS )
1531         goto error;
1532
1533     return VLC_SUCCESS;
1534
1535 error:
1536     return VLC_EGENERIC;
1537 }
1538
1539
1540 /*****************************************************************************
1541  *
1542  *****************************************************************************/
1543 static void StreamRead( void *p_private, unsigned int i_size,
1544                         unsigned int i_truncated_bytes, struct timeval pts,
1545                         unsigned int duration )
1546 {
1547     live_track_t   *tk = (live_track_t*)p_private;
1548     demux_t        *p_demux = tk->p_demux;
1549     demux_sys_t    *p_sys = p_demux->p_sys;
1550     block_t        *p_block;
1551
1552     //msg_Dbg( p_demux, "pts: %d", pts.tv_sec );
1553
1554     int64_t i_pts = (uint64_t)pts.tv_sec * UINT64_C(1000000) +
1555         (uint64_t)pts.tv_usec;
1556
1557     /* XXX Beurk beurk beurk Avoid having negative value XXX */
1558     i_pts &= UINT64_C(0x00ffffffffffffff);
1559
1560     if( tk->b_quicktime && tk->p_es == NULL )
1561     {
1562         QuickTimeGenericRTPSource *qtRTPSource =
1563             (QuickTimeGenericRTPSource*)tk->sub->rtpSource();
1564         QuickTimeGenericRTPSource::QTState &qtState = qtRTPSource->qtState;
1565         uint8_t *sdAtom = (uint8_t*)&qtState.sdAtom[4];
1566
1567         if( tk->fmt.i_cat == VIDEO_ES ) {
1568             if( qtState.sdAtomSize < 16 + 32 )
1569             {
1570                 /* invalid */
1571                 p_sys->event = 0xff;
1572                 tk->waiting = 0;
1573                 return;
1574             }
1575             tk->fmt.i_codec = VLC_FOURCC(sdAtom[0],sdAtom[1],sdAtom[2],sdAtom[3]);
1576             tk->fmt.video.i_width  = (sdAtom[28] << 8) | sdAtom[29];
1577             tk->fmt.video.i_height = (sdAtom[30] << 8) | sdAtom[31];
1578
1579             if( tk->fmt.i_codec == VLC_FOURCC('a', 'v', 'c', '1') )
1580             {
1581                 uint8_t *pos = (uint8_t*)qtRTPSource->qtState.sdAtom + 86;
1582                 uint8_t *endpos = (uint8_t*)qtRTPSource->qtState.sdAtom
1583                                   + qtRTPSource->qtState.sdAtomSize;
1584                 while (pos+8 < endpos) {
1585                     unsigned int atomLength = pos[0]<<24 | pos[1]<<16 | pos[2]<<8 | pos[3];
1586                     if( atomLength == 0 || atomLength > (unsigned int)(endpos-pos)) break;
1587                     if( memcmp(pos+4, "avcC", 4) == 0 &&
1588                         atomLength > 8 &&
1589                         atomLength <= INT_MAX )
1590                     {
1591                         tk->fmt.i_extra = atomLength-8;
1592                         tk->fmt.p_extra = malloc( tk->fmt.i_extra );
1593                         memcpy(tk->fmt.p_extra, pos+8, atomLength-8);
1594                         break;
1595                     }
1596                     pos += atomLength;
1597                 }
1598             }
1599             else
1600             {
1601                 tk->fmt.i_extra        = qtState.sdAtomSize - 16;
1602                 tk->fmt.p_extra        = malloc( tk->fmt.i_extra );
1603                 memcpy( tk->fmt.p_extra, &sdAtom[12], tk->fmt.i_extra );
1604             }
1605         }
1606         else {
1607             if( qtState.sdAtomSize < 4 )
1608             {
1609                 /* invalid */
1610                 p_sys->event = 0xff;
1611                 tk->waiting = 0;
1612                 return;
1613             }
1614             tk->fmt.i_codec = VLC_FOURCC(sdAtom[0],sdAtom[1],sdAtom[2],sdAtom[3]);
1615         }
1616         tk->p_es = es_out_Add( p_demux->out, &tk->fmt );
1617     }
1618
1619 #if 0
1620     fprintf( stderr, "StreamRead size=%d pts=%lld\n",
1621              i_size,
1622              pts.tv_sec * 1000000LL + pts.tv_usec );
1623 #endif
1624
1625     /* grow buffer if it looks like buffer is too small, but don't eat
1626      * up all the memory on strange streams */
1627     if( i_truncated_bytes > 0 && tk->i_buffer < 2000000 )
1628     {
1629         void *p_tmp;
1630         msg_Dbg( p_demux, "lost %d bytes", i_truncated_bytes );
1631         msg_Dbg( p_demux, "increasing buffer size to %d", tk->i_buffer * 2 );
1632         tk->i_buffer *= 2;
1633         p_tmp = realloc( tk->p_buffer, tk->i_buffer );
1634         if( p_tmp == NULL )
1635         {
1636             msg_Warn( p_demux, "realloc failed" );
1637         }
1638         else
1639         {
1640             tk->p_buffer = (uint8_t*)p_tmp;
1641         }
1642     }
1643     if( i_size > tk->i_buffer )
1644     {
1645         msg_Warn( p_demux, "buffer overflow" );
1646     }
1647     /* FIXME could i_size be > buffer size ? */
1648     if( tk->fmt.i_codec == VLC_FOURCC('s','a','m','r') ||
1649         tk->fmt.i_codec == VLC_FOURCC('s','a','w','b') )
1650     {
1651         AMRAudioSource *amrSource = (AMRAudioSource*)tk->sub->readSource();
1652
1653         p_block = block_New( p_demux, i_size + 1 );
1654         p_block->p_buffer[0] = amrSource->lastFrameHeader();
1655         memcpy( p_block->p_buffer + 1, tk->p_buffer, i_size );
1656     }
1657     else if( tk->fmt.i_codec == VLC_FOURCC('H','2','6','1') )
1658     {
1659 #if LIVEMEDIA_LIBRARY_VERSION_INT >= 1081468800
1660         H261VideoRTPSource *h261Source = (H261VideoRTPSource*)tk->sub->rtpSource();
1661         uint32_t header = h261Source->lastSpecialHeader();
1662 #else
1663         uint32_t header = 0;
1664         msg_Warn( p_demux, "need livemedia library >= \"2004.04.09\"" );
1665 #endif
1666         p_block = block_New( p_demux, i_size + 4 );
1667         memcpy( p_block->p_buffer, &header, 4 );
1668         memcpy( p_block->p_buffer + 4, tk->p_buffer, i_size );
1669
1670         if( tk->sub->rtpSource()->curPacketMarkerBit() )
1671             p_block->i_flags |= BLOCK_FLAG_END_OF_FRAME;
1672     }
1673     else if( tk->fmt.i_codec == VLC_FOURCC('h','2','6','4') )
1674     {
1675         if( (tk->p_buffer[0] & 0x1f) >= 24 )
1676             msg_Warn( p_demux, "unsupported NAL type for H264" );
1677
1678         /* Normal NAL type */
1679         p_block = block_New( p_demux, i_size + 4 );
1680         p_block->p_buffer[0] = 0x00;
1681         p_block->p_buffer[1] = 0x00;
1682         p_block->p_buffer[2] = 0x00;
1683         p_block->p_buffer[3] = 0x01;
1684         memcpy( &p_block->p_buffer[4], tk->p_buffer, i_size );
1685     }
1686     else if( tk->b_asf )
1687     {
1688         int i_copy = __MIN( p_sys->asfh.i_min_data_packet_size, (int)i_size );
1689         p_block = block_New( p_demux, p_sys->asfh.i_min_data_packet_size );
1690
1691         memcpy( p_block->p_buffer, tk->p_buffer, i_copy );
1692     }
1693     else
1694     {
1695         p_block = block_New( p_demux, i_size );
1696         memcpy( p_block->p_buffer, tk->p_buffer, i_size );
1697     }
1698
1699     /* Update NPT */
1700     //msg_Dbg( p_demux, "current %d, start_seq %u", (int)tk->sub->rtpSource()->curPacketRTPSeqNum(), tk->i_start_seq );
1701     if( (tk->fmt.i_cat == VIDEO_ES) && (p_sys->i_pcr < i_pts) &&
1702         (i_pts > 0) && (p_sys->i_pcr > 0) )
1703     {
1704         p_sys->i_npt += __MAX( 0, i_pts - p_sys->i_pcr );
1705         p_sys->i_pcr = i_pts;
1706         //msg_Dbg( p_demux, "npt update" );
1707     }
1708     else if( /*tk->fmt.i_cat == VIDEO_ES &&*/ p_sys->i_pcr < i_pts )
1709     {
1710         p_sys->i_pcr = i_pts;
1711     }
1712     //msg_Dbg( p_demux, "npt %lld", p_sys->i_npt );
1713
1714     if( (i_pts != tk->i_pts) && (!tk->b_muxed) )
1715     {
1716         p_block->i_pts = i_pts;
1717     }
1718     if( !tk->b_muxed )
1719     {
1720         /*FIXME: for h264 you should check that packetization-mode=1 in sdp-file */
1721         p_block->i_dts = ( tk->fmt.i_codec == VLC_FOURCC( 'm', 'p', 'g', 'v' ) ) ? 0 : i_pts;
1722     }
1723
1724     if( tk->b_muxed )
1725     {
1726         stream_DemuxSend( tk->p_out_muxed, p_block );
1727     }
1728     else if( tk->b_asf )
1729     {
1730         stream_DemuxSend( p_sys->p_out_asf, p_block );
1731     }
1732     else
1733     {
1734         es_out_Send( p_demux->out, tk->p_es, p_block );
1735     }
1736
1737     /* warn that's ok */
1738     p_sys->event = 0xff;
1739
1740     /* we have read data */
1741     tk->waiting = 0;
1742     p_demux->p_sys->b_no_data = false;
1743     p_demux->p_sys->i_no_data_ti = 0;
1744
1745     if( i_pts > 0 && !tk->b_muxed )
1746     {
1747         tk->i_pts = i_pts;
1748     }
1749 }
1750
1751 /*****************************************************************************
1752  *
1753  *****************************************************************************/
1754 static void StreamClose( void *p_private )
1755 {
1756     live_track_t   *tk = (live_track_t*)p_private;
1757     demux_t        *p_demux = tk->p_demux;
1758     demux_sys_t    *p_sys = p_demux->p_sys;
1759
1760     msg_Dbg( p_demux, "StreamClose" );
1761
1762     p_sys->event = 0xff;
1763     p_demux->b_error = true;
1764 }
1765
1766
1767 /*****************************************************************************
1768  *
1769  *****************************************************************************/
1770 static void TaskInterrupt( void *p_private )
1771 {
1772     demux_t *p_demux = (demux_t*)p_private;
1773
1774     p_demux->p_sys->i_no_data_ti++;
1775
1776     /* Avoid lock */
1777     p_demux->p_sys->event = 0xff;
1778 }
1779
1780 /*****************************************************************************
1781  *
1782  *****************************************************************************/
1783 static void TimeoutPrevention( timeout_thread_t *p_timeout )
1784 {
1785     p_timeout->b_die = false;
1786     p_timeout->i_remain = (int64_t)p_timeout->p_sys->i_timeout -2;
1787     p_timeout->i_remain *= 1000000;
1788
1789     vlc_thread_ready( p_timeout );
1790
1791     /* Avoid lock */
1792     while( !p_timeout->b_die )
1793     {
1794         if( p_timeout->i_remain <= 0 )
1795         {
1796             char *psz_bye = NULL;
1797             p_timeout->i_remain = (int64_t)p_timeout->p_sys->i_timeout -2;
1798             p_timeout->i_remain *= 1000000;
1799             msg_Dbg( p_timeout, "reset the timeout timer" );
1800             if( p_timeout->b_handle_keep_alive == true )
1801             {
1802 #if LIVEMEDIA_LIBRARY_VERSION_INT >= 1138089600
1803                 p_timeout->p_sys->rtsp->getMediaSessionParameter( *p_timeout->p_sys->ms, NULL, psz_bye );
1804 #endif
1805                 p_timeout->p_sys->b_timeout_call = false;
1806             }
1807             else
1808             {
1809                 p_timeout->p_sys->b_timeout_call = true;
1810             }
1811         }
1812         p_timeout->i_remain -= 200000;
1813         msleep( 200000 ); /* 200 ms */
1814     }
1815 }
1816
1817 /*****************************************************************************
1818  *
1819  *****************************************************************************/
1820 static int b64_decode( char *dest, char *src );
1821
1822 static int ParseASF( demux_t *p_demux )
1823 {
1824     demux_sys_t    *p_sys = p_demux->p_sys;
1825
1826     const char *psz_marker = "a=pgmpu:data:application/vnd.ms.wms-hdr.asfv1;base64,";
1827     char *psz_asf = strcasestr( p_sys->p_sdp, psz_marker );
1828     char *psz_end;
1829     block_t *p_header;
1830
1831     /* Parse the asf header */
1832     if( psz_asf == NULL )
1833         return VLC_EGENERIC;
1834
1835     psz_asf += strlen( psz_marker );
1836     psz_asf = strdup( psz_asf );    /* Duplicate it */
1837     psz_end = strchr( psz_asf, '\n' );
1838
1839     while( psz_end > psz_asf && ( *psz_end == '\n' || *psz_end == '\r' ) )
1840         *psz_end-- = '\0';
1841
1842     if( psz_asf >= psz_end )
1843     {
1844         free( psz_asf );
1845         return VLC_EGENERIC;
1846     }
1847
1848     /* Always smaller */
1849     p_header = block_New( p_demux, psz_end - psz_asf );
1850     p_header->i_buffer = b64_decode( (char*)p_header->p_buffer, psz_asf );
1851     //msg_Dbg( p_demux, "Size=%d Hdrb64=%s", p_header->i_buffer, psz_asf );
1852     if( p_header->i_buffer <= 0 )
1853     {
1854         free( psz_asf );
1855         return VLC_EGENERIC;
1856     }
1857
1858     /* Parse it to get packet size */
1859     asf_HeaderParse( &p_sys->asfh, p_header->p_buffer, p_header->i_buffer );
1860
1861     /* Send it to demuxer */
1862     stream_DemuxSend( p_sys->p_out_asf, p_header );
1863
1864     free( psz_asf );
1865     return VLC_SUCCESS;
1866 }
1867
1868
1869 static unsigned char* parseH264ConfigStr( char const* configStr,
1870                                           unsigned int& configSize )
1871 {
1872     char *dup, *psz;
1873     int i, i_records = 1;
1874
1875     if( configSize )
1876     configSize = 0;
1877
1878     if( configStr == NULL || *configStr == '\0' )
1879         return NULL;
1880
1881     psz = dup = strdup( configStr );
1882
1883     /* Count the number of comma's */
1884     for( psz = dup; *psz != '\0'; ++psz )
1885     {
1886         if( *psz == ',')
1887         {
1888             ++i_records;
1889             *psz = '\0';
1890         }
1891     }
1892
1893     unsigned char *cfg = new unsigned char[5 * strlen(dup)];
1894     psz = dup;
1895     for( i = 0; i < i_records; i++ )
1896     {
1897         cfg[configSize++] = 0x00;
1898         cfg[configSize++] = 0x00;
1899         cfg[configSize++] = 0x00;
1900         cfg[configSize++] = 0x01;
1901
1902         configSize += b64_decode( (char*)&cfg[configSize], psz );
1903         psz += strlen(psz)+1;
1904     }
1905
1906     free( dup );
1907     return cfg;
1908 }
1909
1910 /*char b64[] = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/";*/
1911 static int b64_decode( char *dest, char *src )
1912 {
1913     const char *dest_start = dest;
1914     int  i_level;
1915     int  last = 0;
1916     int  b64[256] = {
1917         -1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,  /* 00-0F */
1918         -1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,  /* 10-1F */
1919         -1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,62,-1,-1,-1,63,  /* 20-2F */
1920         52,53,54,55,56,57,58,59,60,61,-1,-1,-1,-1,-1,-1,  /* 30-3F */
1921         -1, 0, 1, 2, 3, 4, 5, 6, 7, 8, 9,10,11,12,13,14,  /* 40-4F */
1922         15,16,17,18,19,20,21,22,23,24,25,-1,-1,-1,-1,-1,  /* 50-5F */
1923         -1,26,27,28,29,30,31,32,33,34,35,36,37,38,39,40,  /* 60-6F */
1924         41,42,43,44,45,46,47,48,49,50,51,-1,-1,-1,-1,-1,  /* 70-7F */
1925         -1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,  /* 80-8F */
1926         -1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,  /* 90-9F */
1927         -1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,  /* A0-AF */
1928         -1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,  /* B0-BF */
1929         -1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,  /* C0-CF */
1930         -1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,  /* D0-DF */
1931         -1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,  /* E0-EF */
1932         -1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1   /* F0-FF */
1933         };
1934
1935     for( i_level = 0; *src != '\0'; src++ )
1936     {
1937         int  c;
1938
1939         c = b64[(unsigned int)*src];
1940         if( c == -1 )
1941         {
1942             continue;
1943         }
1944
1945         switch( i_level )
1946         {
1947             case 0:
1948                 i_level++;
1949                 break;
1950             case 1:
1951                 *dest++ = ( last << 2 ) | ( ( c >> 4)&0x03 );
1952                 i_level++;
1953                 break;
1954             case 2:
1955                 *dest++ = ( ( last << 4 )&0xf0 ) | ( ( c >> 2 )&0x0f );
1956                 i_level++;
1957                 break;
1958             case 3:
1959                 *dest++ = ( ( last &0x03 ) << 6 ) | c;
1960                 i_level = 0;
1961         }
1962         last = c;
1963     }
1964
1965     *dest = '\0';
1966
1967     return dest - dest_start;
1968 }