zoneminder/src/zm_rtp_data.cpp

//
// ZoneMinder RTP Data Class Implementation, $Date$, $Revision$
// Copyright (C) 2003-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_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 %ld on port %d", mRtpSource.getSsrc(), mRtpSource.getLocalDataPort() );

    SockAddrInet localAddr;
    UdpInetServer rtpDataSocket;
    if ( mRtpSource.getLocalHost() != "" )
        localAddr.resolve( mRtpSource.getLocalHost().c_str(), mRtpSource.getLocalDataPort(), "udp" );
    else
        localAddr.resolve( mRtpSource.getLocalDataPort(), "udp" );
    if ( !rtpDataSocket.bind( localAddr ) )
        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[BUFSIZ];
    while ( !mStop && select.wait() >= 0 )
    {
         Select::CommsList readable = select.getReadable();
         if ( readable.size() == 0 )
         {
             Error( "RTP timed out" );
             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
             {
                 Fatal( "Barfed" );
             }
         }
    }
    rtpDataSocket.close();
    mRtspThread.stop();
    return( 0 );
}