UDPSocketDataIO.h

/* This file is Copyright 2000-2008 Meyer Sound Laboratories Inc.  See the included LICENSE.txt file for details. */

#ifndef MuscleUDPSocketDataIO_h
#define MuscleUDPSocketDataIO_h

#include "support/MuscleSupport.h"

#include "dataio/DataIO.h"
#include "util/NetworkUtilityFunctions.h"

BEGIN_NAMESPACE(muscle);

class UDPSocketDataIO : public DataIO
{
public:
   UDPSocketDataIO(const SocketRef & sock, bool blocking) : _sock(sock)
   {
      (void) SetBlockingIOEnabled(blocking);
   }

   virtual ~UDPSocketDataIO() 
   {
      // empty
   }

   virtual int32 Read(void * buffer, uint32 size) 
   {
      ip_address tmpAddr = invalidIP;
      uint16 tmpPort = 0;
      int32 ret = ReceiveDataUDP(_sock, buffer, size, _blocking, &tmpAddr, &tmpPort);
      _recvFrom.SetIPAddress(tmpAddr);
      _recvFrom.SetPort(tmpPort);
      return ret;
   }

   virtual int32 Write(const void * buffer, uint32 size) {return SendDataUDP(_sock, buffer, size, _blocking, _sendTo.GetIPAddress(), _sendTo.GetPort());}

   virtual status_t Seek(int64 /*seekOffset*/, int /*whence*/) {return B_ERROR;}

   virtual int64 GetPosition() const {return -1;}

   virtual void FlushOutput() {/* empty */}
   
   virtual void Shutdown() {_sock.Reset();}

   virtual const SocketRef & GetSelectSocket() const {return _sock;}

   void SetSendDestination(const IPAddressAndPort & dest) {_sendTo = dest;}

   const IPAddressAndPort & GetSendDestination() const {return _sendTo;}

   status_t SetBlockingIOEnabled(bool blocking)
   {
      status_t ret = SetSocketBlockingEnabled(_sock, blocking);
      if (ret == B_NO_ERROR) _blocking = blocking;
      return ret;
   }

   bool IsBlockingIOEnabled() const {return _blocking;}

   const IPAddressAndPort & GetSourceOfLastReadPacket() const {return _recvFrom;}

private:
   SocketRef _sock;
   bool _blocking;

   IPAddressAndPort _recvFrom;
   IPAddressAndPort _sendTo;
};

END_NAMESPACE(muscle);

#endif

---