使用套接字函数设计网络聊天室

#include "winsock2.h"
#pragma comment (lib,"ws2_32.lib")

WSADATA wsd;
WSAStartup(MAKEWORD(2,2),&wsd);

SOCKET m_server,m_client;
SOCKET m_Clients[MAXNUM];      //客户端套接字
int    m_CurClient;            //当前连接的客户数量

//创建套接字
m_server = socket(AF_INET,SOCK_STREAM,0);
//将网络中的事件关联到窗口的消息函数中
WSAAsyncSelect(m_server,m_hWnd,20000,FD_WRITE|FD_READ|FD_ACCEPT);
m_client = 0;
m_serverIP = "";
for (int i = 0; i< MAXNUM;i++)
m_Clients[i]= 0;
m_CurClient = 0;

void CServerDlg::OnOK()
{
//服务器端地址
sockaddr_in serveraddr;serveraddr.sin_family = AF_INET;m_IP.GetWindowText(m_serverIP);//要注意，这里的m_IP是IP控件类型的变量，m_serverIP是CString类型的变量 它们都在类的成员变量中先定义了//设置本机地址
serveraddr.sin_addr.S_un.S_addr  = inet_addr(m_serverIP);UpdateData(TRUE);
//设置端口号
serveraddr.sin_port = htons(atoi(m_port));//m_port也是CString类型的变量//绑定地址
if (bind(m_server,(sockaddr*)&serveraddr,sizeof(serveraddr)))
{
MessageBox("绑定地址失败.");
return;
}
//开始监听
listen(m_server,50);
}

void CServerDlg::HandleData()
{
sockaddr_in serveraddr;
char buffer[1024];
int len =sizeof(serveraddr);

//接收客户端的数据
int curlink = -1;int num = -1;
for (int p = 0;p< MAXNUM; p++)
{
num= recv(m_Clients[p],buffer,1024,0);
if (num != -1)
{
curlink = p;
break;
}
}
buffer[num]= 0;
if (num==-1) //接受客户端的连接
{
if (m_CurClient < MAXNUM)
{
m_Clients[m_CurClient] = accept(m_server,(struct sockaddr*)
&serveraddr,&len);
m_CurClient+=1;
}
return;
}//将接收的数据发送给客户端（这里是程序的核心，服务器端把接收到的数据又发送出去了）
for (int j = 0;j< m_CurClient;j++)
if (j != curlink)
send(m_Clients[j],buffer,num,0);
}

BOOL CServerDlg::PreTranslateMessage(MSG* pMsg)
{
if (pMsg->message==20000)
{
HandleData();
return TRUE;
}
else
return CDialog::PreTranslateMessage(pMsg);
}

#include "winsock2.h"
#pragma comment (lib,"ws2_32.lib")

WSADATA wsd;
WSAStartup(MAKEWORD(2,2),&wsd);

m_client = socket(AF_INET,SOCK_STREAM,0);

void CClientDlg::OnOK()
{
//服务器端地址
sockaddr_in serveraddr;UpdateData(TRUE);
serveraddr.sin_family = AF_INET;
serveraddr.sin_port = htons(m_port);
serveraddr.sin_addr.S_un.S_addr = inet_addr(m_IP);if (connect(m_client,(sockaddr*)&serveraddr,sizeof(serveraddr))!=0)
{
MessageBox("连接失败");
return;
}
else
MessageBox("连接成功");
WSAAsyncSelect(m_client,m_hWnd,1000,FD_READ);
CString str,info ;
m_name.GetWindowText(str);info.Format("%s------>%s",str,"进入聊天室");
int i = send(m_client,info.GetBuffer(0),info.GetLength(),0);
}

void CClientDlg::OnButton1()
{
CString str,name,info ;
m_name.GetWindowText(name);
m_info.GetWindowText(str);
if (!name.IsEmpty()&&!str.IsEmpty())
{
info.Format("%s说: %s",name,str);
//开始发送数据
int i = send(m_client,info.GetBuffer(0),info.GetLength(),0);
m_list.AddString(info);
m_info.SetWindowText("");
}
}

void CClientDlg::ReceiveData(){
char buffer[1024];
//接收服务器端传来的数据,
int num = recv(m_client,buffer,1024,0);
buffer[num] = 0;
//将接收的数据添加到列表框中
m_list.AddString(buffer);
}

BOOL CClientDlg::PreTranslateMessage(MSG* pMsg)
{
if (pMsg->message==1000)
{
ReceiveData();
return TRUE;
}
else
return CDialog::PreTranslateMessage(pMsg);
}