X-Git-Url: https://git.sesse.net/?a=blobdiff_plain;f=modules%2Fdemux%2Flive555.cpp;h=79ac789846d8aeaa5881e45254930422b3dad346;hb=15b18c04b30d456147f53445f98fa138f8cfa20b;hp=118360699380b8e3f8ff5a42a29ae0da676e5ce3;hpb=b045fd3dea3cedf806b66c05149c795dc71a383b;p=vlc diff --git a/modules/demux/live555.cpp b/modules/demux/live555.cpp index 1183606993..79ac789846 100644 --- a/modules/demux/live555.cpp +++ b/modules/demux/live555.cpp @@ -99,17 +99,13 @@ vlc_module_begin () set_capability( "demux", 50 ) set_shortname( "RTP/RTSP") set_callbacks( Open, Close ) - add_shortcut( "live" ) - add_shortcut( "livedotcom" ) + add_shortcut( "live", "livedotcom" ) set_category( CAT_INPUT ) set_subcategory( SUBCAT_INPUT_DEMUX ) add_submodule () set_description( N_("RTSP/RTP access and demux") ) - add_shortcut( "rtsp" ) - add_shortcut( "sdp" ) - add_shortcut( "live" ) - add_shortcut( "livedotcom" ) + add_shortcut( "rtsp", "pnm", "sdp", "live", "livedotcom" ) set_capability( "access_demux", 0 ) set_callbacks( Open, Close ) add_bool( "rtsp-tcp", false, NULL, @@ -136,6 +132,7 @@ vlc_module_begin () change_safe() add_bool( "rtsp-kasenna", false, NULL, KASENNA_TEXT, KASENNA_LONGTEXT, true ) + change_safe() add_bool( "rtsp-wmserver", false, NULL, WMSERVER_TEXT, WMSERVER_LONGTEXT, true) change_safe() @@ -163,6 +160,7 @@ typedef struct bool b_muxed; bool b_quicktime; bool b_asf; + block_t *p_asf_block; bool b_discard_trunc; stream_t *p_out_muxed; /* for muxed stream */ @@ -224,6 +222,7 @@ struct demux_sys_t bool b_get_param; /* Does the server support GET_PARAMETER */ bool b_paused; /* Are we paused? */ + bool b_error; float f_seek_request;/* In case we receive a seek request while paused*/ }; @@ -272,10 +271,7 @@ static int Open ( vlc_object_t *p_this ) return VLC_EGENERIC; } } - else - { - var_Create( p_demux, "rtsp-caching", VLC_VAR_INTEGER|VLC_VAR_DOINHERIT ); - } + var_Create( p_demux, "rtsp-caching", VLC_VAR_INTEGER|VLC_VAR_DOINHERIT ); p_demux->pf_demux = Demux; p_demux->pf_control= Control; @@ -306,6 +302,7 @@ static int Open ( vlc_object_t *p_this ) p_sys->b_get_param = false; p_sys->b_paused = false; p_sys->f_seek_request = -1; + p_sys->b_error = false; /* parse URL for rtsp://[user:[passwd]@]serverip:port/options */ vlc_UrlParse( &p_sys->url, p_sys->psz_path, 0 ); @@ -345,7 +342,7 @@ static int Open ( vlc_object_t *p_this ) int i_read = stream_Read( p_demux->s, &p_sdp[i_sdp], i_sdp_max - i_sdp - 1 ); - if( !vlc_object_alive (p_demux) || p_demux->b_error ) + if( !vlc_object_alive (p_demux) ) { free( p_sdp ); goto error; @@ -367,7 +364,7 @@ static int Open ( vlc_object_t *p_this ) } i_sdp_max += 1000; - p_sdp = (uint8_t*)realloc( p_sdp, i_sdp_max ); + p_sdp = (uint8_t*)xrealloc( p_sdp, i_sdp_max ); } p_sys->p_sdp = (char*)p_sdp; } @@ -425,9 +422,13 @@ static void Close( vlc_object_t *p_this ) { demux_t *p_demux = (demux_t*)p_this; demux_sys_t *p_sys = p_demux->p_sys; - int i; - for( i = 0; i < p_sys->i_track; i++ ) + if( p_sys->rtsp && p_sys->ms ) p_sys->rtsp->teardownMediaSession( *p_sys->ms ); + if( p_sys->ms ) Medium::close( p_sys->ms ); + if( p_sys->rtsp ) RTSPClient::close( p_sys->rtsp ); + if( p_sys->env ) p_sys->env->reclaim(); + + for( int i = 0; i < p_sys->i_track; i++ ) { live_track_t *tk = p_sys->track[i]; @@ -439,16 +440,12 @@ static void Close( vlc_object_t *p_this ) if( p_sys->i_track ) free( p_sys->track ); if( p_sys->p_out_asf ) stream_Delete( p_sys->p_out_asf ); - if( p_sys->rtsp && p_sys->ms ) p_sys->rtsp->teardownMediaSession( *p_sys->ms ); if( p_sys->p_timeout ) { vlc_cancel( p_sys->p_timeout->handle ); vlc_join( p_sys->p_timeout->handle, NULL ); free( p_sys->p_timeout ); } - if( p_sys->ms ) Medium::close( p_sys->ms ); - if( p_sys->rtsp ) RTSPClient::close( p_sys->rtsp ); - if( p_sys->env ) p_sys->env->reclaim(); delete p_sys->scheduler; free( p_sys->p_sdp ); free( p_sys->psz_path ); @@ -468,7 +465,6 @@ static int Connect( demux_t *p_demux ) { demux_sys_t *p_sys = p_demux->p_sys; Authenticator authenticator; - bool b_firstpass = true; char *psz_user = NULL; char *psz_pwd = NULL; char *psz_url = NULL; @@ -477,34 +473,32 @@ static int Connect( demux_t *p_demux ) int i_http_port = 0; int i_ret = VLC_SUCCESS; - /* Create the url using the port number if available */ - if( p_sys->url.i_port == 0 ) - { - p_sys->url.i_port = 554; - if( asprintf( &psz_url, "rtsp://%s", p_sys->psz_path ) == -1 ) - return VLC_ENOMEM; - } - else - { - if( asprintf( &psz_url, "rtsp://%s:%d%s", p_sys->url.psz_host, - p_sys->url.i_port, p_sys->url.psz_path ) == -1 ) - return VLC_ENOMEM; - } - /* Get the user name and password */ if( p_sys->url.psz_username || p_sys->url.psz_password ) { + /* Create the URL by stripping away the username/password part */ + if( p_sys->url.i_port == 0 ) + p_sys->url.i_port = 554; + if( asprintf( &psz_url, "rtsp://%s:%d%s", + strempty( p_sys->url.psz_host ), + p_sys->url.i_port, + strempty( p_sys->url.psz_path ) ) == -1 ) + return VLC_ENOMEM; + psz_user = strdup( strempty( p_sys->url.psz_username ) ); psz_pwd = strdup( strempty( p_sys->url.psz_password ) ); } else { + if( asprintf( &psz_url, "rtsp://%s", p_sys->psz_path ) == -1 ) + return VLC_ENOMEM; + psz_user = var_CreateGetString( p_demux, "rtsp-user" ); psz_pwd = var_CreateGetString( p_demux, "rtsp-pwd" ); } createnew: - if( !vlc_object_alive (p_demux) || p_demux->b_error ) + if( !vlc_object_alive (p_demux) ) { i_ret = VLC_EGENERIC; goto bailout; @@ -515,7 +509,7 @@ createnew: if( ( p_sys->rtsp = RTSPClient::createNew( *p_sys->env, var_CreateGetInteger( p_demux, "verbose" ) > 1, - "VLC media player", i_http_port ) ) == NULL ) + "LibVLC/"VERSION, i_http_port ) ) == NULL ) { msg_Err( p_demux, "RTSPClient::createNew failed (%s)", p_sys->env->getResultMsg() ); @@ -594,14 +588,6 @@ describe: } msg_Dbg( p_demux, "DESCRIBE failed with %d: %s", i_code, psz_error ); - if( b_firstpass ) - { /* describeURL always returns an "RTSP/1.0 401 Unauthorized" the - * first time. This is a workaround to avoid asking for a - * user/passwd the first time the code passess here. */ - i_code = 0; - b_firstpass = false; - } - if( i_code == 401 ) { msg_Dbg( p_demux, "authentication failed" ); @@ -683,7 +669,7 @@ static int SessionsSetup( demux_t *p_demux ) Boolean bInit; live_track_t *tk; - if( !vlc_object_alive (p_demux) || p_demux->b_error ) + if( !vlc_object_alive (p_demux) ) { delete iter; return VLC_EGENERIC; @@ -712,7 +698,7 @@ static int SessionsSetup( demux_t *p_demux ) } if( !strcmp( sub->codecName(), "X-ASF-PF" ) ) - bInit = sub->initiate( 4 ); /* Constant ? */ + bInit = sub->initiate( 0 ); else bInit = sub->initiate(); @@ -780,12 +766,13 @@ static int SessionsSetup( demux_t *p_demux ) tk->p_es = NULL; tk->b_quicktime = false; tk->b_asf = false; + tk->p_asf_block = NULL; tk->b_muxed = false; tk->b_discard_trunc = false; tk->p_out_muxed = NULL; tk->waiting = 0; tk->b_rtcp_sync = false; - tk->i_pts = 0; + tk->i_pts = VLC_TS_INVALID; tk->i_npt = 0.; tk->i_buffer = 65536; tk->p_buffer = (uint8_t *)malloc( 65536 ); @@ -866,7 +853,7 @@ static int SessionsSetup( demux_t *p_demux ) i_extra ) ) ) { tk->fmt.i_extra = i_extra; - tk->fmt.p_extra = malloc( i_extra ); + tk->fmt.p_extra = xmalloc( i_extra ); memcpy( tk->fmt.p_extra, p_extra, i_extra ); delete[] p_extra; } @@ -886,7 +873,7 @@ static int SessionsSetup( demux_t *p_demux ) i_extra ) ) ) { tk->fmt.i_extra = i_extra; - tk->fmt.p_extra = malloc( i_extra ); + tk->fmt.p_extra = xmalloc( i_extra ); memcpy( tk->fmt.p_extra, p_extra, i_extra ); delete[] p_extra; } @@ -921,6 +908,7 @@ static int SessionsSetup( demux_t *p_demux ) if( !strcmp( sub->codecName(), "MPV" ) ) { tk->fmt.i_codec = VLC_CODEC_MPGV; + tk->fmt.b_packetized = false; } else if( !strcmp( sub->codecName(), "H263" ) || !strcmp( sub->codecName(), "H263-1998" ) || @@ -944,7 +932,7 @@ static int SessionsSetup( demux_t *p_demux ) i_extra ) ) ) { tk->fmt.i_extra = i_extra; - tk->fmt.p_extra = malloc( i_extra ); + tk->fmt.p_extra = xmalloc( i_extra ); memcpy( tk->fmt.p_extra, p_extra, i_extra ); delete[] p_extra; @@ -965,7 +953,7 @@ static int SessionsSetup( demux_t *p_demux ) i_extra ) ) ) { tk->fmt.i_extra = i_extra; - tk->fmt.p_extra = malloc( i_extra ); + tk->fmt.p_extra = xmalloc( i_extra ); memcpy( tk->fmt.p_extra, p_extra, i_extra ); delete[] p_extra; } @@ -1028,7 +1016,8 @@ static int SessionsSetup( demux_t *p_demux ) if( tk->p_es || tk->b_quicktime || tk->b_muxed || tk->b_asf ) { /* Append */ - p_sys->track = (live_track_t**)realloc( p_sys->track, sizeof( live_track_t ) * ( p_sys->i_track + 1 ) ); + p_sys->track = (live_track_t**)xrealloc( p_sys->track, + sizeof( live_track_t ) * ( p_sys->i_track + 1 ) ); p_sys->track[p_sys->i_track++] = tk; } else @@ -1155,7 +1144,7 @@ static int Demux( demux_t *p_demux ) if( p_sys->i_pcr > 0 ) { if( b_send_pcr ) - es_out_Control( p_demux->out, ES_OUT_SET_PCR, p_sys->i_pcr ); + es_out_Control( p_demux->out, ES_OUT_SET_PCR, 1 + p_sys->i_pcr ); } /* First warn we want to read data */ @@ -1193,7 +1182,7 @@ static int Demux( demux_t *p_demux ) es_out_Control( p_demux->out, ES_OUT_RESET_PCR ); tk->b_rtcp_sync = true; /* reset PCR */ - tk->i_pts = 0; + tk->i_pts = VLC_TS_INVALID; tk->i_npt = 0.; p_sys->i_pcr = 0; p_sys->i_npt = 0.; @@ -1235,7 +1224,7 @@ static int Demux( demux_t *p_demux ) msg_Warn( p_demux, "no data received in 10s, eof ?" ); return 0; } - return p_demux->b_error ? 0 : 1; + return p_sys->b_error ? 0 : 1; } /***************************************************************************** @@ -1322,7 +1311,7 @@ static int Control( demux_t *p_demux, int i_query, va_list args ) for( i = 0; i < p_sys->i_track; i++ ) { p_sys->track[i]->b_rtcp_sync = false; - p_sys->track[i]->i_pts = 0; + p_sys->track[i]->i_pts = VLC_TS_INVALID; } /* Retrieve the starttime if possible */ @@ -1434,7 +1423,7 @@ static int Control( demux_t *p_demux, int i_query, va_list args ) return VLC_SUCCESS; if( ( b_pause && !p_sys->rtsp->pauseMediaSession( *p_sys->ms ) ) || ( !b_pause && !p_sys->rtsp->playMediaSession( *p_sys->ms, - p_sys->f_seek_request ) ) ) + p_sys->f_seek_request, -1.0f, p_sys->ms->scale() ) ) ) { msg_Err( p_demux, "PLAY or PAUSE failed %s", p_sys->env->getResultMsg() ); return VLC_EGENERIC; @@ -1461,7 +1450,7 @@ static int Control( demux_t *p_demux, int i_query, va_list args ) { live_track_t *tk = p_sys->track[i]; tk->b_rtcp_sync = false; - tk->i_pts = 0; + tk->i_pts = VLC_TS_INVALID; p_sys->i_pcr = 0; es_out_Control( p_demux->out, ES_OUT_RESET_PCR ); } @@ -1514,6 +1503,7 @@ static int RollOverTcp( demux_t *p_demux ) if( tk->b_muxed ) stream_Delete( tk->p_out_muxed ); if( tk->p_es ) es_out_Del( p_demux->out, tk->p_es ); + if( tk->p_asf_block ) block_Release( tk->p_asf_block ); es_format_Clean( &tk->fmt ); free( tk->p_buffer ); free( tk ); @@ -1531,6 +1521,7 @@ static int RollOverTcp( demux_t *p_demux ) p_sys->i_track = 0; p_sys->b_no_data = true; p_sys->i_no_data_ti = 0; + p_sys->p_out_asf = NULL; /* Reopen rtsp client */ if( ( i_return = Connect( p_demux ) ) != VLC_SUCCESS ) @@ -1562,6 +1553,91 @@ error: } +/***************************************************************************** + * + *****************************************************************************/ +static block_t *StreamParseAsf( demux_t *p_demux, live_track_t *tk, + bool b_marker, + const uint8_t *p_data, unsigned i_size ) +{ + const unsigned i_packet_size = p_demux->p_sys->asfh.i_min_data_packet_size; + block_t *p_list = NULL; + + while( i_size >= 4 ) + { + unsigned i_flags = p_data[0]; + unsigned i_length_offset = (p_data[1] << 16) | + (p_data[2] << 8) | + (p_data[3] ); + bool b_key = i_flags & 0x80; + bool b_length = i_flags & 0x40; + bool b_relative_ts = i_flags & 0x20; + bool b_duration = i_flags & 0x10; + bool b_location_id = i_flags & 0x08; + + //msg_Dbg( p_demux, "ASF: marker=%d size=%d : %c=%d id=%d", + // b_marker, i_size, b_length ? 'L' : 'O', i_length_offset ); + unsigned i_header_size = 4; + if( b_relative_ts ) + i_header_size += 4; + if( b_duration ) + i_header_size += 4; + if( b_location_id ) + i_header_size += 4; + + if( i_header_size > i_size ) + { + msg_Warn( p_demux, "Invalid header size" ); + break; + } + + /* XXX + * When b_length is true, the streams I found do not seems to respect + * the documentation. + * From them, I have failed to find which choice between '__MIN()' or + * 'i_length_offset - i_header_size' is the right one. + */ + unsigned i_payload; + if( b_length ) + i_payload = __MIN( i_length_offset, i_size - i_header_size); + else + i_payload = i_size - i_header_size; + + if( !tk->p_asf_block ) + { + tk->p_asf_block = block_New( p_demux, i_packet_size ); + if( !tk->p_asf_block ) + break; + tk->p_asf_block->i_buffer = 0; + } + unsigned i_offset = b_length ? 0 : i_length_offset; + if( i_offset == tk->p_asf_block->i_buffer && i_offset + i_payload <= i_packet_size ) + { + memcpy( &tk->p_asf_block->p_buffer[i_offset], &p_data[i_header_size], i_payload ); + tk->p_asf_block->i_buffer += i_payload; + if( b_marker ) + { + /* We have a complete packet */ + tk->p_asf_block->i_buffer = i_packet_size; + block_ChainAppend( &p_list, tk->p_asf_block ); + tk->p_asf_block = NULL; + } + } + else + { + /* Reset on broken stream */ + msg_Err( p_demux, "Broken packet detected (%d vs %d or %d + %d vs %d)", + i_offset, tk->p_asf_block->i_buffer, i_offset, i_payload, i_packet_size); + tk->p_asf_block->i_buffer = 0; + } + + /* */ + p_data += i_header_size + i_payload; + i_size -= i_header_size + i_payload; + } + return p_list; +} + /***************************************************************************** * *****************************************************************************/ @@ -1617,7 +1693,7 @@ static void StreamRead( void *p_private, unsigned int i_size, atomLength <= INT_MAX ) { tk->fmt.i_extra = atomLength-8; - tk->fmt.p_extra = malloc( tk->fmt.i_extra ); + tk->fmt.p_extra = xmalloc( tk->fmt.i_extra ); memcpy(tk->fmt.p_extra, pos+8, atomLength-8); break; } @@ -1627,7 +1703,7 @@ static void StreamRead( void *p_private, unsigned int i_size, else { tk->fmt.i_extra = qtState.sdAtomSize - 16; - tk->fmt.p_extra = malloc( tk->fmt.i_extra ); + tk->fmt.p_extra = xmalloc( tk->fmt.i_extra ); memcpy( tk->fmt.p_extra, &sdAtom[12], tk->fmt.i_extra ); } } @@ -1716,10 +1792,9 @@ static void StreamRead( void *p_private, unsigned int i_size, } else if( tk->b_asf ) { - int i_copy = __MIN( p_sys->asfh.i_min_data_packet_size, (int)i_size ); - p_block = block_New( p_demux, p_sys->asfh.i_min_data_packet_size ); - - memcpy( p_block->p_buffer, tk->p_buffer, i_copy ); + p_block = StreamParseAsf( p_demux, tk, + tk->sub->rtpSource()->curPacketMarkerBit(), + tk->p_buffer, i_size ); } else { @@ -1732,32 +1807,26 @@ static void StreamRead( void *p_private, unsigned int i_size, p_sys->i_pcr = i_pts; } - if( (i_pts != tk->i_pts) && (!tk->b_muxed) ) - { - p_block->i_pts = i_pts; - } - /* Update our global npt value */ if( tk->i_npt > 0 && tk->i_npt > p_sys->i_npt && tk->i_npt < p_sys->i_npt_length) p_sys->i_npt = tk->i_npt; - if( !tk->b_muxed ) + if( p_block ) { - /*FIXME: for h264 you should check that packetization-mode=1 in sdp-file */ - p_block->i_dts = ( tk->fmt.i_codec == VLC_CODEC_MPGV ) ? 0 : i_pts; - } + if( !tk->b_muxed && !tk->b_asf ) + { + if( i_pts != tk->i_pts ) + p_block->i_pts = VLC_TS_0 + i_pts; + /*FIXME: for h264 you should check that packetization-mode=1 in sdp-file */ + p_block->i_dts = ( tk->fmt.i_codec == VLC_CODEC_MPGV ) ? VLC_TS_INVALID : (VLC_TS_0 + i_pts); + } - if( tk->b_muxed ) - { - stream_DemuxSend( tk->p_out_muxed, p_block ); - } - else if( tk->b_asf ) - { - stream_DemuxSend( p_sys->p_out_asf, p_block ); - } - else - { - es_out_Send( p_demux->out, tk->p_es, p_block ); + if( tk->b_muxed ) + stream_DemuxSend( tk->p_out_muxed, p_block ); + else if( tk->b_asf ) + stream_DemuxSend( p_sys->p_out_asf, p_block ); + else + es_out_Send( p_demux->out, tk->p_es, p_block ); } /* warn that's ok */ @@ -1786,7 +1855,7 @@ static void StreamClose( void *p_private ) msg_Dbg( p_demux, "StreamClose" ); p_sys->event = 0xff; - p_demux->b_error = true; + p_sys->b_error = true; }