//	Implementation of class CTCPServer
//	Author: ShiXingdong
//	Last modify: Jun 1st, 2002

#include <string>
#include "TCPServer.h"

using namespace std;

const int g_nBufSize=10240;	// buffer size = 10 KB

bool g_bStopWorking;	//停止工作的標志
bool g_bTurnOn;			//服務器已打開的標志
bool  (*g_pReceptionist)(SOCKET sct);	//相應客戶端請求的函數的指針
DWORD WINAPI g_AcceptClient(LPVOID lpParam);	//接收客戶端連接的函數
DWORD WINAPI g_ProcessClientRequest(LPVOID lpParam);	//響應客戶端請求的函數


bool CTCPServer::Initialize(unsigned short nPortNum)
{
	g_bStopWorking=false;
	g_bTurnOn=false;
	g_pReceptionist=NULL;

	m_nPortNum=nPortNum;
	m_ServerAddress.sin_family=AF_INET;
	m_ServerAddress.sin_port=htons(nPortNum);
	m_ServerAddress.sin_addr.s_addr=htonl(INADDR_ANY);

	//建立套接字
	if((m_sctListenSocket=socket(AF_INET,SOCK_STREAM,IPPROTO_IP))==INVALID_SOCKET)
		return(false);
	//幫定套接字
	if(bind(m_sctListenSocket,(sockaddr*)&m_ServerAddress,sizeof(m_ServerAddress))==SOCKET_ERROR)
		return(false);
	//偵聽
	listen(m_sctListenSocket,SOMAXCONN);
	return(true);
}

CTCPServer::CTCPServer(unsigned short nPortNum)
{
	m_bInitSuccess=false;
	WSADATA data;
	if(WSAStartup(MAKEWORD(2,2),&data)!=0)
		return;
	if(!Initialize(nPortNum))
		return;
	DWORD dwThreadID;
	
	//建立接收客戶端連接的線程
	if(CreateThread(NULL,0,g_AcceptClient,(LPVOID)m_sctListenSocket,0,&dwThreadID)==NULL)
		return;
	m_bInitSuccess=true;
	return;
}

CTCPServer::~CTCPServer()
{
	g_bStopWorking=true;
	WSACleanup();
}


bool CTCPServer::SetReceptionist(bool (*pFunc)(SOCKET sct))
{
	g_pReceptionist=pFunc;
	return(true);
}


bool CTCPServer::IsOK() const
{
	return(m_bInitSuccess&&(g_pReceptionist!=NULL));
}

string CTCPServer::GetIPAddr() const
{
	char szHostName[128];
	struct hostent * pHost;
	int i; 

	if(gethostname(szHostName,128)==0)
	{
		pHost = gethostbyname(szHostName);
		for(i=0; (pHost!= NULL) && (pHost->h_addr_list[i]!= NULL); i++ )
		{
			return(string(inet_ntoa (*(struct in_addr *)pHost->h_addr_list[i])));
		}
	}
	return(string(""));
}

bool CTCPServer::TurnOn()
{
	g_bTurnOn=true;
	return(true);
}

bool CTCPServer::TurnOff()
{
	g_bTurnOn=false;
	return(true);
}

DWORD WINAPI g_AcceptClient(LPVOID lpParam)
{
	SOCKET sctListenSocket=(SOCKET)lpParam;
	SOCKET *pConnectSocket;
	sockaddr_in ClientAddress;
	int iAddressSize=sizeof(ClientAddress);
	int bReuse=true;
	DWORD dwThreadID;
	while(!g_bStopWorking)
	{	
		if(!g_bTurnOn)
			continue;
		pConnectSocket=new SOCKET;
		*pConnectSocket=accept(sctListenSocket,(sockaddr*)&ClientAddress,&iAddressSize);
		setsockopt(*pConnectSocket,SOL_SOCKET,SO_REUSEADDR,(char *)&bReuse,sizeof(bReuse));
		if(*pConnectSocket!=INVALID_SOCKET)
			CreateThread(NULL,0,g_ProcessClientRequest,(LPVOID)*pConnectSocket,0,&dwThreadID);
	}
	return(0);
}

DWORD WINAPI g_ProcessClientRequest(LPVOID lpParam)
{
	SOCKET sctClientSocket=(SOCKET)lpParam;

	while(!g_bStopWorking)
	{
		if(!g_bTurnOn)	//服務器未打開
			continue;
		if(g_pReceptionist==NULL)	//未設置接待函數
			continue;
		if(!g_pReceptionist(sctClientSocket))	//不再響應客戶端請求
			break;
	}
	return(0);
}