MprFromNet.h

Go to the documentation of this file.

//  
// Copyright (C) 2006-2012 SIPez LLC.  All rights reserved.
//
// Copyright (C) 2004-2008 SIPfoundry Inc.
// Licensed by SIPfoundry under the LGPL license.
//
// Copyright (C) 2004-2006 Pingtel Corp.  All rights reserved.
// Licensed to SIPfoundry under a Contributor Agreement.
//
// $$


#ifndef _MprFromNet_h_
#define _MprFromNet_h_

#include "rtcp/RtcpConfig.h"

// SYSTEM INCLUDES
#ifdef _WIN32 /* [ */
#include <winsock2.h>
#endif /* _WIN32 ] */

// APPLICATION INCLUDES
#include <os/OsMsgQ.h>
#include "os/OsDefs.h"
#include "mp/MpUdpBuf.h"
#include "mp/MpRtpBuf.h"
#ifdef INCLUDE_RTCP /* [ */
#include "rtcp/IRTPDispatch.h"
#include "rtcp/INetDispatch.h"
#else
#include <mp/NetInTask.h>
#endif /* INCLUDE_RTCP ] */

// DEFINES
// MACROS
// EXTERNAL FUNCTIONS
// EXTERNAL VARIABLES
// CONSTANTS
// STRUCTS
// TYPEDEFS
// FORWARD DECLARATIONS
class MprDecode;
class MprDejitter;
class MprRtpDispatcher;
class OsEvent;
class OsSocket;
class MpResourceMsg;
class NetInTask;
class UtlString;
class MpFlowGraphBase;

class MprFromNet
{
/* //////////////////////////// PUBLIC //////////////////////////////////// */
public:

/* ============================ CREATORS ================================== */


   MprFromNet();

   virtual ~MprFromNet();


/* ============================ MANIPULATORS ============================== */

   UtlBoolean handleMessage(MpResourceMsg& rMsg);

   static OsStatus setSockets(const UtlString& resourceName, OsMsgQ& flowgraphMessageQueue, OsSocket* rtpSocket, OsSocket* rtcpSocket);

   OsStatus setSockets(OsSocket& rRtpSocket, OsSocket& rRtcpSocket);
   static OsStatus resetSockets(const UtlString& resourceName, OsMsgQ& flowgraphMessageQueue);

   OsStatus resetSockets();
   OsStatus pushPacket(const MpUdpBufPtr &buf, bool isRtcp);

   OsStatus enableSsrcDiscard(UtlBoolean enable, RtpSRC ssrc);
   OsStatus setRtpDispatcher(MprRtpDispatcher *pRtpDispatcher);

/* ============================ ACCESSORS ================================= */


#ifdef INCLUDE_RTCP /* [ */
   void setDispatchers(IRTPDispatch *piRTPDispatch, INetDispatch *piRTCPDispatch);

#else /* INCLUDE_RTCP ] [ */
   OsStatus getRtcpStats(MprRtcpStats& stats);
#endif /* INCLUDE_RTCP ] */

   // For debug purposes allow labeling this with the containing flowgraph
   OsStatus setFlowGraph(MpFlowGraphBase* flowgraph);


/* ============================ INQUIRY =================================== */



/* //////////////////////////// PROTECTED ///////////////////////////////// */
protected:

/* //////////////////////////// PRIVATE /////////////////////////////////// */
private:
   OsMutex          mDiscardCtlMutex;
   OsMutex          mRegistrationSyncMutex;
   NetInTask*       mNetInTask;
   UtlBoolean       mRegistered;
   MprRtpDispatcher* mpRtpDispatcher;
   UtlBoolean       mDiscardSelectedStream;
   RtpSRC           mDiscardedSSRC;
   MpFlowGraphBase* mpFlowGraph;
#ifdef INCLUDE_RTCP /* [ */
   INetDispatch*    mpiRTCPDispatch;
   IRTPDispatch*    mpiRTPDispatch;

#else /* INCLUDE_RTCP ] [ */
   rtpHandle        mInRtpHandle;
   int              mRtcpCount;
#endif /* INCLUDE_RTCP ] */

   int             mNumPushed;     
   int             mNumPktsRtcp;   
   int             mNumPktsRtp;    
   int             mNumEncDropped; 
   int             mNumLoopDropped;

#ifndef INCLUDE_RTCP /* [ */
   OsStatus rtcpStats(struct RtpHeader *h);
#endif /* INCLUDE_RTCP ] */

   UtlBoolean resetSocketsInternal(OsEvent *pEvent = NULL);
   static MpRtpBufPtr parseRtpPacket(const MpUdpBufPtr &buf);

   MprFromNet(const MprFromNet& rMprFromNet);

   MprFromNet& operator=(const MprFromNet& rhs);

};

/* ============================ INLINE METHODS ============================ */
#endif  // _MprFromNet_h_