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

2023-08-03 16:01:28

目的：实现类似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("聊天室开启成功！\r\n");     //此处使用的是编辑框控件，所以需使用replaceSel        
           CString temp=time.Format("%Y-%m-%d");          //增加时间排头

37         m_editbox.ReplaceSel("日期："+temp+"\r\n");     //编辑框输出信息

38         temp=time.Format("%H:%M:%S");

39         theApp.m_str_name=m_str_name;        //如果想要实现聊天界面标题为登录者

40         m_editbox.ReplaceSel(temp+" 管理员"+theApp.m_str_name+"开放聊天室\r\n");

41     }

42     if(serversocket->Listen())        //打开监听

43     {

44         m_editbox.ReplaceSel("等待成员加入。。。\r\n");

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());

        }

    }

}

码农公寓

相关文章