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;

class DT_DLL_RTIUTIL DtUdpMsgSocket : public DtMsgSocket
{
public:

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

   virtual ~DtUdpMsgSocket();

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

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

   virtual bool flush();

   virtual int recvMessage(caddr_t *buffptr);

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

   virtual void enableBundling(size_t maxBundleSize);

   virtual void disableBundling();

   virtual void enableCompression(MAKRti::DtU8 compressionLevel);

   virtual void disableCompression();

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

   virtual bool setMcastTtlOption( MAKRti::DtU32 ttl );

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

   virtual MAKRti::DtU32 maxMsgSize() const;

   virtual bool isOpen() const;

   MAKRti::DtInetUdpSocket* inetUdpSocket();

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

protected:

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

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

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

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

protected:

   MAKRti::DtInetEndpoint myEndpoint;

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

   DtMsgBundler myDefaultMsgBundler;

   bool myBundlingEnabled;

   DtMsgCompressor myMsgCompressor;

   bool myCompressionEnabled;

   MAKRti::DtInetEndpoint myLastSourceEndpoint;

   size_t myMaxPacketSize;

   unsigned int myMaxSendRetries;

   double mySendRetryPeriod;
   
   int myFilterPort;

   static const MAKRti::DtU32 theDefaultSockOpts;

};


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