学步园

最近做一个关于服务器和客户端互相通信的程序，题目有两个要求：1.服务器只能接收一个客户端的连接，当有第二个客户端连接时就会失败；2.当已经连接的客户端断开后，其他的客户端就可以去连接服务器了。

这样的要求看似很简单，但做的过程中我才发现，客户端程序很好写，可服务器端有点麻烦了。首先只能接收一个客户端连接的服务器程序，这个很另类，网上找了大多都是研究怎么让服务器可以同时与多个客户端打交道，于是只能自己慢慢搞了。

实践过程中，我发现，只要服务器端在调用了accept函数之后把用于listen用的socket给关闭掉，这样就能让第二个客户端连接时返回SOCKET_ERROR。第一个要求算是完成了，下面是第二个要求。要完成第二个要求首先就得能知道客户端断开连接了，然后再重新listen并accept，重复第一次的步骤。这个步骤说来也简单，但到底选用哪个模型来实现呢，细节问题怎么处理呢？可是苦了我这个菜鸟。还好最后找到方法了，把代码贴到这里保存下

void CTCPservDlg::OnCstart() //开启服务器初始化的部分<br /> {<br /> // TODO: Add your control notification handler code here<br /> // 创建套节字<br /> sListen = ::socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);<br /> if(sListen == INVALID_SOCKET)<br /> {<br /> MessageBox("Failed socket() /n");<br /> return ;<br /> }</p> <p> // 填充sockaddr_in结构<br /> sockaddr_in sin;<br /> sin.sin_family = AF_INET;<br /> sin.sin_port = htons(6000);<br /> sin.sin_addr.S_un.S_addr = INADDR_ANY;;</p> <p> BOOL reuse=TRUE;<br /> setsockopt(sListen,SOL_SOCKET,SO_REUSEADDR,(char*)&reuse,sizeof(BOOL));</p> <p> // 绑定这个套节字到一个本地地址<br /> if(::bind(sListen, (LPSOCKADDR)&sin, sizeof(sin)) == SOCKET_ERROR)<br /> {<br /> MessageBox("Failed bind() /n");<br /> return ;<br /> }<br /> // 将套接字设为窗口通知消息类型。<br /> ::WSAAsyncSelect(sListen, m_hWnd, WM_SOCKET, FD_ACCEPT|FD_CLOSE);</p> <p> // 进入监听模式<br /> if(::listen(sListen, 2) == SOCKET_ERROR)<br /> {<br /> MessageBox("Failed listen() /n");<br /> return ;<br /> }<br /> GetDlgItem(IDCSTART)->EnableWindow(FALSE);<br /> // MessageBox("服务器成功开启,等待一个客户接入!");<br /> }</p> <p>void CTCPservDlg::OnSock(WPARAM wparam,LPARAM lparam)<br /> {<br /> switch(LOWORD(lparam))<br /> {<br /> case FD_ACCEPT: // 监听中的套接字检测到有连接进入<br /> {<br /> SOCKET client = ::accept(sListen, NULL, NULL);<br /> MessageBox("接收到一个连接");<br /> ::closesocket(sListen);<br /> ::WSAAsyncSelect(client, m_hWnd, WM_SOCKET, FD_READ|FD_WRITE|FD_CLOSE);<br /> sSend=client;<br /> }<br /> break;<br /> case FD_WRITE:<br /> {<br /> CString s;<br /> GetDlgItemText(IDC_SEND,s);<br /> send(sSend,s.GetBuffer(0),s.GetLength(),0);<br /> }<br /> break;<br /> case FD_READ:<br /> {<br /> char szText[1024] = { 0 };<br /> if(::recv(sListen, szText, 1024, 0) == -1)<br /> ::closesocket(sListen);<br /> else<br /> MessageBox(szText);<br /> }<br /> break;<br /> case FD_CLOSE:<br /> {<br /> // MessageBox("对方关闭连接");<br /> OnCstart();<br /> // ::closesocket(sListen);<br /> }<br /> break;<br /> }</p> <p>}