2013-03-17 07:45:21 +08:00
//
// ZoneMinder RTSP Class Implementation, $Date$, $Revision$
// Copyright (C) 2001-2008 Philip Coombes
//
// This program is free software; you can redistribute it and/or
// modify it under the terms of the GNU General Public License
// as published by the Free Software Foundation; either version 2
// of the License, or (at your option) any later version.
//
// This program 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 General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with this program; if not, write to the Free Software
// Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
//
# include "zm.h"
# if HAVE_LIBAVFORMAT
# include "zm_rtsp.h"
# include "zm_rtp_data.h"
# include "zm_rtp_ctrl.h"
# include "zm_db.h"
# include <sys/time.h>
# include <signal.h>
# include <stdlib.h>
# include <errno.h>
int RtspThread : : smMinDataPort = 0 ;
int RtspThread : : smMaxDataPort = 0 ;
RtspThread : : PortSet RtspThread : : smAssignedPorts ;
bool RtspThread : : sendCommand ( std : : string message )
{
2016-04-04 22:11:48 +08:00
if ( mNeedAuth ) {
StringVector parts = split ( message , " " ) ;
if ( parts . size ( ) > 1 )
message + = mAuthenticator - > getAuthHeader ( parts [ 0 ] , parts [ 1 ] ) ;
}
message + = stringtf ( " User-Agent: ZoneMinder/%s \r \n " , ZM_VERSION ) ;
message + = stringtf ( " CSeq: %d \r \n \r \n " , + + mSeq ) ;
Debug ( 2 , " Sending RTSP message: %s " , message . c_str ( ) ) ;
if ( mMethod = = RTP_RTSP_HTTP )
{
message = base64Encode ( message ) ;
Debug ( 2 , " Sending encoded RTSP message: %s " , message . c_str ( ) ) ;
if ( mRtspSocket2 . send ( message . c_str ( ) , message . size ( ) ) ! = ( int ) message . length ( ) )
2013-03-17 07:45:21 +08:00
{
2016-04-04 22:11:48 +08:00
Error ( " Unable to send message '%s': %s " , message . c_str ( ) , strerror ( errno ) ) ;
return ( false ) ;
2013-03-17 07:45:21 +08:00
}
2016-04-04 22:11:48 +08:00
}
else
{
if ( mRtspSocket . send ( message . c_str ( ) , message . size ( ) ) ! = ( int ) message . length ( ) )
2013-03-17 07:45:21 +08:00
{
2016-04-04 22:11:48 +08:00
Error ( " Unable to send message '%s': %s " , message . c_str ( ) , strerror ( errno ) ) ;
return ( false ) ;
2013-03-17 07:45:21 +08:00
}
2016-04-04 22:11:48 +08:00
}
return ( true ) ;
2013-03-17 07:45:21 +08:00
}
bool RtspThread : : recvResponse ( std : : string & response )
{
2016-04-04 22:11:48 +08:00
if ( mRtspSocket . recv ( response ) < 0 )
Error ( " Recv failed; %s " , strerror ( errno ) ) ;
Debug ( 2 , " Received RTSP response: %s (%zd bytes) " , response . c_str ( ) , response . size ( ) ) ;
float respVer = 0 ;
respCode = - 1 ;
char respText [ ZM_NETWORK_BUFSIZ ] ;
if ( sscanf ( response . c_str ( ) , " RTSP/%f %3d %[^ \r \n ] \r \n " , & respVer , & respCode , respText ) ! = 3 )
{
if ( isalnum ( response [ 0 ] ) )
2013-03-17 07:45:21 +08:00
{
2016-04-04 22:11:48 +08:00
Error ( " Response parse failure in '%s' " , response . c_str ( ) ) ;
2013-03-17 07:45:21 +08:00
}
2016-04-04 22:11:48 +08:00
else
2013-03-17 07:45:21 +08:00
{
2016-04-04 22:11:48 +08:00
Error ( " Response parse failure, %zd bytes follow " , response . size ( ) ) ;
if ( response . size ( ) )
Hexdump ( Logger : : ERROR , response . data ( ) , min ( response . size ( ) , 16 ) ) ;
2013-03-17 07:45:21 +08:00
}
2016-04-04 22:11:48 +08:00
return ( false ) ;
}
if ( respCode = = 401 )
{
Debug ( 2 , " Got 401 access denied response code, check WWW-Authenticate header and retry " ) ;
mAuthenticator - > checkAuthResponse ( response ) ;
mNeedAuth = true ;
return ( false ) ;
}
else if ( respCode ! = 200 )
{
Error ( " Unexpected response code %d, text is '%s' " , respCode , respText ) ;
return ( false ) ;
}
return ( true ) ;
2013-03-17 07:45:21 +08:00
}
int RtspThread : : requestPorts ( )
{
2016-04-04 22:11:48 +08:00
if ( ! smMinDataPort )
{
char sql [ ZM_SQL_SML_BUFSIZ ] ;
strncpy ( sql , " select Id from Monitors where Function != 'None' and Type = 'Remote' and Protocol = 'rtsp' and Method = 'rtpUni' order by Id asc " , sizeof ( sql ) ) ;
if ( mysql_query ( & dbconn , sql ) )
2013-03-17 07:45:21 +08:00
{
2016-04-04 22:11:48 +08:00
Error ( " Can't run query: %s " , mysql_error ( & dbconn ) ) ;
exit ( mysql_errno ( & dbconn ) ) ;
}
2013-03-17 07:45:21 +08:00
2016-04-04 22:11:48 +08:00
MYSQL_RES * result = mysql_store_result ( & dbconn ) ;
if ( ! result )
{
Error ( " Can't use query result: %s " , mysql_error ( & dbconn ) ) ;
exit ( mysql_errno ( & dbconn ) ) ;
2013-03-17 07:45:21 +08:00
}
2016-04-04 22:11:48 +08:00
int nMonitors = mysql_num_rows ( result ) ;
int position = 0 ;
if ( nMonitors )
2013-03-17 07:45:21 +08:00
{
2016-04-04 22:11:48 +08:00
for ( int i = 0 ; MYSQL_ROW dbrow = mysql_fetch_row ( result ) ; i + + )
{
int id = atoi ( dbrow [ 0 ] ) ;
if ( mId = = id )
2013-03-17 07:45:21 +08:00
{
2016-04-04 22:11:48 +08:00
position = i ;
break ;
2013-03-17 07:45:21 +08:00
}
2016-04-04 22:11:48 +08:00
}
2013-03-17 07:45:21 +08:00
}
2016-04-04 22:11:48 +08:00
else
{
// Minor hack for testing when not strictly enabled
nMonitors = 1 ;
position = 0 ;
}
mysql_free_result ( result ) ;
int portRange = int ( ( ( config . max_rtp_port - config . min_rtp_port ) + 1 ) / nMonitors ) ;
smMinDataPort = config . min_rtp_port + ( position * portRange ) ;
smMaxDataPort = smMinDataPort + portRange - 1 ;
Debug ( 2 , " Assigned RTP port range is %d-%d " , smMinDataPort , smMaxDataPort ) ;
}
for ( int i = smMinDataPort ; i < = smMaxDataPort ; i + + )
{
PortSet : : const_iterator iter = smAssignedPorts . find ( i ) ;
if ( iter = = smAssignedPorts . end ( ) )
{
smAssignedPorts . insert ( i ) ;
return ( i ) ;
}
}
Panic ( " Can assign RTP port, no ports left in pool " ) ;
return ( - 1 ) ;
2013-03-17 07:45:21 +08:00
}
void RtspThread : : releasePorts ( int port )
{
2016-04-04 22:11:48 +08:00
if ( port > 0 )
smAssignedPorts . erase ( port ) ;
2013-03-17 07:45:21 +08:00
}
2015-08-21 23:29:54 +08:00
RtspThread : : RtspThread ( int id , RtspMethod method , const std : : string & protocol , const std : : string & host , const std : : string & port , const std : : string & path , const std : : string & auth , bool rtsp_describe ) :
2016-04-04 22:11:48 +08:00
mId ( id ) ,
mMethod ( method ) ,
mProtocol ( protocol ) ,
mHost ( host ) ,
mPort ( port ) ,
mPath ( path ) ,
mRtspDescribe ( rtsp_describe ) ,
mSessDesc ( 0 ) ,
mFormatContext ( 0 ) ,
mSeq ( 0 ) ,
mSession ( 0 ) ,
mSsrc ( 0 ) ,
mDist ( UNDEFINED ) ,
mRtpTime ( 0 ) ,
mStop ( false )
2013-03-17 07:45:21 +08:00
{
2016-04-04 22:11:48 +08:00
mUrl = mProtocol + " :// " + mHost + " : " + mPort ;
if ( ! mPath . empty ( ) )
{
if ( mPath [ 0 ] = = ' / ' )
mUrl + = mPath ;
2014-04-29 05:37:31 +08:00
else
2016-04-04 22:11:48 +08:00
mUrl + = ' / ' + mPath ;
}
mSsrc = rand ( ) ;
Debug ( 2 , " RTSP Local SSRC is %x " , mSsrc ) ;
if ( mMethod = = RTP_RTSP_HTTP )
mHttpSession = stringtf ( " %d " , rand ( ) ) ;
mNeedAuth = false ;
StringVector parts = split ( auth , " : " ) ;
if ( parts . size ( ) > 1 )
mAuthenticator = new zm : : Authenticator ( parts [ 0 ] , parts [ 1 ] ) ;
else
mAuthenticator = new zm : : Authenticator ( parts [ 0 ] , " " ) ;
2013-03-17 07:45:21 +08:00
}
RtspThread : : ~ RtspThread ( )
{
2016-04-04 22:11:48 +08:00
if ( mFormatContext )
{
2015-05-29 23:38:02 +08:00
# if LIBAVFORMAT_VERSION_CHECK(52, 96, 0, 96, 0)
2016-04-04 22:11:48 +08:00
avformat_free_context ( mFormatContext ) ;
2015-02-18 20:25:31 +08:00
# else
2016-04-04 22:11:48 +08:00
av_free_format_context ( mFormatContext ) ;
2015-02-18 20:25:31 +08:00
# endif
2016-04-04 22:11:48 +08:00
mFormatContext = NULL ;
}
if ( mSessDesc )
{
delete mSessDesc ;
mSessDesc = NULL ;
}
delete mAuthenticator ;
2013-03-17 07:45:21 +08:00
}
int RtspThread : : run ( )
{
2016-04-04 22:11:48 +08:00
std : : string message ;
std : : string response ;
response . reserve ( ZM_NETWORK_BUFSIZ ) ;
2016-05-06 20:32:21 +08:00
if ( ! mRtspSocket . connect ( mHost . c_str ( ) , mPort . c_str ( ) ) )
2016-04-04 22:11:48 +08:00
Fatal ( " Unable to connect RTSP socket " ) ;
//Select select( 0.25 );
//select.addReader( &mRtspSocket );
//while ( select.wait() )
//{
//mRtspSocket.recv( response );
//Debug( 4, "Drained %d bytes from RTSP socket", response.size() );
//}
bool authTried = false ;
if ( mMethod = = RTP_RTSP_HTTP )
{
2016-05-06 20:32:21 +08:00
if ( ! mRtspSocket2 . connect ( mHost . c_str ( ) , mPort . c_str ( ) ) )
2016-04-04 22:11:48 +08:00
Fatal ( " Unable to connect auxiliary RTSP/HTTP socket " ) ;
2013-03-17 07:45:21 +08:00
//Select select( 0.25 );
2016-04-04 22:11:48 +08:00
//select.addReader( &mRtspSocket2 );
2013-03-17 07:45:21 +08:00
//while ( select.wait() )
//{
2016-04-04 22:11:48 +08:00
//mRtspSocket2.recv( response );
//Debug( 4, "Drained %d bytes from HTTP socket", response.size() );
2013-03-17 07:45:21 +08:00
//}
2014-04-29 05:37:31 +08:00
2016-04-04 22:11:48 +08:00
//possibly retry sending the message for authentication
int respCode = - 1 ;
char respText [ 256 ] ;
do {
message = " GET " + mPath + " HTTP/1.0 \r \n " ;
message + = " X-SessionCookie: " + mHttpSession + " \r \n " ;
if ( mNeedAuth ) {
message + = mAuthenticator - > getAuthHeader ( " GET " , mPath ) ;
authTried = true ;
}
message + = " Accept: application/x-rtsp-tunnelled \r \n " ;
message + = " \r \n " ;
Debug ( 2 , " Sending HTTP message: %s " , message . c_str ( ) ) ;
if ( mRtspSocket . send ( message . c_str ( ) , message . size ( ) ) ! = ( int ) message . length ( ) )
{
Error ( " Unable to send message '%s': %s " , message . c_str ( ) , strerror ( errno ) ) ;
return ( - 1 ) ;
}
if ( mRtspSocket . recv ( response ) < 0 )
{
Error ( " Recv failed; %s " , strerror ( errno ) ) ;
return ( - 1 ) ;
}
Debug ( 2 , " Received HTTP response: %s (%zd bytes) " , response . c_str ( ) , response . size ( ) ) ;
float respVer = 0 ;
respCode = - 1 ;
if ( sscanf ( response . c_str ( ) , " HTTP/%f %3d %[^ \r \n ] \r \n " , & respVer , & respCode , respText ) ! = 3 )
{
if ( isalnum ( response [ 0 ] ) )
2013-03-17 07:45:21 +08:00
{
2016-05-06 20:32:21 +08:00
Error ( " Response parse failure in '%s' " , response . c_str ( ) ) ;
2013-03-17 07:45:21 +08:00
}
2016-04-04 22:11:48 +08:00
else
2013-03-17 07:45:21 +08:00
{
2016-05-06 20:32:21 +08:00
Error ( " Response parse failure, %zd bytes follow " , response . size ( ) ) ;
if ( response . size ( ) )
Hexdump ( Logger : : ERROR , response . data ( ) , min ( response . size ( ) , 16 ) ) ;
2013-03-17 07:45:21 +08:00
}
return ( - 1 ) ;
2016-04-04 22:11:48 +08:00
}
// If Server requests authentication, check WWW-Authenticate header and fill required fields
// for requested authentication method
if ( respCode = = 401 & & ! authTried ) {
mNeedAuth = true ;
mAuthenticator - > checkAuthResponse ( response ) ;
Debug ( 2 , " Processed 401 response " ) ;
mRtspSocket . close ( ) ;
2016-05-06 20:32:21 +08:00
if ( ! mRtspSocket . connect ( mHost . c_str ( ) , mPort . c_str ( ) ) )
2016-04-04 22:11:48 +08:00
Fatal ( " Unable to reconnect RTSP socket " ) ;
Debug ( 2 , " connection should be reopened now " ) ;
}
} while ( respCode = = 401 & & ! authTried ) ;
if ( respCode ! = 200 )
{
Error ( " Unexpected response code %d, text is '%s' " , respCode , respText ) ;
return ( - 1 ) ;
}
2015-01-31 22:31:27 +08:00
2016-04-04 22:11:48 +08:00
message = " POST " + mPath + " HTTP/1.0 \r \n " ;
message + = " X-SessionCookie: " + mHttpSession + " \r \n " ;
if ( mNeedAuth )
message + = mAuthenticator - > getAuthHeader ( " POST " , mPath ) ;
message + = " Content-Length: 32767 \r \n " ;
message + = " Content-Type: application/x-rtsp-tunnelled \r \n " ;
message + = " \r \n " ;
Debug ( 2 , " Sending HTTP message: %s " , message . c_str ( ) ) ;
if ( mRtspSocket2 . send ( message . c_str ( ) , message . size ( ) ) ! = ( int ) message . length ( ) )
2015-08-21 23:29:54 +08:00
{
2016-04-04 22:11:48 +08:00
Error ( " Unable to send message '%s': %s " , message . c_str ( ) , strerror ( errno ) ) ;
return ( - 1 ) ;
2015-08-21 23:29:54 +08:00
}
2016-04-04 22:11:48 +08:00
}
2015-01-31 22:31:27 +08:00
2016-04-04 22:11:48 +08:00
std : : string localHost = " " ;
int localPorts [ 2 ] = { 0 , 0 } ;
2013-03-17 07:45:21 +08:00
2016-04-04 22:11:48 +08:00
// Request supported RTSP commands by the server
message = " OPTIONS " + mUrl + " RTSP/1.0 \r \n " ;
if ( ! sendCommand ( message ) )
return ( - 1 ) ;
2013-03-17 07:45:21 +08:00
2016-04-04 22:11:48 +08:00
// A negative return here may indicate auth failure, but we will have setup the auth mechanisms so we need to retry.
if ( ! recvResponse ( response ) ) {
if ( mNeedAuth ) {
Debug ( 2 , " Resending OPTIONS due to possible auth requirement " ) ;
if ( ! sendCommand ( message ) )
return ( - 1 ) ;
if ( ! recvResponse ( response ) )
2013-03-17 07:45:21 +08:00
return ( - 1 ) ;
2016-04-04 22:11:48 +08:00
} else {
return ( - 1 ) ;
2013-03-17 07:45:21 +08:00
}
2016-04-04 22:11:48 +08:00
} // end if failed response maybe due to auth
char publicLine [ 256 ] = " " ;
StringVector lines = split ( response , " \r \n " ) ;
for ( size_t i = 0 ; i < lines . size ( ) ; i + + )
sscanf ( lines [ i ] . c_str ( ) , " Public: %[^ \r \n ] \r \n " , publicLine ) ;
// Check if the server supports the GET_PARAMETER command
// If yes, it is likely that the server will request this command as a keepalive message
bool sendKeepalive = false ;
if ( publicLine [ 0 ] & & strstr ( publicLine , " GET_PARAMETER " ) )
sendKeepalive = true ;
message = " DESCRIBE " + mUrl + " RTSP/1.0 \r \n " ;
bool res ;
do {
if ( mNeedAuth )
authTried = true ;
sendCommand ( message ) ;
sleep ( 1 ) ;
res = recvResponse ( response ) ;
if ( ! res & & respCode = = 401 )
mNeedAuth = true ;
} while ( ! res & & respCode = = 401 & & ! authTried ) ;
const std : : string endOfHeaders = " \r \n \r \n " ;
size_t sdpStart = response . find ( endOfHeaders ) ;
if ( sdpStart = = std : : string : : npos )
return ( - 1 ) ;
2013-03-17 07:45:21 +08:00
2016-04-04 22:11:48 +08:00
if ( mRtspDescribe )
{
std : : string DescHeader = response . substr ( 0 , sdpStart ) ;
Debug ( 1 , " Processing DESCRIBE response header '%s' " , DescHeader . c_str ( ) ) ;
2013-03-17 07:45:21 +08:00
2016-04-04 22:11:48 +08:00
lines = split ( DescHeader , " \r \n " ) ;
for ( size_t i = 0 ; i < lines . size ( ) ; i + + )
{
// If the device sends us a url value for Content-Base in the response header, we should use that instead
if ( ( lines [ i ] . size ( ) > 13 ) & & ( lines [ i ] . substr ( 0 , 13 ) = = " Content-Base: " ) )
{
mUrl = trimSpaces ( lines [ i ] . substr ( 13 ) ) ;
Info ( " Received new Content-Base in DESCRIBE response header. Updated device Url to: '%s' " , mUrl . c_str ( ) ) ;
break ;
}
}
}
sdpStart + = endOfHeaders . length ( ) ;
std : : string sdp = response . substr ( sdpStart ) ;
Debug ( 1 , " Processing SDP '%s' " , sdp . c_str ( ) ) ;
try
{
mSessDesc = new SessionDescriptor ( mUrl , sdp ) ;
mFormatContext = mSessDesc - > generateFormatContext ( ) ;
}
catch ( const Exception & e )
{
Error ( e . getMessage ( ) . c_str ( ) ) ;
return ( - 1 ) ;
}
#if 0
// New method using ffmpeg native functions
std : : string authUrl = mUrl ;
if ( ! mAuth . empty ( ) )
authUrl . insert ( authUrl . find ( " :// " ) + 3 , mAuth + " @ " ) ;
if ( av_open_input_file ( & mFormatContext , authUrl . c_str ( ) , NULL , 0 , NULL ) ! = 0 )
{
Error ( " Unable to open input '%s' " , authUrl . c_str ( ) ) ;
return ( - 1 ) ;
}
2013-03-17 07:45:21 +08:00
# endif
2016-04-04 22:11:48 +08:00
uint32_t rtpClock = 0 ;
std : : string trackUrl = mUrl ;
std : : string controlUrl ;
_AVCODECID codecId ;
if ( mFormatContext - > nb_streams > = 1 )
{
for ( unsigned int i = 0 ; i < mFormatContext - > nb_streams ; i + + )
2013-03-17 07:45:21 +08:00
{
2016-04-04 22:11:48 +08:00
SessionDescriptor : : MediaDescriptor * mediaDesc = mSessDesc - > getStream ( i ) ;
2015-05-29 23:38:02 +08:00
# if (LIBAVCODEC_VERSION_CHECK(52, 64, 0, 64, 0) || LIBAVUTIL_VERSION_CHECK(50, 14, 0, 14, 0))
2016-04-04 22:11:48 +08:00
if ( mFormatContext - > streams [ i ] - > codec - > codec_type = = AVMEDIA_TYPE_VIDEO )
2013-03-17 07:45:21 +08:00
# else
2016-04-04 22:11:48 +08:00
if ( mFormatContext - > streams [ i ] - > codec - > codec_type = = CODEC_TYPE_VIDEO )
2013-03-17 07:45:21 +08:00
# endif
2016-04-04 22:11:48 +08:00
{
// Check if control Url is absolute or relative
controlUrl = mediaDesc - > getControlUrl ( ) ;
if ( std : : equal ( trackUrl . begin ( ) , trackUrl . end ( ) , controlUrl . begin ( ) ) )
2013-03-17 07:45:21 +08:00
{
2016-04-04 22:11:48 +08:00
trackUrl = controlUrl ;
2013-03-17 07:45:21 +08:00
}
2016-04-04 22:11:48 +08:00
else
2013-03-17 07:45:21 +08:00
{
2016-04-04 22:11:48 +08:00
if ( * trackUrl . rbegin ( ) ! = ' / ' ) {
trackUrl + = " / " + controlUrl ;
} else {
trackUrl + = controlUrl ;
}
2013-03-17 07:45:21 +08:00
}
2016-04-04 22:11:48 +08:00
rtpClock = mediaDesc - > getClock ( ) ;
codecId = mFormatContext - > streams [ i ] - > codec - > codec_id ;
// Hackery pokery
//rtpClock = mFormatContext->streams[i]->codec->sample_rate;
break ;
}
2013-03-17 07:45:21 +08:00
}
2016-04-04 22:11:48 +08:00
}
2013-03-17 07:45:21 +08:00
2016-04-04 22:11:48 +08:00
switch ( mMethod )
{
case RTP_UNICAST :
{
localPorts [ 0 ] = requestPorts ( ) ;
localPorts [ 1 ] = localPorts [ 0 ] + 1 ;
2013-03-17 07:45:21 +08:00
2016-04-04 22:11:48 +08:00
message = " SETUP " + trackUrl + " RTSP/1.0 \r \n Transport: RTP/AVP;unicast;client_port= " + stringtf ( " %d " , localPorts [ 0 ] ) + " - " + stringtf ( " %d " , localPorts [ 1 ] ) + " \r \n " ;
break ;
}
case RTP_MULTICAST :
2013-03-17 07:45:21 +08:00
{
2016-04-04 22:11:48 +08:00
message = " SETUP " + trackUrl + " RTSP/1.0 \r \n Transport: RTP/AVP;multicast \r \n " ;
break ;
2013-03-17 07:45:21 +08:00
}
2016-04-04 22:11:48 +08:00
case RTP_RTSP :
case RTP_RTSP_HTTP :
2013-03-17 07:45:21 +08:00
{
2016-04-04 22:11:48 +08:00
message = " SETUP " + trackUrl + " RTSP/1.0 \r \n Transport: RTP/AVP/TCP;unicast \r \n " ;
break ;
2013-03-17 07:45:21 +08:00
}
2016-04-04 22:11:48 +08:00
default :
{
Panic ( " Got unexpected method %d " , mMethod ) ;
break ;
}
}
2013-03-17 07:45:21 +08:00
2016-04-04 22:11:48 +08:00
if ( ! sendCommand ( message ) )
return ( - 1 ) ;
if ( ! recvResponse ( response ) )
return ( - 1 ) ;
2013-03-17 07:45:21 +08:00
2016-04-04 22:11:48 +08:00
lines = split ( response , " \r \n " ) ;
std : : string session ;
int timeout = 0 ;
char transport [ 256 ] = " " ;
2013-03-17 07:45:21 +08:00
2016-04-04 22:11:48 +08:00
for ( size_t i = 0 ; i < lines . size ( ) ; i + + )
{
if ( ( lines [ i ] . size ( ) > 8 ) & & ( lines [ i ] . substr ( 0 , 8 ) = = " Session: " ) )
2013-03-17 07:45:21 +08:00
{
2016-04-04 22:11:48 +08:00
StringVector sessionLine = split ( lines [ i ] . substr ( 9 ) , " ; " ) ;
session = trimSpaces ( sessionLine [ 0 ] ) ;
if ( sessionLine . size ( ) = = 2 )
sscanf ( trimSpaces ( sessionLine [ 1 ] ) . c_str ( ) , " timeout=%d " , & timeout ) ;
2013-03-17 07:45:21 +08:00
}
2016-04-04 22:11:48 +08:00
sscanf ( lines [ i ] . c_str ( ) , " Transport: %s " , transport ) ;
}
if ( session . empty ( ) )
Fatal ( " Unable to get session identifier from response '%s' " , response . c_str ( ) ) ;
Debug ( 2 , " Got RTSP session %s, timeout %d secs " , session . c_str ( ) , timeout ) ;
if ( ! transport [ 0 ] )
Fatal ( " Unable to get transport details from response '%s' " , response . c_str ( ) ) ;
Debug ( 2 , " Got RTSP transport %s " , transport ) ;
std : : string method = " " ;
int remotePorts [ 2 ] = { 0 , 0 } ;
int remoteChannels [ 2 ] = { 0 , 0 } ;
std : : string distribution = " " ;
unsigned long ssrc = 0 ;
StringVector parts = split ( transport , " ; " ) ;
for ( size_t i = 0 ; i < parts . size ( ) ; i + + )
{
if ( parts [ i ] = = " unicast " | | parts [ i ] = = " multicast " )
distribution = parts [ i ] ;
else if ( startsWith ( parts [ i ] , " server_port= " ) )
{
method = " RTP/UNICAST " ;
StringVector subparts = split ( parts [ i ] , " = " ) ;
StringVector ports = split ( subparts [ 1 ] , " - " ) ;
remotePorts [ 0 ] = strtol ( ports [ 0 ] . c_str ( ) , NULL , 10 ) ;
remotePorts [ 1 ] = strtol ( ports [ 1 ] . c_str ( ) , NULL , 10 ) ;
}
else if ( startsWith ( parts [ i ] , " interleaved= " ) )
{
method = " RTP/RTSP " ;
StringVector subparts = split ( parts [ i ] , " = " ) ;
StringVector channels = split ( subparts [ 1 ] , " - " ) ;
remoteChannels [ 0 ] = strtol ( channels [ 0 ] . c_str ( ) , NULL , 10 ) ;
remoteChannels [ 1 ] = strtol ( channels [ 1 ] . c_str ( ) , NULL , 10 ) ;
}
else if ( startsWith ( parts [ i ] , " port= " ) )
{
method = " RTP/MULTICAST " ;
StringVector subparts = split ( parts [ i ] , " = " ) ;
StringVector ports = split ( subparts [ 1 ] , " - " ) ;
localPorts [ 0 ] = strtol ( ports [ 0 ] . c_str ( ) , NULL , 10 ) ;
localPorts [ 1 ] = strtol ( ports [ 1 ] . c_str ( ) , NULL , 10 ) ;
}
else if ( startsWith ( parts [ i ] , " destination= " ) )
{
StringVector subparts = split ( parts [ i ] , " = " ) ;
localHost = subparts [ 1 ] ;
}
else if ( startsWith ( parts [ i ] , " ssrc= " ) )
{
StringVector subparts = split ( parts [ i ] , " = " ) ;
ssrc = strtoll ( subparts [ 1 ] . c_str ( ) , NULL , 16 ) ;
}
}
Debug ( 2 , " RTSP Method is %s " , method . c_str ( ) ) ;
Debug ( 2 , " RTSP Distribution is %s " , distribution . c_str ( ) ) ;
Debug ( 2 , " RTSP SSRC is %lx " , ssrc ) ;
Debug ( 2 , " RTSP Local Host is %s " , localHost . c_str ( ) ) ;
Debug ( 2 , " RTSP Local Ports are %d/%d " , localPorts [ 0 ] , localPorts [ 1 ] ) ;
Debug ( 2 , " RTSP Remote Ports are %d/%d " , remotePorts [ 0 ] , remotePorts [ 1 ] ) ;
Debug ( 2 , " RTSP Remote Channels are %d/%d " , remoteChannels [ 0 ] , remoteChannels [ 1 ] ) ;
message = " PLAY " + mUrl + " RTSP/1.0 \r \n Session: " + session + " \r \n Range: npt=0.000- \r \n " ;
if ( ! sendCommand ( message ) )
return ( - 1 ) ;
if ( ! recvResponse ( response ) )
return ( - 1 ) ;
2013-03-17 07:45:21 +08:00
2016-04-04 22:11:48 +08:00
lines = split ( response , " \r \n " ) ;
std : : string rtpInfo ;
for ( size_t i = 0 ; i < lines . size ( ) ; i + + )
{
if ( ( lines [ i ] . size ( ) > 9 ) & & ( lines [ i ] . substr ( 0 , 9 ) = = " RTP-Info: " ) )
rtpInfo = trimSpaces ( lines [ i ] . substr ( 9 ) ) ;
// Check for a timeout again. Some rtsp devices don't send a timeout until after the PLAY command is sent
if ( ( lines [ i ] . size ( ) > 8 ) & & ( lines [ i ] . substr ( 0 , 8 ) = = " Session: " ) & & ( timeout = = 0 ) )
2013-03-17 07:45:21 +08:00
{
2016-04-04 22:11:48 +08:00
StringVector sessionLine = split ( lines [ i ] . substr ( 9 ) , " ; " ) ;
if ( sessionLine . size ( ) = = 2 )
sscanf ( trimSpaces ( sessionLine [ 1 ] ) . c_str ( ) , " timeout=%d " , & timeout ) ;
if ( timeout > 0 )
Debug ( 2 , " Got timeout %d secs from PLAY command response " , timeout ) ;
2014-12-02 07:15:08 +08:00
}
2016-04-04 22:11:48 +08:00
}
int seq = 0 ;
unsigned long rtpTime = 0 ;
StringVector streams ;
if ( rtpInfo . empty ( ) )
{
Debug ( 1 , " RTP Info Empty. Starting values for Sequence and Rtptime shall be zero. " ) ;
}
else
{
Debug ( 2 , " Got RTP Info %s " , rtpInfo . c_str ( ) ) ;
// More than one stream can be included in the RTP Info
streams = split ( rtpInfo . c_str ( ) , " , " ) ;
for ( size_t i = 0 ; i < streams . size ( ) ; i + + )
2014-12-02 07:15:08 +08:00
{
2016-04-04 22:11:48 +08:00
// We want the stream that matches the trackUrl we are using
if ( streams [ i ] . find ( controlUrl . c_str ( ) ) ! = std : : string : : npos )
{
// Parse the sequence and rtptime values
parts = split ( streams [ i ] . c_str ( ) , " ; " ) ;
for ( size_t j = 0 ; j < parts . size ( ) ; j + + )
2013-03-17 07:45:21 +08:00
{
2016-04-04 22:11:48 +08:00
if ( startsWith ( parts [ j ] , " seq= " ) )
{
StringVector subparts = split ( parts [ j ] , " = " ) ;
seq = strtol ( subparts [ 1 ] . c_str ( ) , NULL , 10 ) ;
}
else if ( startsWith ( parts [ j ] , " rtptime= " ) )
{
StringVector subparts = split ( parts [ j ] , " = " ) ;
rtpTime = strtol ( subparts [ 1 ] . c_str ( ) , NULL , 10 ) ;
}
2013-03-17 07:45:21 +08:00
}
2016-04-04 22:11:48 +08:00
break ;
}
2013-03-17 07:45:21 +08:00
}
2016-04-04 22:11:48 +08:00
}
2013-03-17 07:45:21 +08:00
2016-04-04 22:11:48 +08:00
Debug ( 2 , " RTSP Seq is %d " , seq ) ;
Debug ( 2 , " RTSP Rtptime is %ld " , rtpTime ) ;
2013-03-17 07:45:21 +08:00
2016-04-04 22:11:48 +08:00
time_t lastKeepalive = time ( NULL ) ;
time_t now ;
message = " GET_PARAMETER " + mUrl + " RTSP/1.0 \r \n Session: " + session + " \r \n " ;
2014-11-08 00:39:50 +08:00
2016-04-04 22:11:48 +08:00
switch ( mMethod )
{
case RTP_UNICAST :
2013-03-17 07:45:21 +08:00
{
2016-04-04 22:11:48 +08:00
RtpSource * source = new RtpSource ( mId , " " , localPorts [ 0 ] , mHost , remotePorts [ 0 ] , ssrc , seq , rtpClock , rtpTime , codecId ) ;
mSources [ ssrc ] = source ;
RtpDataThread rtpDataThread ( * this , * source ) ;
RtpCtrlThread rtpCtrlThread ( * this , * source ) ;
rtpDataThread . start ( ) ;
rtpCtrlThread . start ( ) ;
while ( ! mStop )
{
now = time ( NULL ) ;
// Send a keepalive message if the server supports this feature and we are close to the timeout expiration
2015-06-22 21:26:17 +08:00
Debug ( 5 , " sendkeepalive %d, timeout %d, now: %d last: %d since: %d " , sendKeepalive , timeout , now , lastKeepalive , ( now - lastKeepalive ) ) ;
2016-04-04 22:11:48 +08:00
if ( sendKeepalive & & ( timeout > 0 ) & & ( ( now - lastKeepalive ) > ( timeout - 5 ) ) )
{
if ( ! sendCommand ( message ) )
return ( - 1 ) ;
lastKeepalive = now ;
}
usleep ( 100000 ) ;
}
2013-03-17 07:45:21 +08:00
#if 0
2016-04-04 22:11:48 +08:00
message = " PAUSE " + mUrl + " RTSP/1.0 \r \n Session: " + session + " \r \n " ;
if ( ! sendCommand ( message ) )
return ( - 1 ) ;
if ( ! recvResponse ( response ) )
return ( - 1 ) ;
2013-03-17 07:45:21 +08:00
# endif
2016-04-04 22:11:48 +08:00
message = " TEARDOWN " + mUrl + " RTSP/1.0 \r \n Session: " + session + " \r \n " ;
if ( ! sendCommand ( message ) )
return ( - 1 ) ;
if ( ! recvResponse ( response ) )
return ( - 1 ) ;
2013-03-17 07:45:21 +08:00
2016-04-04 22:11:48 +08:00
rtpDataThread . stop ( ) ;
rtpCtrlThread . stop ( ) ;
2013-03-17 07:45:21 +08:00
2016-04-04 22:11:48 +08:00
//rtpDataThread.kill( SIGTERM );
//rtpCtrlThread.kill( SIGTERM );
2013-03-17 07:45:21 +08:00
2016-04-04 22:11:48 +08:00
rtpDataThread . join ( ) ;
rtpCtrlThread . join ( ) ;
delete mSources [ ssrc ] ;
mSources . clear ( ) ;
2013-03-17 07:45:21 +08:00
2016-04-04 22:11:48 +08:00
releasePorts ( localPorts [ 0 ] ) ;
2013-03-17 07:45:21 +08:00
2016-04-04 22:11:48 +08:00
break ;
}
case RTP_RTSP :
case RTP_RTSP_HTTP :
{
RtpSource * source = new RtpSource ( mId , " " , remoteChannels [ 0 ] , mHost , remoteChannels [ 0 ] , ssrc , seq , rtpClock , rtpTime , codecId ) ;
mSources [ ssrc ] = source ;
// These never actually run
RtpDataThread rtpDataThread ( * this , * source ) ;
RtpCtrlThread rtpCtrlThread ( * this , * source ) ;
Select select ( double ( config . http_timeout ) / 1000.0 ) ;
select . addReader ( & mRtspSocket ) ;
Buffer buffer ( ZM_NETWORK_BUFSIZ ) ;
std : : string keepaliveMessage = " OPTIONS " + mUrl + " RTSP/1.0 \r \n " ;
std : : string keepaliveResponse = " RTSP/1.0 200 OK \r \n " ;
while ( ! mStop & & select . wait ( ) > = 0 )
{
Select : : CommsList readable = select . getReadable ( ) ;
if ( readable . size ( ) = = 0 )
2013-03-17 07:45:21 +08:00
{
2016-04-04 22:11:48 +08:00
Error ( " RTSP timed out " ) ;
break ;
}
2013-03-17 07:45:21 +08:00
2016-04-04 22:11:48 +08:00
static char tempBuffer [ ZM_NETWORK_BUFSIZ ] ;
ssize_t nBytes = mRtspSocket . recv ( tempBuffer , sizeof ( tempBuffer ) ) ;
buffer . append ( tempBuffer , nBytes ) ;
Debug ( 4 , " Read %zd bytes on sd %d, %d total " , nBytes , mRtspSocket . getReadDesc ( ) , buffer . size ( ) ) ;
2013-03-17 07:45:21 +08:00
2016-04-04 22:11:48 +08:00
while ( buffer . size ( ) > 0 )
2013-03-17 07:45:21 +08:00
{
2016-04-04 22:11:48 +08:00
if ( buffer [ 0 ] = = ' $ ' )
{
if ( buffer . size ( ) < 4 )
break ;
unsigned char channel = buffer [ 1 ] ;
unsigned short len = ntohs ( * ( ( unsigned short * ) ( buffer + 2 ) ) ) ;
2013-03-17 07:45:21 +08:00
2016-04-04 22:11:48 +08:00
Debug ( 4 , " Got %d bytes left, expecting %d byte packet on channel %d " , buffer . size ( ) , len , channel ) ;
if ( ( unsigned short ) buffer . size ( ) < ( len + 4 ) )
{
Debug ( 4 , " Missing %d bytes, rereading " , ( len + 4 ) - buffer . size ( ) ) ;
break ;
}
if ( channel = = remoteChannels [ 0 ] )
2013-03-17 07:45:21 +08:00
{
2016-04-04 22:11:48 +08:00
Debug ( 4 , " Got %d bytes on data channel %d, packet length is %d " , buffer . size ( ) , channel , len ) ;
Hexdump ( 4 , ( char * ) buffer , 16 ) ;
rtpDataThread . recvPacket ( buffer + 4 , len ) ;
Debug ( 4 , " Received " ) ;
2013-03-17 07:45:21 +08:00
}
2016-04-04 22:11:48 +08:00
else if ( channel = = remoteChannels [ 1 ] )
{
// len = ntohs( *((unsigned short *)(buffer+2)) );
// Debug( 4, "Got %d bytes on control channel %d", nBytes, channel );
Debug ( 4 , " Got %d bytes on control channel %d, packet length is %d " , buffer . size ( ) , channel , len ) ;
Hexdump ( 4 , ( char * ) buffer , 16 ) ;
rtpCtrlThread . recvPackets ( buffer + 4 , len ) ;
}
else
{
Error ( " Unexpected channel selector %d in RTSP interleaved data " , buffer [ 1 ] ) ;
buffer . clear ( ) ;
break ;
}
buffer . consume ( len + 4 ) ;
nBytes - = len + 4 ;
}
else
{
if ( keepaliveResponse . compare ( 0 , keepaliveResponse . size ( ) , ( char * ) buffer , keepaliveResponse . size ( ) ) = = 0 )
{
Debug ( 4 , " Got keepalive response '%s' " , ( char * ) buffer ) ;
//buffer.consume( keepaliveResponse.size() );
if ( char * charPtr = ( char * ) memchr ( ( char * ) buffer , ' $ ' , buffer . size ( ) ) )
{
int discardBytes = charPtr - ( char * ) buffer ;
buffer - = discardBytes ;
}
else
{
buffer . clear ( ) ;
}
}
else
{
if ( char * charPtr = ( char * ) memchr ( ( char * ) buffer , ' $ ' , buffer . size ( ) ) )
{
int discardBytes = charPtr - ( char * ) buffer ;
Warning ( " Unexpected format RTSP interleaved data, resyncing by %d bytes " , discardBytes ) ;
Hexdump ( - 1 , ( char * ) buffer , discardBytes ) ;
buffer - = discardBytes ;
}
else
{
Warning ( " Unexpected format RTSP interleaved data, dumping %d bytes " , buffer . size ( ) ) ;
Hexdump ( - 1 , ( char * ) buffer , 32 ) ;
buffer . clear ( ) ;
}
}
}
}
// Send a keepalive message if the server supports this feature and we are close to the timeout expiration
// FIXME: Is this really necessary when using tcp ?
now = time ( NULL ) ;
// Send a keepalive message if the server supports this feature and we are close to the timeout expiration
Debug ( 5 , " sendkeepalive %d, timeout %d, now: %d last: %d since: %d " , sendKeepalive , timeout , now , lastKeepalive , ( now - lastKeepalive ) ) ;
if ( sendKeepalive & & ( timeout > 0 ) & & ( ( now - lastKeepalive ) > ( timeout - 5 ) ) )
{
if ( ! sendCommand ( message ) )
return ( - 1 ) ;
lastKeepalive = now ;
}
buffer . tidy ( 1 ) ;
}
2013-03-17 07:45:21 +08:00
#if 0
2016-04-04 22:11:48 +08:00
message = " PAUSE " + mUrl + " RTSP/1.0 \r \n Session: " + session + " \r \n " ;
if ( ! sendCommand ( message ) )
return ( - 1 ) ;
if ( ! recvResponse ( response ) )
return ( - 1 ) ;
2013-03-17 07:45:21 +08:00
# endif
2016-04-04 22:11:48 +08:00
// Send a teardown message but don't expect a response as this may not be implemented on the server when using TCP
message = " TEARDOWN " + mUrl + " RTSP/1.0 \r \n Session: " + session + " \r \n " ;
if ( ! sendCommand ( message ) )
return ( - 1 ) ;
delete mSources [ ssrc ] ;
mSources . clear ( ) ;
break ;
}
case RTP_MULTICAST :
{
RtpSource * source = new RtpSource ( mId , localHost , localPorts [ 0 ] , mHost , remotePorts [ 0 ] , ssrc , seq , rtpClock , rtpTime , codecId ) ;
mSources [ ssrc ] = source ;
RtpDataThread rtpDataThread ( * this , * source ) ;
RtpCtrlThread rtpCtrlThread ( * this , * source ) ;
rtpDataThread . start ( ) ;
rtpCtrlThread . start ( ) ;
while ( ! mStop )
{
// Send a keepalive message if the server supports this feature and we are close to the timeout expiration
if ( sendKeepalive & & ( timeout > 0 ) & & ( ( time ( NULL ) - lastKeepalive ) > ( timeout - 5 ) ) )
2013-03-17 07:45:21 +08:00
{
2016-04-04 22:11:48 +08:00
if ( ! sendCommand ( message ) )
return ( - 1 ) ;
lastKeepalive = time ( NULL ) ;
2013-03-17 07:45:21 +08:00
}
2016-04-04 22:11:48 +08:00
usleep ( 100000 ) ;
}
#if 0
message = " PAUSE " + mUrl + " RTSP/1.0 \r \n Session: " + session + " \r \n " ;
if ( ! sendCommand ( message ) )
return ( - 1 ) ;
if ( ! recvResponse ( response ) )
return ( - 1 ) ;
# endif
message = " TEARDOWN " + mUrl + " RTSP/1.0 \r \n Session: " + session + " \r \n " ;
if ( ! sendCommand ( message ) )
return ( - 1 ) ;
if ( ! recvResponse ( response ) )
return ( - 1 ) ;
rtpDataThread . stop ( ) ;
rtpCtrlThread . stop ( ) ;
rtpDataThread . join ( ) ;
rtpCtrlThread . join ( ) ;
delete mSources [ ssrc ] ;
mSources . clear ( ) ;
releasePorts ( localPorts [ 0 ] ) ;
break ;
}
default :
{
Panic ( " Got unexpected method %d " , mMethod ) ;
break ;
2013-03-17 07:45:21 +08:00
}
2016-04-04 22:11:48 +08:00
}
2013-03-17 07:45:21 +08:00
2016-04-04 22:11:48 +08:00
return ( 0 ) ;
2013-03-17 07:45:21 +08:00
}
# endif // HAVE_LIBAVFORMAT