3.8 一個使用CSocket類的網絡通信實例

本例采用CSocket編程模型建立一個聊天程序的簡單實例。建立項目時注意選上“Windows套接字”復選框。

3.8.1 服務器端應用程序設計(ServerDemo)

1）界面

各控件屬性如下

列表框控件IDC_Log屬性“sort”值為false,屬性“Horizontal Scroll”值為true。

按鈕IDC_Send屬性Disable值為true。

2）定義CSocket類的派生類CServSocket和CRecvSocket。

    從CSocket編程模型知道，服務器端需要兩種套接字，一個用來偵聽連接請求，一個用來與請求連接的套接字建立連接。因此，為程序添加兩個CSocket派生類：SServSocket和CRecvSocket，它們與對話框類密切配合，共同完成程序所要求實現的功能。

3）建立套接字與對話框類的關聯

在程序中，對話框類要用到套接字類，而套接字類在響應某些消息，如在函數OnAccept、OnReceive中進行處理時，也要改變對話框的某些控件狀態，以反映給用戶這些事情的發生。

這里存在著兩個類相互使用的情況，把套接字類對象定義成對話框類的成員變量，同時在套接字類中也把對話框類定義為成員變量。如何實現這樣的用法呢？在對話框類頭文件中加入套接字頭文件的聲明，然后在套接字類頭文件中加入對話框類頭文件的聲明，這樣的做法顯然行不通。

具體做法應該如下：

首先，在ServerDemoDlg.h中加入套接字類頭文件的聲明，語句#pragma once的后面加入如下語句：

#include "ServSocket.h"

#include "RecvSocket.h"

然后在該文件中為CServerDemoDlg類增加兩個公有成員變量，語句如下：

CServSocket *ServSock;

CRecvSocket *RecvSock;

這樣在對話框類中就可以使用套接字類了。

繼續在套接字類中加入對話框類信息。

首先，在ServSocket.h文件的開頭，語句#pragma once的后面加入如下語句：

class CServerDemoDlg;

然后，在該文件中為CServSocket類添加一個公有成員變量和一個構造函數：

CServSocket(CServerDemoDlg *Dlg);

CServerDemoDlg *m_Dlg;

接著在ServSocket.cpp文件中添加新的構造函數的實現，并添加一條關于ServerDemoDlg.h文件的預編譯聲明，代碼如下：

#include “ServerDemoDlg.h”

CServSocket::CServSocket(CServerDemoDlg *Dlg)

{

       m_Dlg=Dlg;

}

這樣，在套接字類中也可以通過成員變量使用對話框了。

使用同樣的方法對CRecvSocket類進行設置，使其也可以通過成員變量使用對話框。

4）為套接字添加串行化讀寫信息的功能。在服務器端的兩個套接字中，只有CRecvSocket套接字是真正與客戶端套接字建立連接，發送與接收數據的，因此，我們只為該類添加串行化讀寫信息功能。在RecvSocket.h文件中為類CRecvSocket添加三個公有成員變量。

    CSocketFile *m_File;

       CArchive *m_ArIn;

       CArchive *m_ArOut;

5）在對話框中初始化套接字并偵聽連接請求。在OnInitDialog函數中添加如下代碼：

    // TODO: 在此添加額外的初始化代碼

     if(ServSock=new CServSocket(this))

     {

         if(ServSock->Create (9547))

         {

              m_LogCtrl.AddString ("等待連接......");

              ServSock->Listen ();

         }

         else

         {

              m_LogCtrl.AddString ("初始化失敗，請重新啟動程序！");

              delete ServSock;

         }

     }

     else

     {

         m_LogCtrl.AddString ("初始化失敗，請重新啟動程序！");

     }

上述代碼主要是創建并初始化ServSock套接字，并開始偵聽連接請求。

6）接受連接請求。由于是CServSocket類的ServSock對象在偵聽連接請求，因此由該類來接受連接請求。

首先，在ServSocket.h文件中加入如下語句：

#iinclude “RecvSocket.h”

然后，重載該類的OnAccept函數，在該函數中添加如下代碼：

CRecvSocket *tempSock;

     if(tempSock=new CRecvSocket(this->m_Dlg ))

     {

         if(Accept(*tempSock))

         {

              tempSock->m_File =new CSocketFile(tempSock);

              tempSock->m_ArIn =new CArchive(tempSock->m_File ,CArchive::load );

              tempSock->m_ArOut =new CArchive(tempSock->m_File ,CArchive::store );

              m_Dlg->RecvSock =tempSock;

              tempSock=NULL;

              m_Dlg->m_LogCtrl .AddString ("連接成功,可以開始傳遞消息");

              m_Dlg->m_SendCtrl.EnableWindow (true);

         }

         else

         {

              m_Dlg->m_LogCtrl .AddString ("客戶端當前的連接嘗試失敗");

              delete tempSock;

         }

     }

     else

     {

         m_Dlg->m_LogCtrl .AddString ("連接套接字初始化失敗");

     }

上述代碼首先調用Accept函數接受連接請求，然后為該連接創建一個CRecvSocket類型的套接字，并為該套接字關聯CArchive對象，使其能實現串行化傳輸信息的功能。最后把關聯好的套接字傳回給對話框對象供其使用。這樣，對話框對象的成員變量RecvSock套接字便與客戶端套接字之間建立了一條信息通道，信息將在兩個套接字之間傳遞。

7）接收信息，連接建立成功后，當有信息到達服務器端時，就會引發RecvSock套接字對象的OnReceive函數，因此需要重載CRecvSocket類的OnReceive函數。添加代碼如下：

CString str;

(*m_ArIn)>>str;

m_Dlg->m_LogCtrl .AddString ("對方發來的信息如下:");

    m_Dlg->m_LogCtrl .AddString (str);

m_Dlg->m_LogCtrl .SetCurSel (m_Dlg->m_LogCtrl .GetCount() - 1);

8）發送信息。為對話框“發送”按鈕添加事件處理函數OnBnClickedSend()，代碼如下：

void CServerDemoDlg::OnBnClickedSend()

{

     // TODO: 在此添加控件通知處理程序代碼

     CString str;

     m_MsgCtrl.GetWindowText (str);

     if(str.GetLength ()==0)

         AfxMessageBox("空信息,所以不發出");

     else

     {

         m_LogCtrl.AddString ("你發出的信息如下:");

         m_LogCtrl.AddString (str);

         m_LogCtrl.SetCurSel (m_LogCtrl.GetCount ()-1);

          *(RecvSock->m_ArOut )<<str;

         RecvSock->m_ArOut ->Flush ();

     }

}

3.8.2 客戶端應用程序設計(項目名稱ClientDemo)

1）界面

各控件屬性如下

列表框控件IDC_Log屬性“sort”值為false,屬性“Horizontal Scroll”值為true。

按鈕IDC_Send屬性Disable值為true。

2）創建套接字類(從CSocket類派生)。客戶端只需要一個套接字，命名為CClientSocket。

3）建立對話框類與套接字類的關聯。

首先，在ClientDemoDlg.h文件的開頭，語句#pragma once后面加入如下語句：

#include “ClientSocket.h”

然后，在該文件中為CClientDemoDlg類添加一個公有成員變量，語句如下：

CClientSocket *ClientSock;

接著，在ClientSocket.h文件的開頭，語句#pragma once后面加入如下語句：

class CClientDemoDlg;

然后，在該文件中為CClientSocket類添加一公有成員變量和一個構造函數，語句如下：

CClientSocket(CClientDemoDlg *Dlg);

CClientDemoDlg *m_Dlg;

接著，在ClientSocket.cpp文件中添加新的構造函數的實現代碼，并添加一條關于CClientDemoDlg.h文件的預編譯聲明，代碼如下：

#include "ClientDemoDlg.h"

CClientSocket::CClientSocket(CClientDemoDlg *Dlg)

{

         m_Dlg=Dlg;

}

這樣，便完成了對話框和套接字之間的連接了。

4）為套接字添加串行化讀寫信息的功能。在ClientSocket.h文件中，為類CClientSocket添加三個公有成員變量，代碼如下：

CSocketFile *m_File;

CArchive *m_ArIn;

CArchive *m_ArOut;

5）在對話框中初始化套接字并建立連接

在對話框類的OnInitDialog函數中添加如下代碼

// TODO: 在此添加額外的初始化代碼

     m_LogCtrl.AddString ("正在連接......");

     if(ClientSock=new CClientSocket(this))

     {

         if(ClientSock->Create())

         {

              if(ClientSock->Connect ("localhost",9547))

              {

                   ClientSock->m_File =new CSocketFile(ClientSock);

                   ClientSock->m_ArIn =new CArchive(ClientSock->m_File ,CArchive::load );

                   ClientSock->m_ArOut =new CArchive(ClientSock->m_File,CArchive::store );

                   m_LogCtrl.AddString ("連接成功,可以開始傳遞消息");

                   m_SendCtrl.EnableWindow (true);

              }

              else

              {

                   m_LogCtrl.AddString ("連接不成功");

                   delete ClientSock;

              }

         }

         else

         {

              m_LogCtrl.AddString ("初始化失敗,請重新啟動程序");

              delete ClientSock;

         }

     }

     else

     {

         m_LogCtrl.AddString ("初始化失敗,請重新啟動程序");

     }

6）接收消息。消息到來時，會引發套接字的OnReceive消息，因此要重載CClientSocket類的OnReceive函數，在其中添加代碼如下

// TODO: 在此添加專用代碼和/或調用基類

    CString str;

     m_Dlg->m_LogCtrl .AddString ("對方發來消息如下:");

     *m_ArIn>>str;

     m_Dlg->m_LogCtrl .AddString (str);

     m_Dlg->m_LogCtrl .SetCurSel (m_Dlg->m_LogCtrl .GetCount ()-1);

7）發送信息。為對話框“發送”按鈕添加事件處理函數OnBnClickedSend()，代碼如下：

    void CClientDemoDlg::OnBnClickedSend()

{

     // TODO: 在此添加控件通知處理程序代碼

     CString str;

     m_MsgCtrl.GetWindowText (str);

     if(str.GetLength ()==0)

         AfxMessageBox("空信息,所以不發出");

     else

     {

         m_LogCtrl.AddString ("你發的信息如下:");

         m_LogCtrl.AddString (str);

         m_LogCtrl.SetCurSel (m_LogCtrl.GetCount ()-1);

         *(ClientSock->m_ArOut )<<str;

         ClientSock->m_ArOut ->Flush ();

     }

}