基于UDP的MFC局域网聊天程序

开发环境VS2010

这个小程序用到了UDP通信和多线程的知识。

已知问题：不能显示中文。发送内容不能过长。滚动条没有自动滚到最后一行。

基本思路：在初始化对话框的时候就创建一个线程，在这个线程里进行套接字初始化，然后一直阻塞到接受到消息。

总结：接受消息显示采用的是往CString 变量里面追加东西，然后再分行显示。这样做个人感觉肯定不是最优，但是我目前只能想到这个办法了。至于不能显示中文的问题，我知道发生在接收消息阶段。整个消息发送接收过程是这样处理的：从控件上获取到CString的，然后转换格式成wchar_t，再格式化成char* 通过sendto函数发送过去。接受端接收了char*也是转成wchar_t再转成CString显示的。这里面格式转换是存在问题的，希望有人能指点。

关键代码存档

在头文件里添加成员变量static DWORD WINAPI ThreadProc1(LPVOID lpParameter); 下面和也是。

View Code

1 /*变量声明*/
2 HANDLE hThread1;
3 HANDLE hThread2;
4 CString m_Edit;  //发送消息的编辑控件的成员变量
5 CIPAddressCtrl m_ip;  //ip控件的成员变量
6 CString m_msg;  // 显示消息控件的成员变量，没用到
7 
8 void OnOK();  /*隐藏默认的OnOK函数，就是按程序运行时下回车时调用的函数，这里我们要重新定义一下*/

View Code

  1 /*在OnInitDialog()中添加额外的初始化代码*/
  2 hThread1 = ::CreateThread(NULL,0,ThreadProc1,this,0,NULL);  /*在初始化对话框的时候就创建线程*/
  3 
  4 
  5 /*点击发送按钮时的代码*/
  6 void CMFCUDPDlg::OnBnClickedButton1()
  7 {
  8     
  9      CSocket socket;
 10      if( !socket.Create(0, SOCK_DGRAM, NULL) )
 11      {
 12         MessageBox(_T("创建套接字失败"));
 13      }
 14      //UpdateData(TRUE);    
 15      
 16      wchar_t *sendMsg;
 17      GetDlgItemTextW(IDC_EDIT2,m_Edit);
 18      
 19      sendMsg=m_Edit.GetBuffer(m_Edit.GetLength()); 
 20      m_Edit.ReleaseBuffer();
 21 
 22      /*把char类型的sendMsg转化为wchar_t类型的WStr。
 23     size_t len = strlen(sendMsg) + 1;
 24     size_t converted = 0;
 25     wchar_t *WStr;
 26     WStr=(wchar_t*)malloc(len*sizeof(wchar_t));
 27     mbstowcs_s(&converted, WStr, len, sendMsg, _TRUNCATE);
 28     */
 29 
 30       /*测试类型转换以后是否有乱码 ,我现在觉得下面这种格式化，还是从wchar_t变成CString方便
 31     m_Edit.Format(_T("%s"),sendMsg); 
 32      AfxMessageBox(m_Edit);
 33     */
 34 
 35      //此时的sendMsg已经变成了wchar_t类型的，接下来要把它变成char*型的
 36       size_t len = wcslen(sendMsg) + 1;
 37       size_t converted = 0;
 38       char *message;
 39       message=(char*)malloc(len*sizeof(char));
 40       wcstombs_s(&converted, message, len, sendMsg, _TRUNCATE);
 41 
 42     /*获取ip空间里的ip地址*/
 43     BYTE f0, f1, f2, f3; 
 44     m_ip.GetAddress(f0, f1, f2, f3); 
 45     CString m_addr; 
 46     m_addr.Format(_T("%d%s%d%s%d%s%d"), f0, ".", f1, ".", f2, ".", f3); 
 47 
 48     int length = socket.SendTo(message, strlen(message), 9000, m_addr, 0);
 49     socket.Close();
 50     
 51 
 52 }
 53 
 54 /*因为线程函数是静态的，不能调用非静态成员，在此创建一个类类型的指针*/
 55 CMFCUDPDlg* p;
 56 /*初始化对话框时，创建线程做的事情*/
 57 DWORD WINAPI CMFCUDPDlg::ThreadProc1(LPVOID lpParameter)
 58 {
 59     
 60         CMFCUDPDlg *p=(CMFCUDPDlg*)lpParameter;
 61         CSocket Server;  
 62 
 63         // 创建用于接收数据报的套接字
 64         if (Server.Create(9000, SOCK_DGRAM, NULL)== 0) 
 65         {
 66             AfxMessageBox(_T("创建Socket失败"));
 67             return -1;
 68         }
 69 
 70     while(TRUE)
 71     {
 72             SOCKADDR_IN ClntAddr; 
 73             int clntAddrLen = sizeof(ClntAddr);
 74             char echoBuffer[1024]=""; 
 75             CString buffer;
 76             static CString save;
 77         
 78             //一直阻塞到有客户端发来数据
 79             int recvMsgSize = Server.ReceiveFrom(echoBuffer,sizeof(echoBuffer), (SOCKADDR*)&ClntAddr, &clntAddrLen, 0);
 80             
 81             if (recvMsgSize < 0)
 82             {
 83               AfxMessageBox(_T("接受数据异常"));
 84             }
 85             else
 86             {
 87                 //AfxMessageBox(_T("有消息"));
 88             }
 89 
 90             //把echoBuffer转格式成wchar_t
 91             size_t len = strlen(echoBuffer) + 1;
 92             size_t converted = 0;
 93             wchar_t *WStr;
 94             WStr=(wchar_t*)malloc(len*sizeof(wchar_t));
 95             mbstowcs_s(&converted, WStr, len, echoBuffer, _TRUNCATE);
 96 
 97             //获取对方ip
 98             char *addr = inet_ntoa(ClntAddr.sin_addr);
 99             //把addr转格式成wchar_t
100             len = strlen(addr) + 1;
101             wchar_t *address;
102             address=(wchar_t*)malloc(len*sizeof(wchar_t));
103             mbstowcs_s(&converted, address, len, addr, _TRUNCATE);
104 
105             /*把发送端的ip地址和消息还有换行都存放在buffer里，然后一直往save里追加*/
106             buffer.Format(_T("【%s】说 :%s\r\n"),address,WStr);
107             save = save+buffer;
108             
109             p->SetDlgItemTextW(IDC_EDIT1,save);
110             p->SetDlgItemTextW(IDC_EDIT2,NULL);
111             //p->m_msg = buffer;
112             //p->UpdateData(false);    
113     }
114     Server.Close();
115 
116     return 0;
117 
118 }

View Code

void CMFCUDPDlg::OnOK()
{
    /*当编辑好发送内容后，直接按回车，就相当于点击了发送按钮*/
    OnBnClickedButton1();
}

提示:如果没有换行，记得将编辑控件的属性里的Multiline属性设置为true，然后记得添加滚动条，需要的话设置成readonly。

感想:我承认里面做了很多烦琐而低效的类型转换。希望有人能指点。化繁为简。