聊天室用户列表功能

实现类似聊天室的那种用户列表的功能，其基本原理是每次用户登陆到聊天室就给服务器发送一个消息，该消息可以是用户的IP地址、主机名或登陆昵称（这里为方便起见用的是主机名），然后服务器收到消息后就将该消息广播到连接到服务器的所有客户端，这样所有的用户就都有了一个包含其他用户消息的列表了。

下面是简单的代码，仅测试这个功能。

服务器端代码：

BOOL CCWINDlg::OnInitDialog() { ............ ............ // TODO: Add extra initialization here m_socket=socket(AF_INET,SOCK_STREAM,0); if(m_socket==INVALID_SOCKET) { MessageBox("创建套接字失败!"); return FALSE; } BOOL Reuse=TRUE; setsockopt(m_socket,SOL_SOCKET,SO_REUSEADDR,(char*)&Reuse,sizeof(BOOL)); SOCKADDR_IN m_addrserv; m_addrserv.sin_addr.S_un.S_addr=inet_addr("127.0.0.1"); m_addrserv.sin_family=AF_INET; m_addrserv.sin_port=htons(6000); int ret; ret=bind(m_socket,(SOCKADDR*)&m_addrserv,sizeof(SOCKADDR)); if(ret==SOCKET_ERROR) { MessageBox("绑定失败!"); return FALSE; } m_sock=socket(AF_INET,SOCK_STREAM,0); listen(m_socket,64); SOCKADDR_IN m_addrclient; int len=sizeof(SOCKADDR); m_sock=accept(m_socket,(SOCKADDR*)&m_addrclient,&len); if(m_sock==INVALID_SOCKET) { MessageBox("接收客户端连接失败!"); return FALSE; } if(SOCKET_ERROR==WSAAsyncSelect(m_sock,m_hWnd,WM_CONNECT,FD_READ)) { MessageBox("注册接受连接事件失败!"); return FALSE; } return TRUE; // return TRUE unless you set the focus to a control }

void CCWINDlg::OnConnect(WPARAM wparam,LPARAM lparam) { switch(LOWORD(lparam)) { //有用户连接就接收用户名并显示 case FD_READ: { char hostbuf[256]; int ret=recv(m_sock,hostbuf,256,0); if(ret==SOCKET_ERROR) { CString str; str.Format("%d,ret=%d",WSAGetLastError(),ret); MessageBox(str); MessageBox("接收用户名失败!"); return ; } hostbuf[ret]='/0'; ::PostMessage(m_hWnd,WM_ADD,(WPARAM)&hostbuf, 0); MessageBox(hostbuf); if(SOCKET_ERROR==WSAAsyncSelect(m_sock,m_hWnd,WM_CONNECT,FD_READ)) { MessageBox("服务器注册读取事件失败!"); return; } } break; } }

void CCWINDlg::OnAdd(WPARAM wparam,LPARAM lparam) { // MessageBox("进入了广播阶段!"); SOCKET s = ::socket(AF_INET, SOCK_DGRAM, 0); // 有效SO_BROADCAST选项 BOOL bBroadcast = TRUE; ::setsockopt(s, SOL_SOCKET, SO_BROADCAST, (char*)&bBroadcast, sizeof(BOOL)); // 设置广播地址，这里的广播端口号（电台）是6000 SOCKADDR_IN bcast; bcast.sin_family = AF_INET; bcast.sin_addr.s_addr = ::inet_addr("255.255.255.255"); bcast.sin_port = htons(6000); char *hostsend=new char[256]; hostsend=(char*)wparam; if(s==INVALID_SOCKET) { MessageBox("创建广播套接字失败!"); return; } int ret=sendto(s,hostsend,strlen(hostsend),0,(SOCKADDR*)&bcast,sizeof(SOCKADDR)); if(ret==SOCKET_ERROR) { MessageBox("广播发送用户名失败!"); return; } }

客户端代码：

BOOL CCWINClientDlg::OnInitDialog() { ............ ............ // TODO: Add extra initialization here m_socket=socket(AF_INET,SOCK_STREAM,0); SOCKADDR_IN m_addrserv; m_addrserv.sin_addr.S_un.S_addr=inet_addr("127.0.0.1"); m_addrserv.sin_family=AF_INET; m_addrserv.sin_port=htons(6000); if(SOCKET_ERROR==connect(m_socket,(SOCKADDR*)&m_addrserv,sizeof(SOCKADDR))) { MessageBox("连接失败!"); return FALSE; } //MessageBox("AS"); char host[256]; gethostname(host,256); int ret; ret=send(m_socket,host,strlen(host),0); if(ret==0) { MessageBox("客户端发送用户名失败!"); return FALSE; } SOCKET s = ::socket(AF_INET, SOCK_DGRAM, 0); // 首先要绑定一个本地地址，指明广播端口号 SOCKADDR_IN sin; sin.sin_family = AF_INET; sin.sin_addr.S_un.S_addr = INADDR_ANY; sin.sin_port = ::ntohs(6000); if(::bind(s, (sockaddr*)&sin, sizeof(sin)) == SOCKET_ERROR) { MessageBox(" bind() failed /n"); return FALSE; } // 接收广播 RECVPARAM *precv=new RECVPARAM; precv->h_sock=s; precv->h_wnd=m_hWnd; HANDLE hthread=CreateThread(NULL,0,RecvProc,(LPVOID)precv,0,NULL); CloseHandle(hthread); return TRUE; // return TRUE unless you set the focus to a control }

DWORD WINAPI CCWINClientDlg::RecvProc(LPVOID lparam) { SOCKET s=((RECVPARAM*)lparam)->h_sock; HWND h=((RECVPARAM*)lparam)->h_wnd; SOCKADDR_IN addrRemote; int nLen = sizeof(addrRemote); char sz[256]; int nRet = ::recvfrom(s, sz, 256, 0, (sockaddr*)&addrRemote, &nLen); if(nRet > 0) { sz[nRet] = '/0'; //AfxMessageBox(sz); ::SendMessage(h,WM_ADD,(WPARAM)&sz,0); } return 1; }

void CCWINClientDlg::OnADD(WPARAM wparam,LPARAM lparam) { char *host=new char[256]; strcpy(host,(char*)wparam); ((CComboBox*)GetDlgItem(IDC_COMBO1))->AddString(host); }

由于是初学网络编程，以上代码写的有点乱七八糟。而且还只实现了一个客户端的情况。后面学习中我将尝试多个客户端情况的网络通信程序设计。以及在多客户端情况下当某用户断开连接时去除用户名的功能。

今天我又学习了select模型，这个模型可以在一个线程里处理多个套接字（好像最多是64个），于是乎我就把昨天写的那程序改了下，把服务器端的代码改用select的处理方式，然后客户端的代码基本差不多，结果能达到预想的处理多个用户连接服务器并向组合框里添加多个用户名的要求。线将改进后的代码贴出来：

void CSelectServDlg::OnStart() { // TODO: Add your control notification handler code here sListen=socket(AF_INET,SOCK_STREAM,0); SOCKADDR_IN m_addrserv; m_addrserv.sin_addr.S_un.S_addr=INADDR_ANY; m_addrserv.sin_family=AF_INET; m_addrserv.sin_port=htons(6000); if(bind(sListen,(SOCKADDR*)&m_addrserv,sizeof(SOCKADDR))==SOCKET_ERROR) { MessageBox("绑定失败!"); return ; } listen(sListen,5); //select模型处理过程 fd_set fdSocket; FD_ZERO(&fdSocket); FD_SET(sListen,&fdSocket); while(1) { fd_set fdRead=fdSocket; int ret=select(0,&fdRead,NULL,NULL,NULL); if(ret>0) { for(int i=0;i<(int)fdSocket.fd_count;i++) { if(FD_ISSET(fdSocket.fd_array[i],&fdSocket)) { if(fdSocket.fd_array[i]==sListen) { if(fdSocket.fd_count<FD_SETSIZE) { SOCKADDR_IN m_addrclient; int len=sizeof(SOCKADDR); SOCKET sNew=accept(sListen,(SOCKADDR*)&m_addrclient,&len); FD_SET(sNew,&fdSocket); CString str; str.Format("有新用户%s连接到服务器",inet_ntoa(m_addrclient.sin_addr)); MessageBox(str); } else { MessageBox("连接的用户数量太多！"); } } else { char host[256]; int ret=recv(fdSocket.fd_array[i],host,256,0); if(ret==SOCKET_ERROR) { MessageBox("接收用户名失败!"); return; } host[ret]='/0'; //下面开始向客户端发广播了 SOCKET s = ::socket(AF_INET, SOCK_DGRAM, 0); // 有效SO_BROADCAST选项 BOOL bBroadcast = TRUE; ::setsockopt(s, SOL_SOCKET, SO_BROADCAST, (char*)&bBroadcast, sizeof(BOOL)); // 设置广播地址，这里的广播端口号（电台）是6000 SOCKADDR_IN bcast; bcast.sin_family = AF_INET; bcast.sin_addr.s_addr = INADDR_BROADCAST; // ::inet_addr("255.255.255.255"); bcast.sin_port = htons(6000); if(SOCKET_ERROR==sendto(s,host,strlen(host),0,(SOCKADDR*)&bcast,sizeof(SOCKADDR))) { MessageBox("广播用户名失败!"); return; } else { closesocket(fdSocket.fd_array[i]); FD_CLR(fdSocket.fd_array[i],&fdSocket); } } } } } else { MessageBox("select失败"); return ; } } }

 这是在一个按钮消息函数里实现的，每次打开服务器后程序都会有点卡，不知道是不是在不停的循环处理套接字的缘故