// NetClient.cpp: implementation of the CNetClient class.
//
//////////////////////////////////////////////////////////////////////

#include "stdafx.h"
#include "NetClient.h"
//#include <fstream.h>

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

//////////////////////////////////////////////////////////////////////
// Construction/Destruction
//////////////////////////////////////////////////////////////////////
CNetClient::CNetClient()
{
	IsStart=false;
	bOprUnInit=false;
	m_pProcessRecvData=NULL;
	hSendEvent=CreateEvent(NULL,false,false,NULL);
	hWorkEvent=CreateEvent(NULL,false,false,NULL);
	hExitEvent=CreateEvent(NULL,false,false,NULL);
}
CNetClient::~CNetClient()
{
	CloseHandle(hSendEvent);
	CloseHandle(hWorkEvent);
	CloseHandle(hExitEvent);
}
DWORD _stdcall SendProc(LPVOID pParam)
{
	CNetClient * pNetClient=(CNetClient*) pParam;
	HANDLE event[2];
	event[0]=pNetClient->hSendEvent;
	event[1]=pNetClient->hExitEvent;

	while(true)
	{
		Sleep(1);
		::EnterCriticalSection(&pNetClient->SendMsgQueSection);
		//隊列為空，等待發送事件觸發
		if(pNetClient->SendMsgQueue.IsEmpty())
		{
			::LeaveCriticalSection(&pNetClient->SendMsgQueSection);
			//為空，或者發送完畢
			ResetEvent(pNetClient->hSendEvent);
			TRACE(L"\nTheSendProc Is Waiting....");	
			DWORD Index=::WaitForMultipleObjects(2,event,false,INFINITE);
			if((Index-WAIT_OBJECT_0)==1)return 0L;
		}
		else
		{
			//取下一個結點，并發送
			MSG_NODE p=pNetClient->SendMsgQueue.DeQueue();
			//釋放隊列
			::LeaveCriticalSection(&pNetClient->SendMsgQueSection);
			DWORD retlen;
			bool bRet=pNetClient->m_sClient.SendMsg(p.pData,p.DataLength,&retlen,WSA_INFINITE);
			if(bRet==false || retlen!=p.DataLength)
			{
				if(GetLastError()!=CLIENT_FUNERROR)
				pNetClient->m_pProcessRecvData(NULL,0,0);
				pNetClient->UnInit();
			}	
		}
	}
	return 0L;
}
DWORD _stdcall WorkProc(LPVOID pParam)
{
	CNetClient* pNetClient=(CNetClient*)pParam;
	HANDLE event[2];
	event[0]=pNetClient->hWorkEvent;
	event[1]=pNetClient->hExitEvent;
	
	while(true)
	{
		Sleep(1);
		::EnterCriticalSection(&pNetClient->RecvMsgQueSection);
		//隊列為空，等待發送事件觸發
		if(pNetClient->RecvMsgQueue.IsEmpty())
		{
			::LeaveCriticalSection(&pNetClient->RecvMsgQueSection);
			//為空，或者發送完畢
			ResetEvent(pNetClient->hWorkEvent);
			TRACE(L"\nTheWorkProc Is Waiting....");			
			DWORD Index=::WaitForMultipleObjects(2,event,false,INFINITE);
			if((Index-WAIT_OBJECT_0)==1)return 0L;
		}
		else
		{
			//取下一個結點，并發送
			MSG_NODE p=pNetClient->RecvMsgQueue.DeQueue();
			//釋放隊列
			::LeaveCriticalSection(&pNetClient->RecvMsgQueSection);
			//調用回調函數，處理數據
			pNetClient->m_pProcessRecvData(p.pData,p.DataLength,pNetClient->m_dwUserData);
		}
	}
	return 0L;
}
DWORD _stdcall RecvProc(LPVOID pParam)
{
	CNetClient * pNetClient=(CNetClient*)pParam;
	char RecvBuf[BUFFER_SIZE];
	DWORD retlen;
	while (true)
	{
		Sleep(1);
		TRACE(L"\nTheRecvThread Is Waiting...");
		if(!pNetClient->m_sClient.RecvMsg(RecvBuf,BUFFER_SIZE,&retlen,WSA_INFINITE) &&  GetLastError()!=CLIENT_FUNERROR)
		{
			if(pNetClient->bOprUnInit)return 0L;//應用程序請求退出	
			//連接已經被斷開，通知上層(通過調用回調函數)
			pNetClient->m_pProcessRecvData(NULL,0,0);
			pNetClient->UnInit();
			return 0L;
		}
		else
		{
			//沒收到字節？還是出錯
			if(retlen==0)
			{
				pNetClient->m_pProcessRecvData(NULL,0,0);
				pNetClient->UnInit();
				return 0L;
			}
			//將接收到的數據放到接收隊列里
			MSG_NODE Msg;
			Msg.DataLength=retlen;
			memcpy(Msg.pData,RecvBuf,retlen);
			//插入消息隊列
			::EnterCriticalSection(&pNetClient->RecvMsgQueSection);
			if(pNetClient->SendMsgQueue.IsEmpty())
			{
				pNetClient->RecvMsgQueue.EnQueue(Msg);
				::LeaveCriticalSection(&pNetClient->RecvMsgQueSection);
				//如果消息隊列為空，告訴工作線程可以進行工作了
				SetEvent(pNetClient->hWorkEvent);
			}
			else
			{
				pNetClient->RecvMsgQueue.EnQueue(Msg);
				::LeaveCriticalSection(&pNetClient->RecvMsgQueSection);
			}
		}
	}
	return 0L;
}
bool CNetClient::Init(ProcessRecvData* pProcessRecvData,LPCSTR szSvrAddr, unsigned long iSvrPort,DWORD userdata)
{
	if(pProcessRecvData==NULL //回調函數空
		|| szSvrAddr==NULL //地址空
		|| IsStart)//已經啟動過了
		return false;	
	::InitializeCriticalSection(&SendMsgQueSection);
	::InitializeCriticalSection(&RecvMsgQueSection);	
	IsStart=false;
	bOprUnInit=false;
	m_pProcessRecvData=pProcessRecvData;
	m_dwUserData=userdata;
	int  bRet=InitNetWork(szSvrAddr,iSvrPort,HostIpAddr);
	if(0==bRet)
	{
		IsStart=true;
		return true;
	}
	else
		return false;
}
void CNetClient::UnInit()
{
	if(!IsStart)return;
	bOprUnInit=true;
	SetEvent(hExitEvent);	
	SetEvent(hExitEvent);
	m_sClient.UnInit();	
	::DeleteCriticalSection(&SendMsgQueSection);
	::DeleteCriticalSection(&RecvMsgQueSection);
	SendMsgQueue.MakeEmpty();
	RecvMsgQueue.MakeEmpty();	
	m_pProcessRecvData=NULL;
	IsStart=false;
}
int CNetClient::InitNetWork(LPCSTR szSvrAddr,unsigned int SvrPort,LPCSTR pHostIpAddress)
{
	CString LogStr;
	int Error=0;
	WSADATA wsaData;
	char Name[100];
	hostent *pHostEntry;
	in_addr rAddr;
	memset((void *)pHostIpAddress,0,sizeof(pHostIpAddress));
	//Net Start Up
	Error=WSAStartup(MAKEWORD(0x02,0x02),&wsaData);
	if(Error!=0)
	{
		Error = WSAGetLastError();
		LogStr.Format(L"WSAStartUp Faild With Error: %d",Error);
		WriteLogString(LogStr);

		return Error;
	}
	//Make Version
	if ( LOBYTE( wsaData.wVersion ) != 2 ||
         HIBYTE( wsaData.wVersion ) != 2 )
	{
		WSACleanup( );
		
		//WriteLogString("The Local Net Version Is not 2");

		return -1;
	}
	//Get Host Ip
	Error = gethostname ( Name, sizeof(Name) );
	if( 0 == Error )
	{
		pHostEntry = gethostbyname( Name );
		if( pHostEntry != NULL )
		{
			memcpy( &rAddr, pHostEntry->h_addr_list[0], sizeof(struct in_addr) );
			sprintf((char * )pHostIpAddress,"%s",inet_ntoa( rAddr ));
		}
		else
		{
			Error = WSAGetLastError();
			LogStr.Format(L"GetHostIp faild with Error: %d",Error);
			//WriteLogString(LogStr);

		}
	}
	else
	{
		Error = WSAGetLastError();
		LogStr.Format(L"gethostname faild with Error: %d",Error);
		//WriteLogString(LogStr);
	}
	//Socket Create
	if(0==Error)
	{
		if(!m_sClient.CreateSocket(SOCK_STREAM))
		{
			Error=WSAGetLastError();
			LogStr.Format(L"Create Client Socket Faild :%d",Error);
			//WriteLogString(LogStr);
			return Error;
		}
	}
	if(0==Error)
	{
		if(!m_sClient.ConnectSocket((char *)szSvrAddr,SvrPort))
		{
			Error=WSAGetLastError();
			LogStr.Format(L"Create Client Socket Faild :%d",Error);
			WriteLogString(LogStr);
			return Error;
		}
	}
	//啟動工作線程，并升高工作線程的等級至最高
	if(0==Error)
	{
		HANDLE WorkHan=0;
		unsigned long WorkID;
		if((WorkHan=CreateThread(NULL,0,WorkProc,this,0,&WorkID))==NULL)
		{
			Error=GetLastError();
			LogStr.Format(L"Create WorkThread Faild With Error %d",Error);
			WriteLogString(LogStr);
			return Error;
		}
		SetThreadPriority(WorkHan,THREAD_PRIORITY_HIGHEST);
		CloseHandle(WorkHan);
	}
	//啟動接收線程
	if(0==Error)
	{
		HANDLE RecvHan=0;
		unsigned long RecvID;
		if((RecvHan=CreateThread(NULL,0,RecvProc,this,0,&RecvID))==NULL)
		{
			Error=GetLastError();
			LogStr.Format(L"Create RecvThread Faild With Error %d",Error);
			WriteLogString(LogStr);
			SetEvent(hExitEvent);//退出先前創建的線程
			return Error;
		}
		CloseHandle(RecvHan);
	}
	//啟動發送線程
	if(0==Error)
	{
		HANDLE ThreHan;
		unsigned long ThrID;
		if((ThreHan=CreateThread(NULL,0,SendProc,this,0,&ThrID))==NULL)
		{
			Error=GetLastError();
			LogStr.Format(L"Create SEND Thred Faild With Error %d",Error);
			WriteLogString(LogStr);