【网络编程】之七、select聊天室

好久没有用MFC 来写代码了，手都生疏了，悲剧啊，好多API 都去查的  哎~~~

好了 下面把 我们聊天室的代码贴出：

select函数封装：

[cpp]  view plain copy

1. BOOL CserverDlg::Select_Server(SOCKET sock, int nTime /* = 100 */, BOOL bRead /* = TRUE */)  
2. {  
3.     fd_set fdset;  
4.     timeval tv;  
5.     FD_ZERO(&fdset);  
6.     FD_SET(sock, &fdset);  
7.     nTime = nTime > 1000 ? 1000 : nTime;  
8.     tv.tv_sec  = 0;  
9.     tv.tv_usec = nTime;  
10.   
11.     int iRet = 0;  
12.     if ( bRead ) {  
13.         iRet = select(0, &fdset, NULL , NULL, &tv);  
14.     }else{  
15.         iRet = select(0, NULL , &fdset, NULL, &tv);  
16.     }  
17.   
18.     if(iRet <= 0) {  
19.         return FALSE;  
20.     } else if (FD_ISSET(sock, &fdset)){  
21.         return TRUE;  
22.     }  
23.     return FALSE;  
24. }  

下面是线程函数，分别是服务器和客户端的：

[cpp]  view plain copy

1. static UINT __cdecl ThreadProc(LPVOID pParam);  

[cpp]  view plain copy

1. UINT CserverDlg::ThreadProc(LPVOID lpParameter )  
2. {  
3.     ASSERT(lpParameter);  
4.     CserverDlg *lthis = (CserverDlg*)lpParameter;  
5.     sockaddr_in clientAddr = {0};  
6.   
7.     WSADATA wsaData;  
8.     WORD sockVersion = MAKEWORD (2, 0);  
9.     ::WSAStartup(sockVersion, &wsaData);  
10.   
11.     lthis ->m_sSock = INVALID_SOCKET;  
12.     lthis ->m_sSock = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);  
13.     if (lthis ->m_sSock == INVALID_SOCKET)  
14.     {  
15.         AfxMessageBox("error");  
16.         ::WSACleanup();  
17.         return TRUE;  
18.     }  
19.   
20.     sockaddr_in ser;  
21.     ser.sin_family = AF_INET;  
22.     ser.sin_port = htons(8888);  
23.     ser.sin_addr.S_un.S_addr = INADDR_ANY;  
24.   
25.     if (::bind(lthis->m_sSock, (LPSOCKADDR)&ser, sizeof(ser)) == SOCKET_ERROR)  
26.     {  
27.         AfxMessageBox("error");  
28.         WSACleanup();  
29.         return TRUE;  
30.     }  
31.   
32.     if (listen(lthis->m_sSock, 2) == SOCKET_ERROR)  
33.     {  
34.         AfxMessageBox("error");  
35.         WSACleanup();  
36.         return TRUE;  
37.     }  
38.   
39.     int nclientAddr = sizeof(clientAddr);  
40.     lthis ->m_sClient = accept(lthis->m_sSock, (SOCKADDR*)&clientAddr, &nclientAddr);  
41.     if (lthis->m_sClient == INVALID_SOCKET)  
42.     {  
43.         AfxMessageBox("error");  
44.         return TRUE;  
45.     }  
46.   
47.     while(TRUE)  
48.     {  
49.         if (lthis->Select_Server(lthis->m_sClient) == TRUE)  
50.         {  
51.             TCHAR recvMeg[1024] = {0};  
52.   
53.             int re = recv(lthis->m_sClient, (char*)recvMeg, 1024, 0);  
54.             if (re > 0)  
55.             {  
56.                 recvMeg[re] = '\n';  
57.                 lthis->recvMeg(recvMeg);  
58.             }  
59.             else  
60.             {  
61.                 lthis->recvMeg("客户端下线");  
62.                 break;  
63.             }  
64.         }  
65.         Sleep(500);  
66.     }  
67.   
68.     if (lthis->m_sClient != INVALID_SOCKET)  
69.     {  
70.         closesocket(lthis->m_sClient);  
71.     }  
72.     if (lthis->m_sSock != INVALID_SOCKET)  
73.     {  
74.         closesocket(lthis->m_sSock);  
75.     }  
76.   
77.     WSACleanup();  
78.     return TRUE;  
79. }  

[cpp]  view plain copy

1. static UINT __cdecl ThreadProc(LPVOID pParam);  

[cpp]  view plain copy

1. UINT CclientDlg::ThreadProc(LPVOID pParam)  
2. {  
3.     ASSERT(pParam);  
4.     CclientDlg* lthis = (CclientDlg*)pParam;  
5.   
6.   
7.     WSADATA wsaData;  
8.     WORD sockVersion = MAKEWORD(2, 0);  
9.     ::WSAStartup(sockVersion, &wsaData);  
10.   
11.     lthis->m_Sock = INVALID_SOCKET;  
12.     lthis->m_Sock = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);  
13.     if (lthis->m_Sock == INVALID_SOCKET)  
14.     {  
15.         AfxMessageBox("error");  
16.         return TRUE;  
17.     }  
18.   
19.     sockaddr_in serAdd;  
20.     serAdd.sin_family = AF_INET;  
21.     serAdd.sin_port = htons(8888);  
22.     serAdd.sin_addr.S_un.S_addr = inet_addr("127.0.0.1");  
23.   
24.     if (connect(lthis->m_Sock, (sockaddr*)&serAdd, sizeof(serAdd)) == SOCKET_ERROR)  
25.     {  
26.         AfxMessageBox("error");  
27.         return TRUE;  
28.     }  
29.   
30.   
31.     while (TRUE)  
32.     {  
33.   
34.         TCHAR buff[1024] = {0};  
35.         if (lthis->Select_Server(lthis->m_Sock))  
36.         {  
37.             int nrecv;  
38.             nrecv = recv(lthis->m_Sock, buff, 1024, 0);  
39.             if (nrecv > 0)  
40.             {  
41.                 buff[nrecv] = '\n';  
42.                 lthis->recvMeg(buff);  
43.             }  
44.             else  
45.             {  
46.                 lthis->recvMeg("服务器终止");  
47.                 break;  
48.             }  
49.   
50.         }  
51.         Sleep(500);  
52.     }  
53.   
54.     if (lthis->m_Sock != INVALID_SOCKET)  
55.     {  
56.         closesocket(lthis->m_Sock);  
57.     }  
58.   
59.     WSACleanup();  
60.     return TRUE;  
61. }  

发送信息：

[cpp]  view plain copy

1. void CserverDlg::OnBnClickedButsend()  
2. {  
3.     // TODO: 在此添加控件通知处理程序代码  
4.     CString strsend;  
5.     ASSERT(m_sClient != INVALID_SOCKET);  
6.     GetDlgItemText(IDC_SEND, strsend);  
7.     if (!strsend.IsEmpty() && Select_Server(m_sClient, 100, FALSE))  
8.     {  
9.         send(m_sClient, strsend, strlen(strsend), 0);  
10.         SetDlgItemText(IDC_SEND, "");  
11.     }  
12.   
13. }  

完整代码下载

2012/8/28

jofranks 于南昌

版权声明：本文为博主原创文章，未经博主允许不得转载。