udpMsgSocket.h

Go to the documentation of this file.

/*********************************************************************
** Copyright (c) 2010 MaK Technologies, Inc.
** All rights reserved.
*********************************************************************/
/*********************************************************************
** $RCSfile: udpMsgSocket.h,v $ $Revision: $ $State: Exp $
*********************************************************************/


#pragma once

#include "msgSocket.h"
#include "msgBundler.h"
#include "msgCompressor.h"

#include <vlutil/vlInetUdpSocket.h>

const size_t DtDefaultMaxPacketSize = 15000;

using namespace MAKRti;

class DT_DLL_RTIUTIL DtUdpMsgSocket : public DtMsgSocket
{
public:

   DtUdpMsgSocket(DtBoost::shared_ptr<DtMsgHeaderReader> hdrReader,
      size_t maxPacketSize = DtDefaultMaxPacketSize);

   virtual ~DtUdpMsgSocket();

   virtual void openSocket(const DtInetEndpoint& dest,
      DtU32 flags = DtDefaultSockOpts, DtU16 localPort = 0,
      DtInetDevice* hostIf = NULL);

   virtual int sendMessage(caddr_t msg, size_t msgSize, bool deleteAfterSend,
      const DtInetAddr& destAddr = DtInetAddr::inaddrAny(), int port = 0);

   virtual bool flush();

   virtual int recvMessage(caddr_t *buffptr);

   virtual int recvMessageWithSrc(caddr_t *buffptr, DtInetEndpoint& src);

   virtual void enableBundling(size_t maxBundleSize);

   virtual void disableBundling();

   virtual void enableCompression(DtU8 compressionLevel);

   virtual void disableCompression();

   virtual bool joinMulticastGroup( const DtInetAddr& grpAddr, 
      DtInetDevice* device = NULL );
   virtual bool joinMulticastGroup( const DtInetEndpoint& grpEndpoint, 
      DtInetDevice* device = NULL );
   virtual bool dropMulticastGroup( const DtInetAddr& grpAddr, 
      DtInetDevice* device = NULL );
   virtual bool dropMulticastGroup( const DtInetEndpoint& grpEndpoint, 
      DtInetDevice* device = NULL );

   virtual bool setMcastTtlOption( DtU32 ttl );

   virtual void setRetryOnSend(unsigned int numRetries, double waitPeriod);

   virtual DtU32 maxMsgSize() const;

   virtual bool isOpen() const;

   DtInetUdpSocket* inetUdpSocket();

   virtual void setFilterPort(int port) { myFilterPort = port; }

protected:

   virtual int sendPacket(caddr_t msg, size_t msgSize,
      const DtInetAddr& destAddr = DtInetAddr::inaddrAny(),
      int port = 0);

   virtual int doSendPacket(caddr_t msg, size_t msgSize,
      const DtInetAddr& destAddr = DtInetAddr::inaddrAny(),
      int port = 0);

   virtual int retrySendTo(caddr_t msg, size_t msgSize,
      const DtInetAddr& destAddr = DtInetAddr::inaddrAny(),
      int port = 0);

   virtual int recv(caddr_t *buffptr, DtInetEndpoint* ep);

protected:

   DtInetEndpoint myEndpoint;

   std::map<DtInetAddr, DtMsgBundler*> myMsgBundlers;

   DtMsgBundler myDefaultMsgBundler;

   bool myBundlingEnabled;

   DtMsgCompressor myMsgCompressor;

   bool myCompressionEnabled;

   DtInetEndpoint myLastSourceEndpoint;

   size_t myMaxPacketSize;

   unsigned int myMaxSendRetries;

   double mySendRetryPeriod;
   
   int myFilterPort;
};


inline DtInetUdpSocket* DtUdpMsgSocket::inetUdpSocket()
{
   return static_cast<DtInetUdpSocket*>(mySocket.get());
}