MFC笔记之多线程聊天室

新手刚接触，跟着孙鑫老师视频一步一步的做。从VC6.0到VS2010好像并不是那么顺利，下面记录下一点收获。

网络编程的一般步骤：

1声明套接字版本（WSAStartup）;2创建套接字(socket);3绑定套接字(bind);4发送接收（sendto/recvfrom）;5关闭（closesocket）

第1~3步代码如下：

 1 CString error;
 2 WORD wver;
 3 wver=MAKEWORD(1,1);
 4 WSAData data;
 5 if(0!=WSAStartup(wver,&data))
 6 {
 7 　　error.Format(_T("声明版本失败，错误代码：%d!"),WSAGetLastError());
 8 　　MessageBox(error);
 9 }
10
11 ms=socket(2,SOCK_DGRAM,0);
12 if(ms==INVALID_SOCKET)
13 {
14 　　error.Format(_T("创建套接字失败，错误代码：%d!"),WSAGetLastError());
15 　　MessageBox(error);
16 }
17
18 SOCKADDR_IN msaddr;
19 msaddr.sin_family=AF_INET;
20 msaddr.sin_addr.S_un.S_addr=htonl(INADDR_ANY);
21 msaddr.sin_port=htons(6000);
22 int len=sizeof(msaddr);
23 if(SOCKET_ERROR==bind(ms,(SOCKADDR*)&msaddr,len))
24 {
25 　　error.Format(_T("绑定套接字失败，错误代码：%d!"),WSAGetLastError());
26 　　MessageBox(error);
27 }

这段代码要写到OnInitDialog()里；其中ms定义为的SOCKET型成员变量。

既然要用到多线程，这里就先插入一点多线程的知识。

1 HANDLE hThread1;
2 hThread1=CreateThread(NULL,0,fun1pro,(LPVOID)ppr,0,NULL);
3 CloseHandle(hThread1);

其中fun1pro为新线程的入口函数，ppr为传入新线程的参数结构体指针；参数结构体根据需要自行定义。然后是线程函数的实现，里面具体实现socket的接收显示。并且在VS2010中线程函数返回值不能为空，这里我用的DWORD类型；且最好声明成静态成员函数。

 1 DWORD WINAPI CMy0712Dlg::fun1pro(LPVOID lpParameter)
 2     {
 3         SOCKET sock;
 4         HWND hw;
 5         CString error;
 6         sock=((pr*)lpParameter)->s;
 7         hw=((pr*)lpParameter)->hs;
 8         CString cbuf,tbuf;
 9         TCHAR buf[100]={0};
10         TCHAR temp[120]={0};
11         CString ttt,sss;
12         SOCKADDR_IN oaddr;
13         int len=sizeof(SOCKADDR);
14         while (TRUE)
15         { int a=recvfrom(sock,(char*)&buf,100,0,(SOCKADDR*)&oaddr,&len);
16             if(SOCKET_ERROR==a)
17             {
18                 error.Format(_T("接受失败，错误代码：%d!"),WSAGetLastError());
19                 AfxMessageBox(error);
20                 break;
21             }
22             else if(a==0)
23             {
24                 error.Format(_T("socket已经关闭！"));
25                 AfxMessageBox(error);
26                 break;
27             }
28             else
29             {
30                 wsprintf(temp,_T("He said: %s"),buf);
31                 memset(buf,0,100);
32                 sss=_T("");
33                 if(!::PostMessageA(hw,WM_RECV,0,(LPARAM)temp))
34                 {
35                 error.Format(_T("消息传递失败，错误代码：%d!"),GetLastError());
36                 AfxMessageBox(error);
37                 error.GetLength();
38                 }
39             }
40         }
41                 return 0;
42     }

这里将接收到的数据存在了buf里，然后加上格式头存在temp里，通过PostMessage将temp作为自定义消息WM_RECV的lParam传递出去。

这里牵扯到自定义消息的定义

 #define WM_RECV    WM_USER+1

还有消息映射

ON_MESSAGE(WM_RECV,OnRecv)

消息函数的声明实现

 afx_msg LRESULT OnRecv(WPARAM wParam,LPARAM lParam);

 1 LRESULT CMy0712Dlg::OnRecv(WPARAM wParam,LPARAM lParam)
 2 {
 3
 4     CString cs(((TCHAR*)lParam));
 5     CString temp;
 6     GetDlgItemText(IDC_EDIT1,temp);
 7     int i=temp.GetLength();
 8     if(i!=0)
 9     {temp+="\r\n";}
10     temp+=cs;
11     SetDlgItemText(IDC_EDIT1,temp);
12     return 0;
13 }

发送函数做到了一个按钮的响应函数里

 1 void CMy0712Dlg::OnBnClickedButton2()
 2 {
 3     // TODO: 在此添加控件通知处理程序代码
 4     CString cs,error,temp;
 5     SOCKADDR_IN addr;
 6     addr.sin_family=AF_INET;
 7     addr.sin_addr.S_un.S_addr=inet_addr("127.0.0.1");
 8     addr.sin_port=htons(6000);
 9     int len=sizeof(addr);
10     GetDlgItemText(IDC_EDIT2,cs);
11     int i=sendto(ms,(char*)cs.GetBuffer(cs.GetLength()*2),cs.GetLength()*2+2,0,(SOCKADDR*)&addr,sizeof(SOCKADDR));
12     if(SOCKET_ERROR==i)
13     {
14         error.Format(_T("数据发送失败，错误代码：%d!"),WSAGetLastError());
15         MessageBox(error);
16     }
17     cs=_T("");
18     SetDlgItemText(IDC_EDIT2,cs);
19 }

这就基本完成了所有代码，但是在默认使用Unicode字符集的VS2010中还是有点小问题的。其中sendto发送了发送缓存两倍的空间，具体在代码中体现。

参考资料：http://blog.163.com/lj_2005/blog/static/458654220115662544227/

MFC笔记之多线程聊天室