基于C/S 结构的IM即时通讯软件--上篇

 目的：实现类似QQ群聊的聊天室，可以看到好友列表及互相传送信息。

分析：可基于C/S结构实现即时通讯

1、创建基于对话框的MFC程序（支持windows套接字），并增加相应的类与结构体，完善对话框界面。

        服务器端：新增两个基于CSocket的类CClientSocket/CServerSocket分别用于服务与监听；并新增一个结构体信息，用于对客户端传输的信息（消息类型与长度）进行甄别。

    

        客户端：新增基于CSocket的类CClientSocket用于服务；并新增一个结构体信息，用于对客户端传输的信息（消息类型与长度）进行甄别。

结构体信息，用于标识传输的内容类型

typedef struct tagHeader
{
    char type;
    int len;
}Header, *pHeader;
#define LOGIN_IO    1   //后面可以继续完善，比如视频/音频/文件
#define SEND_MESSAGE 3
;

2、完善框架结构

        服务器：在Clientsocket.h声明Dlg类，便于进行指针操作，并标识登录的用户昵称，创建CPtrList链队列数据结构。

class CCase006Dlg;
class CClientSocket : public CSocket
{
public:
    CString m_strName;        //请求登录的用户
    CPtrList *clist;        //储存队列的结构
    CCase006Dlg* m_dlgserver;    //对话框指针机制
    CClientSocket(CPtrList*list);    //生成一个socket队列

    CCase006Dlg* m_Dlg;
    void GetDlg(CCase011Dlg *dlg);
12　}

         在Clientsocket.cpp中包含结构体头文件，并实现指针函数

#include "tagHeader.h"

// CClientSocket
CClientSocket::CClientSocket(CPtrList *list) : m_dlgserver(NULL)
{
    clist=list;
}

       在CSeverSocket.h中定义列表对象-CPtrList connectList;

       在CSeverSocket.cpp中添加结构体头文件;

       在CXXXXDlg.h中声明套接字类，并定义指针对象与用户列表更新函数

 1 class CServerSocket;    //监听套接字类
 2 class CClientSocket;    //创建与客户端通信的套接字类
 3 
     // CCase006Dlg 对话框
 4 class CCase006Dlg : public CDialog
 5 {
 6 // 构造
 7 public:
 8     CCase006Dlg(CWnd* pParent = NULL);    // 标准构造函数
 9     CString m_str_name;         //用于标识管理员名
10     CClientSocket* clientsocket;    //定义套接字指针
11     CServerSocket* m_serversocket;
12     void UpdateUser(CClientSocket* psocket);   //声明用户更新函数

       在CXXXXDlg.cpp中包含头文件，并在初始化函数中初始化Socket指针对象

#include "ServerSocket.h"
#include "ClientSocket.h"
#include "tagHeader.h"

BOOL CCase006Dlg::OnInitDialog()
{
    serversocket=NULL;
    clientsocket =NULL;
        ......
}

        客户端：在 Clientsocket.h中定义登录用户名，并定义一个窗口指针及函数

1 class CCase005Dlg;   //注意声明类
2 class CClientSocket : public CSocket
3 {
4 public:
5     CCase005Dlg *m_Dlg;
6     CString m_str_name;        //登录用户
7     void GetDlg(CCase005Dlg *dlg);
8 };

        在ClientSocket.cpp中声明结构体，并实现窗口指针获取函数

#include "Case005Dlg.h"
#include "tagHeader.h"

// CClientSocket 成员函数
void CClientSocket::GetDlg(CCase005Dlg *dlg)
{
    m_dlg=dlg;
}

       在CXXXXDlg.h中声明套接字类，并定义指针对象与用户列表更新函数以及是否获取信息函数

#include "ClientSocket.h"

class CCase005Dlg : public CDialog
{
// 构造
public:
    CClientSocket *m_clientsocket;
    void Updateuser();
    BOOL GetmsgFromRoom();
         ......
}

        在CXXXXDlg.cpp中包含头文件，并在初始化函数中初始化Socket指针对象

#include "ClientSocket.h"
#include "tagHeader.h"

        此时已完成基本框架体系结构，接下来实现通信相关函数。

3、实现界面事件函数

    服务器端：单击" 开始" 按钮开始监听

 1 void CCase011Dlg::OnBnClickedBnOpen()
 2 {
 3     // TODO: 在此添加控件通知处理程序代码
 4     UpdateData();
 5     serversocket=new CServerSocket;       //开启服务，定义一个服务对象
 6 
 7     BYTE nfield[4];
 8     CString strIP;
 9     UpdateData();    //需要更新
10 
11     //将IP传给地址框
12     m_edit_ip.GetAddress(nfield[0],nfield[1],nfield[2],nfield[3]);
13     strIP.Format("%d.%d.%d.%d",nfield[0],nfield[1],nfield[2],nfield[3]);
14     CTime time=CTime::GetCurrentTime();
15 
16     if(m_str_name.IsEmpty())                    //对输入信息判定，，可简写
17     {
18         AfxMessageBox("请先登记管理员名！");
19         return ;
20     }
21     if(m_edit_ip.IsBlank())
22     {
23         AfxMessageBox("请配置聊天室IP");
24         return ;
25     }
26     if(m_str_port.IsEmpty())
27     {
28         AfxMessageBox("请配置要开放的端口");
29         return ;
30     }
31     
32     //开启聊天室，首先创建用于监听的套接字
33     if(serversocket->Create(atoi(m_int_port),1,strIP))
34     {
35         m_editbox.SetWindowText("");
36         m_editbox.ReplaceSel("聊天室开启成功！
");     //此处使用的是编辑框控件，所以需使用replaceSel        
           CString temp=time.Format("%Y-%m-%d");          //增加时间排头
37         m_editbox.ReplaceSel("日期："+temp+"
");     //编辑框输出信息
38         temp=time.Format("%H:%M:%S");
39         theApp.m_str_name=m_str_name;        //如果想要实现聊天界面标题为登录者
40         m_editbox.ReplaceSel(temp+" 管理员"+theApp.m_str_name+"开放聊天室
");
41     }
42     if(serversocket->Listen())        //打开监听
43     {
44         m_editbox.ReplaceSel("等待成员加入。。。
");
45     }
46     //目前无人，因此仅有管理员一人在线
47     m_listbox.ResetContent();
48     m_listbox.AddString(theApp.m_str_name+"(管理员)");
49     this->SetWindowTextA("管理员:"+m_str_name+"服务器端");
50 }

       更新用户列表处理函数

void CCase011Dlg::UpdateUser(CClientSocket* psocket)
{
    m_listbox.ResetContent();        //清空列表
    m_listbox.AddString(theApp.m_strName+"(管理员)");    //添加管理员用户

    CString userInfo;
    userInfo = theApp.m_str_ame+"(管理员)"+"&";    //添加管理员信息
    if(clientsocket!=NULL)
    {
        CClientSocket* psock=NULL;
        POSITION pos=clientsocket->clist->GetHeadPosition();    //获取列表头位置
        while(pos!=NULL)
        {
            psock=(CClientSocket*)clientsocket->clist->GetNext(pos);  
            m_listbox.AddString(psock->m_str_name);   
            userInfo+=(psock->m_strName+"&");        //逐个增加已经记录的用户名
        }
        Header head;
        head.type=LOGIN_IO;
        head.len=userInfo.GetLength();
        POSITION po=clientsocket->clist->GetHeadPosition();
        while(po!=NULL)            //将最新的用户列表转发给各个用户
        {
            psock=(CClientSocket*)clientsocket->clist->GetNext(po);
            psock->Send((char*)&head,sizeof(Header));
            psock->Send((LPCTSTR)userInfo,userInfo.GetLength());
        }
    }
}