Utilisateur : Fcolin Date : 16/06/00 Heure : 10:14 Créé (vss 1)

1 files changed, 169 insertions, 0 deletions

diff --git a/Ivy/ThreadedSocket.h b/Ivy/ThreadedSocket.h
new file mode 100644
index 0000000..19fca4f
--- /dev/null
+++ b/Ivy/ThreadedSocket.h
@@ -0,0 +1,169 @@
+// ThreadedSocket.h: interface for the CThreadedSocket class.

+//

+//////////////////////////////////////////////////////////////////////

+

+#if !defined(AFX_THREADEDSOCKET_H__DC45EFD8_343F_11D3_8A15_00A0245B298A__INCLUDED_)

+#define AFX_THREADEDSOCKET_H__DC45EFD8_343F_11D3_8A15_00A0245B298A__INCLUDED_

+

+#if _MSC_VER > 1000

+#pragma once

+#endif // _MSC_VER > 1000

+

+#include <winsock2.h>

+

+class CThreadedSocketException //: public CException

+{

+//	DECLARE_DYNAMIC(CThreadedSocketException)

+public:

+// Constructor

+	CThreadedSocketException(char* pchMessage);

+

+public:

+	~CThreadedSocketException() {}

+	virtual bool GetErrorMessage(string& lpstrError, unsigned int nMaxError,

+		unsigned int * pnHelpContext = 0);

+	int GetError() { return m_nError; };

+private:

+	int m_nError;

+	char * m_strMessage;

+};

+

+class CThreadedSocket  //: public CObject

+{

+//	DECLARE_DYNAMIC(CThreadedSocket);

+//	typedef unsigned int SOCKET;

+//	typedef struct sockaddr SOCKADDR;

+

+public:

+	CThreadedSocket();

+	virtual ~CThreadedSocket();

+// Construction

+public:

+	void Create(unsigned int nSocketPort = 0, int nSocketType=SOCK_STREAM,

+		long lEvent = FD_READ | FD_WRITE | FD_OOB | FD_ACCEPT | FD_CONNECT | FD_CLOSE,

+		const char * lpszSocketAddress = 0);

+

+// Attributes

+public:

+	SOCKET m_hSocket;

+

+	operator SOCKET() const;

+	

+	void GetPeerName(string & rPeerAddress, unsigned int& rPeerPort);

+

+	void GetSockName(string & rSocketAddress, unsigned int& rSocketPort);

+

+// Operations

+public:

+	void Socket(int nSocketType=SOCK_STREAM, long lEvent =

+		FD_READ | FD_WRITE | FD_OOB | FD_ACCEPT | FD_CONNECT | FD_CLOSE,

+		int nProtocolType = 0, int nAddressFormat = PF_INET);

+	virtual void Accept(CThreadedSocket& rConnectedSocket,

+		SOCKADDR* lpSockAddr = NULL, int* lpSockAddrLen = NULL);

+

+	void Bind(unsigned int nSocketPort, const char * lpszSocketAddress = 0);

+

+	virtual void Close();

+

+	bool Connect(const char * lpszHostAddress, unsigned int nHostPort);

+	bool Connect(const SOCKADDR* lpSockAddr, int nSockAddrLen);

+	

+	virtual int Receive(void* lpBuf, int nBufLen, int nFlags = 0);

+

+	int ReceiveFrom(void* lpBuf, int nBufLen,

+		string & rSocketAddress, unsigned int & rSocketPort, int nFlags = 0);

+

+	enum { receives = 0, sends = 1, both = 2 };

+

+	virtual void Send(const void* lpBuf, int nBufLen, int nFlags = 0);

+

+	void SendTo(const void* lpBuf, int nBufLen,

+		unsigned int nHostPort, const char * lpszHostAddress = 0, int nFlags = 0);

+	void AsyncSelect(long lEvent =

+		FD_READ | FD_WRITE | FD_OOB | FD_ACCEPT | FD_CONNECT | FD_CLOSE);

+

+	

+inline CThreadedSocket::operator SOCKET() const

+	{ return m_hSocket; }

+inline void GetPeerName(SOCKADDR* lpSockAddr, int* lpSockAddrLen)

+	{ 

+	if ( getpeername(m_hSocket, lpSockAddr, lpSockAddrLen) == SOCKET_ERROR)

+		throw( new CThreadedSocketException( "getpeername" ) );

+	}

+inline void GetSockName(SOCKADDR* lpSockAddr, int* lpSockAddrLen)

+	{ if ( getsockname(m_hSocket, lpSockAddr, lpSockAddrLen)== SOCKET_ERROR)

+		throw( new CThreadedSocketException( "getsockname" ) );

+	}

+inline void SetSockOpt(int nOptionName, const void* lpOptionValue, int nOptionLen, int nLevel = SOL_SOCKET)

+	{ 

+	if ( setsockopt(m_hSocket, nLevel, nOptionName, (LPCSTR)lpOptionValue, nOptionLen)== SOCKET_ERROR)

+		throw( new CThreadedSocketException( "setsockopt" ) );

+	}

+inline void GetSockOpt(int nOptionName, void* lpOptionValue, int* lpOptionLen, int nLevel = SOL_SOCKET)

+	{

+	if ( getsockopt(m_hSocket, nLevel, nOptionName, (LPSTR)lpOptionValue, lpOptionLen)== SOCKET_ERROR)

+		throw( new CThreadedSocketException( "getsockopt" ) );

+	}

+

+static inline int CThreadedSocket::GetLastError()

+	{ return WSAGetLastError(); }

+inline void Bind(const SOCKADDR* lpSockAddr, int nSockAddrLen)

+	{ 

+	if ( bind(m_hSocket, lpSockAddr, nSockAddrLen)== SOCKET_ERROR ) 

+		throw( new CThreadedSocketException( "bind" ));

+	}

+

+inline void IOCtl(long lCommand, unsigned long* lpArgument)

+	{ if ( ioctlsocket(m_hSocket, lCommand, lpArgument)== SOCKET_ERROR)

+		throw( new CThreadedSocketException( "bind" ) );

+	}

+inline void Listen(int nConnectionBacklog=5)

+	{ 

+	if (  listen(m_hSocket, nConnectionBacklog)== SOCKET_ERROR)

+		throw( new CThreadedSocketException( "listen" ) );

+	}

+inline int ReceiveFrom(void* lpBuf, int nBufLen, SOCKADDR* lpSockAddr, int* lpSockAddrLen, int nFlags = 0)

+	{

+	int lg = recvfrom(m_hSocket, (LPSTR)lpBuf, nBufLen, nFlags, lpSockAddr, lpSockAddrLen); 

+	if ( lg == SOCKET_ERROR )

+		throw( new CThreadedSocketException( "recvfrom" ) );

+	return lg;

+	}

+inline bool ShutDown(int nHow = sends)

+	{ 

+	if (  shutdown(m_hSocket,nHow) == SOCKET_ERROR ) 

+		throw( new CThreadedSocketException( "shutdown" ) );

+	}

+inline void SendTo(const void* lpBuf, int nBufLen, const SOCKADDR* lpSockAddr, int nSockAddrLen, int nFlags = 0)

+	{

+	if (  sendto(m_hSocket, (LPSTR)lpBuf, nBufLen, nFlags, lpSockAddr, nSockAddrLen)== SOCKET_ERROR )

+		throw( new CThreadedSocketException( "sendto" ) );

+	}

+

+

+

+// Overridable callbacks

+protected:

+	virtual void OnWakeup();

+	virtual void OnReceive(int nErrorCode);

+	virtual void OnSend(int nErrorCode);

+	virtual void OnOutOfBandData(int nErrorCode);

+	virtual void OnAccept(int nErrorCode);

+	virtual void OnConnect(int nErrorCode);

+	virtual void OnClose(int nErrorCode);

+// Implementation 

+	void StartListener(long lEvent);

+	long m_EventMask;

+	unsigned long thread_id;

+	HANDLE h_thread;

+	WSAEVENT m_hEvent[2];   // socket event for receive and send actions

+	unsigned int SocketThread();	//Client Thread

+

+// wrapper for thread routine

+static DWORD WINAPI SocketThreadProc( void * pParam );

+static void Init();

+static bool Initialized;

+

+};

+

+#endif // !defined(AFX_THREADEDSOCKET_H__DC45EFD8_343F_11D3_8A15_00A0245B298A__INCLUDED_)