Multithreaded Non-blocking Socket Server and Client, Based on Synchronized Socket

// Synchronized socket client connector
// Author: Kevin Hua
// Create: 2002-2-26 15:20
// Revision: 1
// Contact: Kevin-Hua@Woncore.com
//////////////////////////////////////////////////////////////////////
#if !defined(AFX_SYNCSOCKETCLIENT_H__6C21FFA2_485A_11D5_9692_0050BA8CD8A0__INCLUDED_)
#define AFX_SYNCSOCKETCLIENT_H__6C21FFA2_485A_11D5_9692_0050BA8CD8A0__INCLUDED_

#if _MSC_VER > 1000
#pragma once
#endif // _MSC_VER > 1000

#pragma warning(disable:4786)

#include <winsock2.h>
#include <string>
#include <process.h>

using namespace std;


#ifdef _DEBUG
#define new DEBUG_NEW
#endif

#pragma warning(disable : 4786)

template<class TSocketWorker>
class CSyncSocketClient
{
public:
	CSyncSocketClient()
	{
		m_nPeerPort = 0;
		m_strPeerAddr = "";
		m_pWorker = NULL;
		m_bCloseWait = TRUE;
		m_nTimeout = -1;

		m_bWSAStartup = TRUE;
		
		WORD wVersionRequested;
		WSADATA wsaData;
		int result;
		 
		wVersionRequested = MAKEWORD(2, 2);
		 
		result = WSAStartup(wVersionRequested, &wsaData);
		if(result != 0)
		{
		    m_bWSAStartup = FALSE;
		}
		 
		if (LOBYTE( wsaData.wVersion) != 2 ||
		        HIBYTE(wsaData.wVersion) != 2) 
		{
		    WSACleanup();
		    m_bWSAStartup = FALSE; 
		}		
	}

	CSyncSocketClient(const char * pszServerAddr, unsigned short nServerPort, BOOL bCloseWait = TRUE)
	{
		m_nPeerPort = nServerPort;
		m_strPeerAddr = pszServerAddr;
		m_pWorker = NULL;
		m_bCloseWait = bCloseWait;
		m_nTimeout = -1;
		
		m_bWSAStartup = TRUE;
		
		WORD wVersionRequested;
		WSADATA wsaData;
		int result;
		 
		wVersionRequested = MAKEWORD(2, 2);
		 
		result = WSAStartup(wVersionRequested, &wsaData);
		if(result != 0)
		{
		    m_bWSAStartup = FALSE;
		}
		 
		if (LOBYTE( wsaData.wVersion) != 2 ||
		        HIBYTE(wsaData.wVersion) != 2) 
		{
		    WSACleanup();
		    m_bWSAStartup = FALSE; 
		}		
	}

	virtual	~CSyncSocketClient()
	{
		CloseConnection();

		if(m_bWSAStartup)
			WSACleanup();
	}

	
	string					m_strPeerAddr;
	int						m_nPeerPort;
	

	BOOL IsConnected()
	{
		if(m_pWorker)
			return m_pWorker->IsConnected();
		else
			return FALSE;
	}
	
	unsigned long GetTimeout()
	{
		return m_nTimeout;
	}

	unsigned long SetTimeout(unsigned long nNewVal)
	{
		if(nNewVal == 0)
			nNewVal = -1;

		unsigned long nOldVal = m_nTimeout;
		m_nTimeout = nNewVal;

		if(m_pWorker)
			m_pWorker->SetTimeout(m_nTimeout);
		
		return nOldVal;
	}

public:						
	void CloseConnection()
	{
		if(m_pWorker)
		{
			m_pWorker->Stop();		
			delete m_pWorker;
			m_pWorker = NULL;
		}
	}

	BOOL Connect()
	{
		if(!m_bWSAStartup)
			return FALSE;
		
		if(IsConnected())
			return FALSE;
		
		if(!m_strPeerAddr.size())
			return FALSE;		 

		SOCKET s;
		struct sockaddr_in peer;


		peer.sin_family = PF_INET;
		peer.sin_port = htons(m_nPeerPort);
		peer.sin_addr.s_addr = inet_addr(m_strPeerAddr.c_str());

		s = socket(AF_INET, SOCK_STREAM, 0);

		if(INVALID_SOCKET == s)
		{
			return FALSE;
		}

		if(SOCKET_ERROR == connect(s, (struct sockaddr*)&peer, sizeof(peer)))
		{
			shutdown(s, SD_SEND);
			closesocket(s);
			return FALSE;
		}

		m_pWorker = new TSocketWorker(s, m_strPeerAddr.c_str(), m_nPeerPort, TRUE);
		if(!m_pWorker)
		{
			shutdown(s, SD_SEND);
			closesocket(s);
			return FALSE;			
		}

		m_pWorker->SetTimeout(m_nTimeout);
		if(!m_pWorker->Start())
		{
			shutdown(s, SD_SEND);
			closesocket(s);
			return FALSE;			
		}	

		return TRUE;
	}
	
	BOOL Connect(const char* pszServerAddr, unsigned short nServerPort, BOOL bCloseWait = TRUE) 
	{
		if(IsConnected())
			return FALSE;
		
		if(!pszServerAddr)
			return FALSE;

		m_bCloseWait = bCloseWait;
		m_strPeerAddr = pszServerAddr;
		m_nPeerPort = nServerPort;

		return Connect();
	}	

private:	
	CSyncSocketClient(const CSyncSocketClient& rhs)
	{
	}
	CSyncSocketClient& operator=(const CSyncSocketClient& rhs)
	{
		return *this;
	}
	
	TSocketWorker*			m_pWorker;
	BOOL					m_bCloseWait;
	unsigned long			m_nTimeout;
	BOOL					m_bWSAStartup;
};


#endif // !defined(AFX_SYNCSOCKETCLIENT_H__6C21FFA2_485A_11D5_9692_0050BA8CD8A0__INCLUDED_)