// Server.cpp: implementation of the CServer class.
//
//////////////////////////////////////////////////////////////////////

#include "stdafx.h"
#include "ChatRoom.h"
#include "ServerClient.h"
#include "ChatRoomDlg.h"


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

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

CServer::CServer()
{
	m_hSocket=NULL;
}

CServer::~CServer()
{
	WSAAsyncSelect(m_hSocket, m_hWnd, 0, 0);
}

BOOL CServer::InitAndListen(HWND hwnd,UINT port)
{
	m_uPort=port;
	m_hWnd=hwnd;

	if(m_hSocket != NULL)
    {
        //如果已經創建了套接字先關閉原來的
		closesocket(m_hSocket);
		m_hSocket = NULL;
	}
	if(m_hSocket == NULL)
	{
		//創建新的套接字，這里是創建的流類型的套接字
		m_hSocket = socket(AF_INET, SOCK_STREAM,0);
		ASSERT(m_hSocket != NULL);
		//ServerInit();
		if(WSAAsyncSelect(m_hSocket, m_hWnd, SER_MESSAGE, FD_ACCEPT|FD_READ|FD_WRITE|FD_CLOSE)>0)
			AfxMessageBox("error select");
	}
	m_addr.sin_family = AF_INET;
	m_addr.sin_addr.S_un.S_addr = INADDR_ANY;
	m_addr.sin_port = htons(m_uPort);

	int ret = 0;
	//綁定一個套接字到本機的地址
	ret = bind(m_hSocket, (LPSOCKADDR)&m_addr, sizeof(m_addr));
	if(ret == SOCKET_ERROR)
	{	//綁定錯誤
		AfxMessageBox("綁定出錯!");
		return FALSE;
	}
	
	//開始一個偵聽過程，等待客戶的連接
	ret = listen(m_hSocket, 5);    //這里的第二個參數表示最多支持的客戶連接數
	if(ret == SOCKET_ERROR)
	{	//listen失敗
		AfxMessageBox("Listen Error");
		return FALSE;
	}

	return TRUE;
}


CClient::CClient()
{
	m_hSocket =NULL;
}

CClient::~CClient()
{

}
BOOL CClient::InitAndConnet(HWND hwnd,UINT port,CString strserver)
{
	m_hWnd=hwnd;
	m_uPort=port;
	m_strServer=strserver;
	if(m_hSocket != NULL)
	{
		//如果原來打開這套接字先關閉
		closesocket(m_hSocket);
		m_hSocket = NULL;
	}
	if(m_hSocket == NULL)
	{
		//創建新的流套接字
		m_hSocket = socket(AF_INET, SOCK_STREAM,0);
		ASSERT(m_hSocket != NULL);
		ClientInit();
	}

	//準備服務器的信息，這里需要指定服務器的地址
	m_addr.sin_family = AF_INET;
	m_addr.sin_addr.S_un.S_addr = inet_addr(m_strServer.GetBuffer(0));
	m_addr.sin_port = htons(m_uPort);   //改變端口號的數據格式
	
	//這里主動連接服務器，該過程將等待一定時間
	int ret = 0;
	int error = 0;
	ret = connect(m_hSocket, (LPSOCKADDR)&m_addr, sizeof(m_addr));
	if(ret == SOCKET_ERROR)
	{//連接失敗
		if(GetLastError()!=WSAEWOULDBLOCK)
		{
			AfxMessageBox(_T("請確認服務器確實已經打開并工作在同樣的端口！"));
			return FALSE;
		}
	}
	return TRUE;
}

void CClient::SendString(CString a)
{
	if(send(m_hSocket,a.GetBuffer(0),a.GetLength(),0)==SOCKET_ERROR)
	{
		AfxMessageBox("Client Send data error");
	}
}

void CClient::GetString(CString &str)
{
	recv(m_hSocket,str.GetBuffer(0),1024,MSG_DONTROUTE);
}

void CClient::ClientInit()
{
	if(WSAAsyncSelect(m_hSocket,m_hWnd,CLI_MESSAGE,FD_READ|FD_WRITE|FD_CLOSE|FD_CONNECT)>0)
	{
		AfxMessageBox("Error in select");
	}
}