serverclient.cpp

使用VC++编写的基于WinSock的聊天室程序

// 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");
	}
}