18.4.3 使用WinSock类设计网络聊天室

void CServerDlg::OnOK() 
{
this->UpdateData();
m_pSocket = new CServerSocket(this);
if (!m_pSocket->Create(70))
{
MessageBox("套接字创建失败");
delete m_pSocket;
m_pSocket = NULL;
return;
}
if (!m_pSocket->Listen())
MessageBox("监听失败");
}

void CServerDlg::AcceptConnect()
{
CClientSocket* socket = new CClientSocket(this);
//接受客户端的连接
if (m_pSocket->Accept(*socket))
m_socketlist.AddTail(socket);
else
delete socket;
}

void CServerDlg::ReceiveData(CClientSocket* socket)
{
char bufferdata[BUFFERSIZE];
//接收客户端传来的数据
int result = socket->Receive(bufferdata,BUFFERSIZE);
bufferdata[result] = 0;
POSITION pos = m_socketlist.GetHeadPosition();
//将数据发送给每个客户端
while (pos!=NULL)
{
CClientSocket* socket = (CClientSocket*)m_socketlist.GetNext(pos);
if (socket != NULL)
socket->Send(bufferdata,result);
}
}

//初始化套接字
WSADATA wsd;
AfxSocketInit(&wsd);

void CClientDlg::OnButtonsend() 
{
// TODO: Add your control notification handler code here
CString str,temp;
m_info.GetWindowText(str);
if (str.IsEmpty()|m_name.IsEmpty())
return;
temp.Format("%s说: %s",m_name,str);
int num = pMysocket->Send(temp.GetBuffer(temp.GetLength()),temp.GetLength());
m_info.SetWindowText("");
m_info.SetFocus();
}

void CClientDlg::ReceiveData()
{
char buffer[200];
//接收传来的数据
int factdata =  pMysocket->Receive(buffer,200);

buffer[factdata] = '\0';
CString str;
str.Format("%s",buffer);
int i = m_list.GetCount();
//将数据添加到列表框中
m_list.InsertString(m_list.GetCount(),str);
}

void CClientDlg::OnButtonjoin() 
{
// TODO: Add your control notification handler code here
UpdateData(true);
CString servername = m_servername; //读取服务器名称
int port;
port = 70; //获取端口

if  (! pMysocket->Connect(servername,port)) //连接服务器
{
MessageBox("连接服务器失败!");
return;
}
CString str;
str.Format("%s----->%s",m_name,"进入聊天室");
int num = pMysocket->Send(str.GetBuffer(0),str.GetLength());
}