125 lines
3.7 KiB
C++
125 lines
3.7 KiB
C++
//
|
|
// ZoneMinder RTP Data 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., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
|
|
//
|
|
|
|
#include "zm.h"
|
|
|
|
#if HAVE_LIBAVFORMAT
|
|
|
|
#include "zm_rtp_data.h"
|
|
|
|
#include "zm_rtsp.h"
|
|
|
|
#include <arpa/inet.h>
|
|
|
|
RtpDataThread::RtpDataThread( RtspThread &rtspThread, RtpSource &rtpSource ) : mRtspThread( rtspThread ), mRtpSource( rtpSource ), mStop( false )
|
|
{
|
|
}
|
|
|
|
bool RtpDataThread::recvPacket( const unsigned char *packet, size_t packetLen )
|
|
{
|
|
const RtpDataHeader *rtpHeader;
|
|
rtpHeader = (RtpDataHeader *)packet;
|
|
|
|
//printf( "D: " );
|
|
//for ( int i = 0; i < 32; i++ )
|
|
//printf( "%02x ", (unsigned char)packet[i] );
|
|
//printf( "\n" );
|
|
|
|
Debug( 5, "Ver: %d", rtpHeader->version );
|
|
Debug( 5, "P: %d", rtpHeader->p );
|
|
Debug( 5, "Pt: %d", rtpHeader->pt );
|
|
Debug( 5, "Mk: %d", rtpHeader->m );
|
|
Debug( 5, "Seq: %d", ntohs(rtpHeader->seqN) );
|
|
Debug( 5, "T/S: %x", ntohl(rtpHeader->timestampN) );
|
|
Debug( 5, "SSRC: %x", ntohl(rtpHeader->ssrcN) );
|
|
|
|
//unsigned short seq = ntohs(rtpHeader->seqN);
|
|
unsigned long ssrc = ntohl(rtpHeader->ssrcN);
|
|
|
|
if ( mRtpSource.getSsrc() && (ssrc != mRtpSource.getSsrc()) )
|
|
{
|
|
Warning( "Discarding packet for unrecognised ssrc %lx", ssrc );
|
|
return( false );
|
|
}
|
|
|
|
return( mRtpSource.handlePacket( packet, packetLen ) );
|
|
}
|
|
|
|
int RtpDataThread::run()
|
|
{
|
|
Debug( 2, "Starting data thread %d on port %d", mRtpSource.getSsrc(), mRtpSource.getLocalDataPort() );
|
|
|
|
SockAddrInet localAddr;
|
|
UdpInetServer rtpDataSocket;
|
|
if ( mRtpSource.getLocalHost() != "" ) {
|
|
if ( !rtpDataSocket.bind( mRtpSource.getLocalHost().c_str(), mRtpSource.getLocalDataPort() ) )
|
|
Fatal( "Failed to bind RTP server" );
|
|
Debug( 3, "Bound to %s:%d", mRtpSource.getLocalHost().c_str(), mRtpSource.getLocalDataPort() );
|
|
}
|
|
else
|
|
{
|
|
if ( !rtpDataSocket.bind( mRtspThread.getAddressFamily() == AF_INET6 ? "::" : "0.0.0.0", mRtpSource.getLocalDataPort() ) )
|
|
Fatal( "Failed to bind RTP server" );
|
|
Debug( 3, "Bound to %s:%d", mRtpSource.getLocalHost().c_str(), mRtpSource.getLocalDataPort() );
|
|
}
|
|
|
|
Select select( 3 );
|
|
select.addReader( &rtpDataSocket );
|
|
|
|
unsigned char buffer[ZM_NETWORK_BUFSIZ];
|
|
while ( !mStop && select.wait() >= 0 )
|
|
{
|
|
if ( mStop )
|
|
break;
|
|
Select::CommsList readable = select.getReadable();
|
|
if ( readable.size() == 0 )
|
|
{
|
|
Error( "RTP timed out" );
|
|
mStop = true;
|
|
break;
|
|
}
|
|
for ( Select::CommsList::iterator iter = readable.begin(); iter != readable.end(); ++iter )
|
|
{
|
|
if ( UdpInetServer *socket = dynamic_cast<UdpInetServer *>(*iter) )
|
|
{
|
|
int nBytes = socket->recv( buffer, sizeof(buffer) );
|
|
Debug( 4, "Got %d bytes on sd %d", nBytes, socket->getReadDesc() );
|
|
if ( nBytes )
|
|
{
|
|
recvPacket( buffer, nBytes );
|
|
}
|
|
else
|
|
{
|
|
mStop = true;
|
|
break;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
Panic( "Barfed" );
|
|
}
|
|
}
|
|
}
|
|
rtpDataSocket.close();
|
|
mRtspThread.stop();
|
|
return( 0 );
|
|
}
|
|
|
|
#endif // HAVE_LIBAVFORMAT
|