2010-10-29 10:43:57 +02:00
/*
* RTSP demuxer
* Copyright ( c ) 2002 Fabrice Bellard
*
* This file is part of FFmpeg .
*
* FFmpeg is free software ; you can redistribute it and / or
* modify it under the terms of the GNU Lesser General Public
* License as published by the Free Software Foundation ; either
* version 2.1 of the License , or ( at your option ) any later version .
*
* FFmpeg is distributed in the hope that it will be useful ,
* but WITHOUT ANY WARRANTY ; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE . See the GNU
* Lesser General Public License for more details .
*
* You should have received a copy of the GNU Lesser General Public
* License along with FFmpeg ; if not , write to the Free Software
* Foundation , Inc . , 51 Franklin Street , Fifth Floor , Boston , MA 02110 - 1301 USA
*/
# include "libavutil/avstring.h"
# include "libavutil/intreadwrite.h"
2011-06-04 13:58:23 +02:00
# include "libavutil/mathematics.h"
2011-05-24 08:44:10 +02:00
# include "libavutil/opt.h"
2010-10-29 10:43:57 +02:00
# include "avformat.h"
# include "internal.h"
# include "network.h"
# include "os_support.h"
# include "rtsp.h"
# include "rdt.h"
2011-03-31 16:40:31 +02:00
# include "url.h"
2010-10-29 10:43:57 +02:00
static int rtsp_read_play ( AVFormatContext * s )
{
RTSPState * rt = s - > priv_data ;
RTSPMessageHeader reply1 , * reply = & reply1 ;
int i ;
char cmd [ 1024 ] ;
av_log ( s , AV_LOG_DEBUG , " hello state=%d \n " , rt - > state ) ;
rt - > nb_byes = 0 ;
if ( ! ( rt - > server_type = = RTSP_SERVER_REAL & & rt - > need_subscription ) ) {
2011-01-05 22:22:15 +01:00
if ( rt - > transport = = RTSP_TRANSPORT_RTP ) {
for ( i = 0 ; i < rt - > nb_rtsp_streams ; i + + ) {
RTSPStream * rtsp_st = rt - > rtsp_streams [ i ] ;
RTPDemuxContext * rtpctx = rtsp_st - > transport_priv ;
if ( ! rtpctx )
continue ;
ff_rtp_reset_packet_queue ( rtpctx ) ;
rtpctx - > last_rtcp_ntp_time = AV_NOPTS_VALUE ;
rtpctx - > first_rtcp_ntp_time = AV_NOPTS_VALUE ;
rtpctx - > base_timestamp = 0 ;
rtpctx - > rtcp_ts_offset = 0 ;
}
}
2010-10-29 10:43:57 +02:00
if ( rt - > state = = RTSP_STATE_PAUSED ) {
cmd [ 0 ] = 0 ;
} else {
snprintf ( cmd , sizeof ( cmd ) ,
2011-03-22 22:38:44 +01:00
" Range: npt=% " PRId64 " .%03 " PRId64 " - \r \n " ,
rt - > seek_timestamp / AV_TIME_BASE ,
rt - > seek_timestamp / ( AV_TIME_BASE / 1000 ) % 1000 ) ;
2010-10-29 10:43:57 +02:00
}
ff_rtsp_send_cmd ( s , " PLAY " , rt - > control_uri , cmd , reply , NULL ) ;
if ( reply - > status_code ! = RTSP_STATUS_OK ) {
return - 1 ;
}
2011-01-05 22:22:50 +01:00
if ( rt - > transport = = RTSP_TRANSPORT_RTP & &
reply - > range_start ! = AV_NOPTS_VALUE ) {
2010-10-29 10:43:57 +02:00
for ( i = 0 ; i < rt - > nb_rtsp_streams ; i + + ) {
RTSPStream * rtsp_st = rt - > rtsp_streams [ i ] ;
RTPDemuxContext * rtpctx = rtsp_st - > transport_priv ;
AVStream * st = NULL ;
2011-01-05 22:22:50 +01:00
if ( ! rtpctx | | rtsp_st - > stream_index < 0 )
2010-10-29 10:43:57 +02:00
continue ;
2011-01-05 22:22:50 +01:00
st = s - > streams [ rtsp_st - > stream_index ] ;
2011-01-05 22:23:12 +01:00
rtpctx - > range_start_offset =
av_rescale_q ( reply - > range_start , AV_TIME_BASE_Q ,
st - > time_base ) ;
2010-10-29 10:43:57 +02:00
}
}
}
rt - > state = RTSP_STATE_STREAMING ;
return 0 ;
}
2011-01-24 10:49:04 +01:00
/* pause the stream */
static int rtsp_read_pause ( AVFormatContext * s )
{
RTSPState * rt = s - > priv_data ;
RTSPMessageHeader reply1 , * reply = & reply1 ;
if ( rt - > state ! = RTSP_STATE_STREAMING )
return 0 ;
else if ( ! ( rt - > server_type = = RTSP_SERVER_REAL & & rt - > need_subscription ) ) {
ff_rtsp_send_cmd ( s , " PAUSE " , rt - > control_uri , NULL , reply , NULL ) ;
if ( reply - > status_code ! = RTSP_STATUS_OK ) {
return - 1 ;
}
}
rt - > state = RTSP_STATE_PAUSED ;
return 0 ;
}
2010-10-29 10:43:57 +02:00
int ff_rtsp_setup_input_streams ( AVFormatContext * s , RTSPMessageHeader * reply )
{
RTSPState * rt = s - > priv_data ;
char cmd [ 1024 ] ;
unsigned char * content = NULL ;
int ret ;
/* describe the stream */
snprintf ( cmd , sizeof ( cmd ) ,
" Accept: application/sdp \r \n " ) ;
if ( rt - > server_type = = RTSP_SERVER_REAL ) {
/**
* The Require : attribute is needed for proper streaming from
* Realmedia servers .
*/
av_strlcat ( cmd ,
" Require: com.real.retain-entity-for-setup \r \n " ,
sizeof ( cmd ) ) ;
}
ff_rtsp_send_cmd ( s , " DESCRIBE " , rt - > control_uri , cmd , reply , & content ) ;
if ( ! content )
return AVERROR_INVALIDDATA ;
if ( reply - > status_code ! = RTSP_STATUS_OK ) {
av_freep ( & content ) ;
return AVERROR_INVALIDDATA ;
}
av_log ( s , AV_LOG_VERBOSE , " SDP: \n %s \n " , content ) ;
/* now we got the SDP description, we parse it */
ret = ff_sdp_parse ( s , ( const char * ) content ) ;
av_freep ( & content ) ;
if ( ret < 0 )
2011-01-28 03:12:22 +01:00
return ret ;
2010-10-29 10:43:57 +02:00
return 0 ;
}
static int rtsp_probe ( AVProbeData * p )
{
if ( av_strstart ( p - > filename , " rtsp: " , NULL ) )
return AVPROBE_SCORE_MAX ;
return 0 ;
}
static int rtsp_read_header ( AVFormatContext * s ,
AVFormatParameters * ap )
{
RTSPState * rt = s - > priv_data ;
int ret ;
ret = ff_rtsp_connect ( s ) ;
if ( ret )
return ret ;
rt - > real_setup_cache = av_mallocz ( 2 * s - > nb_streams * sizeof ( * rt - > real_setup_cache ) ) ;
if ( ! rt - > real_setup_cache )
return AVERROR ( ENOMEM ) ;
2010-11-07 00:26:49 +01:00
rt - > real_setup = rt - > real_setup_cache + s - > nb_streams ;
2010-10-29 10:43:57 +02:00
2011-05-24 08:44:10 +02:00
if ( rt - > initial_pause ) {
2010-10-29 10:43:57 +02:00
/* do not start immediately */
} else {
if ( rtsp_read_play ( s ) < 0 ) {
ff_rtsp_close_streams ( s ) ;
ff_rtsp_close_connections ( s ) ;
return AVERROR_INVALIDDATA ;
}
}
return 0 ;
}
int ff_rtsp_tcp_read_packet ( AVFormatContext * s , RTSPStream * * prtsp_st ,
uint8_t * buf , int buf_size )
{
RTSPState * rt = s - > priv_data ;
int id , len , i , ret ;
RTSPStream * rtsp_st ;
2011-01-29 17:46:18 +01:00
av_dlog ( s , " tcp_read_packet: \n " ) ;
2010-10-29 10:43:57 +02:00
redo :
for ( ; ; ) {
RTSPMessageHeader reply ;
2011-01-02 11:06:21 +01:00
ret = ff_rtsp_read_reply ( s , & reply , NULL , 1 , NULL ) ;
2010-10-29 10:43:57 +02:00
if ( ret < 0 )
return ret ;
if ( ret = = 1 ) /* received '$' */
break ;
/* XXX: parse message */
if ( rt - > state ! = RTSP_STATE_STREAMING )
return 0 ;
}
2011-03-31 16:40:31 +02:00
ret = ffurl_read_complete ( rt - > rtsp_hd , buf , 3 ) ;
2010-10-29 10:43:57 +02:00
if ( ret ! = 3 )
return - 1 ;
id = buf [ 0 ] ;
len = AV_RB16 ( buf + 1 ) ;
2011-01-29 17:46:18 +01:00
av_dlog ( s , " id=%d len=%d \n " , id , len ) ;
2011-09-20 21:58:15 +02:00
if ( len > buf_size | | len < 8 )
2010-10-29 10:43:57 +02:00
goto redo ;
/* get the data */
2011-03-31 16:40:31 +02:00
ret = ffurl_read_complete ( rt - > rtsp_hd , buf , len ) ;
2010-10-29 10:43:57 +02:00
if ( ret ! = len )
return - 1 ;
if ( rt - > transport = = RTSP_TRANSPORT_RDT & &
ff_rdt_parse_header ( buf , len , & id , NULL , NULL , NULL , NULL ) < 0 )
return - 1 ;
/* find the matching stream */
for ( i = 0 ; i < rt - > nb_rtsp_streams ; i + + ) {
rtsp_st = rt - > rtsp_streams [ i ] ;
if ( id > = rtsp_st - > interleaved_min & &
id < = rtsp_st - > interleaved_max )
goto found ;
}
goto redo ;
found :
* prtsp_st = rtsp_st ;
return len ;
}
2011-01-10 11:45:24 +01:00
static int resetup_tcp ( AVFormatContext * s )
{
RTSPState * rt = s - > priv_data ;
char host [ 1024 ] ;
int port ;
av_url_split ( NULL , 0 , NULL , 0 , host , sizeof ( host ) , & port , NULL , 0 ,
s - > filename ) ;
ff_rtsp_undo_setup ( s ) ;
return ff_rtsp_make_setup_request ( s , host , port , RTSP_LOWER_TRANSPORT_TCP ,
rt - > real_challenge ) ;
}
2010-10-29 10:43:57 +02:00
static int rtsp_read_packet ( AVFormatContext * s , AVPacket * pkt )
{
RTSPState * rt = s - > priv_data ;
int ret ;
RTSPMessageHeader reply1 , * reply = & reply1 ;
char cmd [ 1024 ] ;
2011-01-10 11:45:24 +01:00
retry :
2010-10-29 10:43:57 +02:00
if ( rt - > server_type = = RTSP_SERVER_REAL ) {
int i ;
for ( i = 0 ; i < s - > nb_streams ; i + + )
rt - > real_setup [ i ] = s - > streams [ i ] - > discard ;
if ( ! rt - > need_subscription ) {
if ( memcmp ( rt - > real_setup , rt - > real_setup_cache ,
sizeof ( enum AVDiscard ) * s - > nb_streams ) ) {
snprintf ( cmd , sizeof ( cmd ) ,
" Unsubscribe: %s \r \n " ,
rt - > last_subscription ) ;
ff_rtsp_send_cmd ( s , " SET_PARAMETER " , rt - > control_uri ,
cmd , reply , NULL ) ;
if ( reply - > status_code ! = RTSP_STATUS_OK )
return AVERROR_INVALIDDATA ;
rt - > need_subscription = 1 ;
}
}
if ( rt - > need_subscription ) {
int r , rule_nr , first = 1 ;
memcpy ( rt - > real_setup_cache , rt - > real_setup ,
sizeof ( enum AVDiscard ) * s - > nb_streams ) ;
rt - > last_subscription [ 0 ] = 0 ;
snprintf ( cmd , sizeof ( cmd ) ,
" Subscribe: " ) ;
for ( i = 0 ; i < rt - > nb_rtsp_streams ; i + + ) {
rule_nr = 0 ;
for ( r = 0 ; r < s - > nb_streams ; r + + ) {
2011-02-11 22:36:47 +01:00
if ( s - > streams [ r ] - > id = = i ) {
2010-10-29 10:43:57 +02:00
if ( s - > streams [ r ] - > discard ! = AVDISCARD_ALL ) {
if ( ! first )
av_strlcat ( rt - > last_subscription , " , " ,
sizeof ( rt - > last_subscription ) ) ;
ff_rdt_subscribe_rule (
rt - > last_subscription ,
sizeof ( rt - > last_subscription ) , i , rule_nr ) ;
first = 0 ;
}
rule_nr + + ;
}
}
}
av_strlcatf ( cmd , sizeof ( cmd ) , " %s \r \n " , rt - > last_subscription ) ;
ff_rtsp_send_cmd ( s , " SET_PARAMETER " , rt - > control_uri ,
cmd , reply , NULL ) ;
if ( reply - > status_code ! = RTSP_STATUS_OK )
return AVERROR_INVALIDDATA ;
rt - > need_subscription = 0 ;
if ( rt - > state = = RTSP_STATE_STREAMING )
rtsp_read_play ( s ) ;
}
}
ret = ff_rtsp_fetch_packet ( s , pkt ) ;
2011-01-10 11:45:24 +01:00
if ( ret < 0 ) {
2011-02-19 19:14:11 +01:00
if ( ret = = AVERROR ( ETIMEDOUT ) & & ! rt - > packets ) {
2011-01-10 11:45:24 +01:00
if ( rt - > lower_transport = = RTSP_LOWER_TRANSPORT_UDP & &
rt - > lower_transport_mask & ( 1 < < RTSP_LOWER_TRANSPORT_TCP ) ) {
RTSPMessageHeader reply1 , * reply = & reply1 ;
av_log ( s , AV_LOG_WARNING , " UDP timeout, retrying with TCP \n " ) ;
if ( rtsp_read_pause ( s ) ! = 0 )
return - 1 ;
// TEARDOWN is required on Real-RTSP, but might make
// other servers close the connection.
if ( rt - > server_type = = RTSP_SERVER_REAL )
ff_rtsp_send_cmd ( s , " TEARDOWN " , rt - > control_uri , NULL ,
reply , NULL ) ;
rt - > session_id [ 0 ] = ' \0 ' ;
if ( resetup_tcp ( s ) = = 0 ) {
rt - > state = RTSP_STATE_IDLE ;
rt - > need_subscription = 1 ;
if ( rtsp_read_play ( s ) ! = 0 )
return - 1 ;
goto retry ;
}
}
}
2010-10-29 10:43:57 +02:00
return ret ;
2011-01-10 11:45:24 +01:00
}
rt - > packets + + ;
2010-10-29 10:43:57 +02:00
/* send dummy request to keep TCP connection alive */
if ( ( av_gettime ( ) - rt - > last_cmd_time ) / 1000000 > = rt - > timeout / 2 ) {
2011-05-09 19:11:16 +02:00
if ( rt - > server_type = = RTSP_SERVER_WMS | |
( rt - > server_type ! = RTSP_SERVER_REAL & &
rt - > get_parameter_supported ) ) {
2010-10-29 10:43:57 +02:00
ff_rtsp_send_cmd_async ( s , " GET_PARAMETER " , rt - > control_uri , NULL ) ;
} else {
ff_rtsp_send_cmd_async ( s , " OPTIONS " , " * " , NULL ) ;
}
}
return 0 ;
}
static int rtsp_read_seek ( AVFormatContext * s , int stream_index ,
int64_t timestamp , int flags )
{
RTSPState * rt = s - > priv_data ;
rt - > seek_timestamp = av_rescale_q ( timestamp ,
s - > streams [ stream_index ] - > time_base ,
AV_TIME_BASE_Q ) ;
switch ( rt - > state ) {
default :
case RTSP_STATE_IDLE :
break ;
case RTSP_STATE_STREAMING :
if ( rtsp_read_pause ( s ) ! = 0 )
return - 1 ;
rt - > state = RTSP_STATE_SEEKING ;
if ( rtsp_read_play ( s ) ! = 0 )
return - 1 ;
break ;
case RTSP_STATE_PAUSED :
rt - > state = RTSP_STATE_IDLE ;
break ;
}
return 0 ;
}
static int rtsp_read_close ( AVFormatContext * s )
{
RTSPState * rt = s - > priv_data ;
ff_rtsp_send_cmd_async ( s , " TEARDOWN " , rt - > control_uri , NULL ) ;
ff_rtsp_close_streams ( s ) ;
ff_rtsp_close_connections ( s ) ;
ff_network_close ( ) ;
rt - > real_setup = NULL ;
av_freep ( & rt - > real_setup_cache ) ;
return 0 ;
}
2011-05-24 08:44:10 +02:00
static const AVOption options [ ] = {
2011-10-04 07:38:01 +02:00
{ " initial_pause " , " Don't start playing the stream immediately " , offsetof ( RTSPState , initial_pause ) , AV_OPT_TYPE_INT , { . dbl = 0 } , 0 , 1 , AV_OPT_FLAG_DECODING_PARAM } ,
2011-05-24 08:44:10 +02:00
{ NULL } ,
} ;
const AVClass rtsp_demuxer_class = {
. class_name = " RTSP demuxer " ,
. item_name = av_default_item_name ,
. option = options ,
. version = LIBAVUTIL_VERSION_INT ,
} ;
2011-01-25 23:03:28 +01:00
AVInputFormat ff_rtsp_demuxer = {
2011-07-16 22:18:12 +02:00
. name = " rtsp " ,
. long_name = NULL_IF_CONFIG_SMALL ( " RTSP input format " ) ,
. priv_data_size = sizeof ( RTSPState ) ,
. read_probe = rtsp_probe ,
. read_header = rtsp_read_header ,
. read_packet = rtsp_read_packet ,
. read_close = rtsp_read_close ,
. read_seek = rtsp_read_seek ,
2010-10-29 10:43:57 +02:00
. flags = AVFMT_NOFILE ,
. read_play = rtsp_read_play ,
. read_pause = rtsp_read_pause ,
2011-05-24 08:44:10 +02:00
. priv_class = & rtsp_demuxer_class ,
2010-10-29 10:43:57 +02:00
} ;