123

CryNetwork/DatagramSocket.h

@@ -0,0 +1,125 @@
+//////////////////////////////////////////////////////////////////////
+//
+//	Crytek Network source code
+//	
+//	File: DatagramSocket.h
+//  Description: 
+//
+//	History:
+//	-July 25,2001:Created by Alberto Demichelis
+//
+//////////////////////////////////////////////////////////////////////
+
+#ifndef _DATAGRAM_SOCKET_H_
+#define _DATAGRAM_SOCKET_H_
+
+#if _MSC_VER > 1000
+#pragma once
+#endif // _MSC_VER > 1000
+
+#define MAX_UDP_PACKET_SIZE 8192
+// struct ip_mreq mreq;
+// mreq is the ip_mreqstructure
+//{
+//  struct in_addr imr_multiaddr;  // The multicast group to join 
+//  struct in_addr imr_interface;  // The interface to join on
+//}
+// multicast sample addr is --> "234.5.6.7"
+
+#include "Stream.h"
+#include <IPAddress.h>
+
+#ifndef _XBOX
+#if _MSC_VER > 1000
+#pragma comment(lib, "WSOCK32.LIB")
+#endif // _MSC_VER > 1000
+#endif
+
+#ifdef LINUX
+#include <time.h>
+#endif //LINUX
+
+/*inline int	gethostname(char *__name, size_t __len)
+{
+#pragma message ("gethostname not implemented")
+	return 0;
+}
+*/
+
+class CDatagramSocket
+{
+public:
+	enum SocketType
+	{
+		Blocking,
+		NonBlocking
+	};
+
+	//! constructor
+	CDatagramSocket()
+	{
+		m_hSocket=INVALID_SOCKET;
+
+		m_nSentBytesInThisSec=0;
+		m_nReceivedBytesInThisSec=0;
+		m_nSentPacketsInThisSec=0;
+		m_nReceivedPacketsInThisSec=0;
+
+		m_fIncomingKbPerSec=0.0f;
+		m_fOutgoingKbPerSec=0.0f;
+		m_nIncomingPacketsPerSec=0;
+		m_nOutgoingPacketsPerSec=0;
+	}
+	//! destructor
+	virtual ~CDatagramSocket()
+	{
+		Close(); 
+		//<<FIXME>>
+	}
+	//!
+	NRESULT Create(SocketType st = NonBlocking);
+	//!
+	NRESULT Listen(WORD wPort, CIPAddress *xaMulticastAddress = NULL, CIPAddress *ipLocalAddress = 0);
+	//!
+	void SetDefaultTarget(CIPAddress &saAddress)
+	{
+		m_saDefaultAddress.Set(saAddress);
+	}
+	//!
+	NRESULT Send(BYTE *pBuffer, int nLenBytes, CIPAddress *saAddress = NULL);
+	//!
+	NRESULT Receive(unsigned char *pBuf/*[MAX_UDP_PACKET_SIZE]*/, int nBufLen, int &nRecvBytes, CIPAddress &pFrom);
+
+	const char *GetHostName();
+	//!
+	void ComputeBandwidth();
+	//!
+	NRESULT GetSocketAddresses(CIPAddress *pAddr, DWORD nMaCIPAddresses);
+	//!
+	int GetLastError(){return WSAGetLastError();}
+	//!
+	void Close();
+
+private:
+	SOCKET						m_hSocket;										//!<
+	SocketType				m_stSocketType;								//!<
+	CIPAddress				m_saDefaultAddress;						//!< Default target host and port [optional] for Send()
+	unsigned int			m_nStartTick;									//!< tick for bandwitdh computation
+	
+	unsigned int			m_nSentBytesInThisSec;				//!< is counting up and reseted every second
+	unsigned int			m_nReceivedBytesInThisSec;		//!< is counting up and reseted every second
+	unsigned int			m_nSentPacketsInThisSec;			//!< is counting up and reseted every second
+	unsigned int			m_nReceivedPacketsInThisSec;	//!< is counting up and reseted every second
+#if defined(LINUX)
+	struct ip_mreq		m_imMulticastReq;							//!< needed for call to IP_DROP_MEMBERSHIP
+#endif
+
+public:
+	float							m_fOutgoingKbPerSec;					//!< is updated every second
+	float							m_fIncomingKbPerSec;					//!< is updated every second
+	unsigned int			m_nOutgoingPacketsPerSec;			//!< is updated every second
+	unsigned int			m_nIncomingPacketsPerSec;			//!< is updated every second
+};
+
+
+#endif //_DATAGRAM_SOCKET_H_