// TCPClient_CE.cpp: implementation of the CTCPClient_CE class.
//
//////////////////////////////////////////////////////////////////////

#include "stdafx.h"
#include "TCPClient.h"
#include "TCPClient_CE.h"

#ifdef _DEBUG
#undef THIS_FILE
static char THIS_FILE[]=__FILE__;
#define new DEBUG_NEW
#endif

//////////////////////////////////////////////////////////////////////
// Construction/Destruction
//////////////////////////////////////////////////////////////////////

//構(gòu)造函數(shù)
CTCPClient_CE::CTCPClient_CE()
{
   //初始化socket環(huán)境
   WSADATA wsd;
   WSAStartup(MAKEWORD(1,1),&wsd);
   //創(chuàng)建線程退出事件句柄
   m_exitThreadEvent = CreateEvent(NULL,FALSE,FALSE,NULL);
}

//析構(gòu)函數(shù)
CTCPClient_CE::~CTCPClient_CE()
{
   //釋放socket資源
   WSACleanup();
   //關(guān)閉線程退出事件句柄
   CloseHandle(m_exitThreadEvent);
}

/*--------------------------------------------------------------------
【函數(shù)介紹】:  此線程用于監(jiān)聽(tīng)TCP客戶端通訊的事件，例如當(dāng)接收到數(shù)據(jù)、
			   連接斷開(kāi)和通訊過(guò)程發(fā)生錯(cuò)誤等事件
【入口參數(shù)】:  lparam:無(wú)類型指針，可以通過(guò)此參數(shù)，向線程中傳入需要用到的資源。
			   在這里我們將CTCPClient_CE類實(shí)例指針傳進(jìn)來(lái)
【出口參數(shù)】:  (無(wú))
【返回  值】:  返回值沒(méi)有特別的意義，在此我們將返回值設(shè)為0。
---------------------------------------------------------------------*/
DWORD CTCPClient_CE::SocketThreadFunc(LPVOID lparam)
{
	CTCPClient_CE *pSocket;
	//得到CTCPClient_CE實(shí)例指針
	pSocket = (CTCPClient_CE*)lparam;
	//定義讀事件集合
	fd_set fdRead;
	int ret;
	//定義事件等待時(shí)間
	TIMEVAL	aTime;
	aTime.tv_sec = 1;
	aTime.tv_usec = 0;
	while (TRUE)
	{
        //收到退出事件，結(jié)束線程
		if (WaitForSingleObject(pSocket->m_exitThreadEvent,0) == WAIT_OBJECT_0)
		{
			break;
		}
		//置空f(shuō)dRead事件為空
		FD_ZERO(&fdRead);
		//給客戶端socket設(shè)置讀事件
		FD_SET(pSocket->m_socket,&fdRead);
		//調(diào)用select函數(shù)，判斷是否有讀事件發(fā)生
		ret = select(0,&fdRead,NULL,NULL,&aTime);
		
		if (ret == SOCKET_ERROR)
		{
			//觸發(fā)錯(cuò)誤事件
			pSocket->OnError(pSocket->m_pOwnerWnd,1);
			//觸發(fā)連接斷開(kāi)事件
			pSocket->OnDisConnect(pSocket->m_pOwnerWnd);
			//關(guān)閉客戶端socket
			closesocket(pSocket->m_socket);
			break;
		}
		
		
		if (ret > 0)
		{
			if (FD_ISSET(pSocket->m_socket,&fdRead))
			{
				//發(fā)生讀事件
				char recvBuf[1024];
				int recvLen;
				ZeroMemory(recvBuf,1024);
				//接收數(shù)據(jù)
				recvLen = recv(pSocket->m_socket,recvBuf, 1024,0); 
				if (recvLen == SOCKET_ERROR)
				{
					int iError = WSAGetLastError();
					//觸發(fā)socket錯(cuò)誤事件
					pSocket->OnError(pSocket->m_pOwnerWnd,iError);
					//觸發(fā)與服務(wù)器斷開(kāi)事件
					pSocket->OnDisConnect(pSocket->m_pOwnerWnd);
					//關(guān)閉客戶端socket
					closesocket(pSocket->m_socket);
					break;
				}
				else if (recvLen == 0)
				{
					//觸發(fā)與服務(wù)器端斷開(kāi)事件
					pSocket->OnDisConnect(pSocket->m_pOwnerWnd);
					//關(guān)閉客戶端socket
					closesocket(pSocket->m_socket);
					break;
				}
				else
				{
				   //觸發(fā)數(shù)據(jù)接收事件
                   pSocket->OnRead(pSocket->m_pOwnerWnd,recvBuf,recvLen);
				}
			}
		}
	}
	return 0;
}

/*--------------------------------------------------------------------
【函數(shù)介紹】: 用于打開(kāi)客戶端socket
【入口參數(shù)】: pWnd用于指定父窗口句柄
【出口參數(shù)】: (無(wú))
【返回  值】: TRUE:打開(kāi)成功;FALSE:打開(kāi)失敗
---------------------------------------------------------------------*/
bool CTCPClient_CE::Open(CWnd * pWnd)
{
   //復(fù)位線程退出事件
   ResetEvent(m_exitThreadEvent);
   //存儲(chǔ)父窗口句柄
   m_pOwnerWnd = pWnd;
   //創(chuàng)建TCP套接字 
   m_socket = socket(AF_INET,SOCK_STREAM,IPPROTO_TCP);
   if (m_socket == SOCKET_ERROR)
   {
	   return FALSE;
   }
   //創(chuàng)建通訊線程
   m_tcpThreadHandle = CreateThread(NULL,0,SocketThreadFunc,this,0,NULL);
   if (m_tcpThreadHandle == NULL)
   {
	   closesocket(m_socket);
	   return FALSE;
   }
   return TRUE;
}


/*--------------------------------------------------------------------
【函數(shù)介紹】: 用于關(guān)閉客戶端socket
【入口參數(shù)】:  (無(wú))
【出口參數(shù)】:  (無(wú))
【返回  值】: TRUE:關(guān)閉成功;FALSE:關(guān)閉失敗
---------------------------------------------------------------------*/
bool CTCPClient_CE::Close()
{
   //發(fā)送通訊線程結(jié)束事件
   SetEvent(m_exitThreadEvent);
   Sleep(1000);
   //關(guān)閉Socket，釋放資源
   int err = closesocket(m_socket);
   if (err == SOCKET_ERROR)
   {
	   return FALSE;
   }
   return TRUE;
}


/*--------------------------------------------------------------------
【函數(shù)介紹】: 用于建立與TCP服務(wù)器連接
【入口參數(shù)】: (無(wú))
【出口參數(shù)】: (無(wú))
【返回  值】: TRUE:建立連接成功;FALSE:建立連接失敗
---------------------------------------------------------------------*/
bool CTCPClient_CE::Connect()
{
   struct sockaddr_in addr;
   int err;

   addr.sin_family = AF_INET;
   addr.sin_port = htons(m_port);
   //此處要將雙字節(jié)轉(zhuǎn)換成單字節(jié)
   char ansiRemoteHost[255];
	ZeroMemory(ansiRemoteHost,255);
	WideCharToMultiByte(CP_ACP,WC_COMPOSITECHECK,m_remoteHost,wcslen(m_remoteHost)
		,ansiRemoteHost,wcslen(m_remoteHost),NULL,NULL);

   addr.sin_addr.s_addr=inet_addr(ansiRemoteHost);
   //此時(shí)采用同步連接方式,connect直接返回成功或是失敗
   err = connect(m_socket,(struct sockaddr *)&addr,sizeof(addr));
   if (err == SOCKET_ERROR) 
   {
	   return FALSE;
   }
   //設(shè)置通訊模式為異步模式
   DWORD ul= 1;
   ioctlsocket(m_socket,FIONBIO,&ul);
   return TRUE;
}

/*--------------------------------------------------------------------
【函數(shù)介紹】: 向服務(wù)器端發(fā)送數(shù)據(jù)
【入口參數(shù)】: buf:待發(fā)送的數(shù)據(jù)
              len:待發(fā)送的數(shù)據(jù)長(zhǎng)度
【出口參數(shù)】: (無(wú))
【返回  值】: TRUE:發(fā)送數(shù)據(jù)成功;FALSE:發(fā)送數(shù)據(jù)失敗
---------------------------------------------------------------------*/
bool CTCPClient_CE::SendData(const char * buf , int len)
{
	int nBytes = 0;
	int nSendBytes=0;
			
	while (nSendBytes < len)
	{
	    nBytes = send(m_socket,buf+nSendBytes,len-nSendBytes,0);
		if (nBytes==SOCKET_ERROR )
		{
			int iErrorCode = WSAGetLastError();
			//觸發(fā)socket的Error事件
			OnError(m_pOwnerWnd,iErrorCode);
			//觸發(fā)與服務(wù)器端斷開(kāi)連接事件
			OnDisConnect(m_pOwnerWnd);
			//關(guān)閉socket
			Close();
			return FALSE;
		}

		nSendBytes = nSendBytes + nBytes;
		
		if (nSendBytes < len)
		{
		    Sleep(1000);
		}
	} 
	return TRUE; 
}