Socket編程指南及實例程序

說明:這是爲了學習Socket編程而轉載某個大牛,blog爲MDA之路的日誌。

例子代碼就在我的博客中,包括六個UDPTCP發送接受的cpp文件,一個基於MFC的局域網聊天小工具工程,和此小工具的所有運行時庫、資源和執行程序。代碼的壓縮包位置是http://www.blogjava.net/Files/wxb_nudt/socket_src.rar

1         前言

在一些常用的編程技術中,Socket網絡編程可以說是最簡單的一種。而且Socket編程需要的基礎知識很少,適合初學者學習網絡編程。目前支持網絡傳輸的技術、語言和工具繁多,但是大部分都是基於Socket開發的,雖說這些“高級”的網絡技術屏蔽了大部分底層實現,號稱能極大程度的簡化開發,而事實上如果你沒有一點Socket基礎,要理解和應用這些技術還是很困難的,而且會讓你成爲“半瓢水”。

深有感觸的是當年我學習CORBA的時候,由於當時各方面的基礎薄弱,整整啃了半年書,最終還是一頭霧水。如果現在讓我帶一個人學CORBA,我一定會安排好順序:首先弄清C++語法;然後是VC編譯環境或者nmake的用法;接下來學習一些網絡基礎知識;然後是Socket編程;這些大概要花費34個月。有了這些基礎學習CORBA一週即可弄懂,兩個月就可以基於CORBA進行開發了。

好了,說了半天其實中心思想就一個,Socket很簡單,很好學!如果你會C++或者JAVA,又懂一點點網絡基礎如TCPUDP的機制,那麼你看完本文就可以熟練進行Socket開發了。

2         Socket簡介(全文摘抄)

(本節內容全部抄自網絡,不保證正確性,有興趣的可以看看!)

80年代初,美國政府的高級研究工程機構(ARPA)給加利福尼亞大學Berkeley分校提供了資金,讓他們在UNIX操作系統下實現TCP/IP協議。在這個項目中,研究人員爲TCP/IP網絡通信開發了一個API(應用程序接口)。這個API稱爲Socket接口(套接字)。今天,SOCKET接口是TCP/IP網絡最爲通用的API,也是在INTERNET上進行應用開發最爲通用的API

90年代初,由Microsoft聯合了其他幾家公司共同制定了一套WINDOWS下的網絡編程接口,即WindowsSockets規範。它是BerkeleySockets的重要擴充,主要是增加了一些異步函數,並增加了符合Windows消息驅動特性的網絡事件異步選擇機制。WINDOWSSOCKETS規範是一套開放的、支持多種協議的Windows下的網絡編程接口。從1991年的1.0版到1995年的2.0.8版,經過不斷完善並在IntelMicrosoftSunSGIInformixNovell等公司的全力支持下,已成爲Windows網絡編程的事實上的標準。目前,在實際應用中的WINDOWSSOKCETS規範主要有1.1版和2.0版。兩者的最重要區別是1.1版只支持TCP/IP協議,而2.0版可以支持多協議。2.0版有良好的向後兼容性,任何使用1.1版的源代碼,二進制文件,應用程序都可以不加修改地在2.0規範下使用。

SOCKET實際在計算機中提供了一個通信端口,可以通過這個端口與任何一個具有SOCKET接口的計算機通信。應用程序在網絡上傳輸,接收的信息都通過這個SOCKET接口來實現。在應用開發中就像使用文件句柄一樣,可以對SOCKET句柄進行讀,寫操作。

3         再說兩句

網上很多文章對於Socket的來龍去脈有如教科書一般的精準。但是涉及具體編程技術就往往被VC等集成開發環境所毒害了,把Windows SDKMFCSocket、多線程、DLL以及編譯鏈接等等技術攪合在一起煮成一鍋夾生飯。

既然要學習Socket,就應該用最簡單直白的方式把Socket的幾個使用要點講出來。我認爲程序員最關心的有以下幾點,按照優先級排列如下:

1.         Socket的機制是什麼?

2.         C/C++Socket需要什麼頭文件、庫文件、DLL,它們可以由誰提供,安裝後一般處於系統的哪個文件夾內?

3.         編寫Socket程序需要的編程基礎是什麼?

4.         Socket庫內最重要的幾個函數和數據類型是什麼?

5.         兩個最簡單的例子程序;

6.         一個貼近應用的稍微複雜的Socket應用程序。

我將一一講述這些要點,並給出從簡到繁,從樸素到花哨的所有源代碼以及編譯鏈接的命令。

4         Socket的機制是什麼?

我們可以簡單的把Socket理解爲一個可以連通網絡上不同計算機程序之間的管道,把一堆數據從管道的A端扔進去,則會從管道的B端(也許同時還可以從CDEF……端冒出來)。管道的端口由兩個因素來唯一確認,即機器的IP地址和程序所使用的端口號。IP地址的含義所有人都知道,所謂端口號就是程序員指定的一個數字,許多著名的木馬程序成天在網絡上掃描不同的端口號就是爲了獲取一個可以連通的端口從而進行破壞。比較著名的端口號有http80端口和ftp21端口(我記錯了麼?)。當然,建議大家自己寫程序不要使用太小的端口號,它們一般被系統佔用了,也不要使用一些著名的端口,一般來說使用1000~5000之內的端口比較好。

Socket可以支持數據的發送和接收,它會定義一種稱爲套接字的變量,發送數據時首先創建套接字,然後使用該套接字的sendto等方法對準某個IP/端口進行數據發送;接收端也首先創建套接字,然後將該套接字綁定到一個IP/端口上,所有發向此端口的數據會被該套接字的recv等函數讀出。如同讀出文件中的數據一樣。

5         所需的頭文件、庫文件和DLL

對於目前使用最廣泛的Windows Socket2.0版本,所需的一些文件如下(以安裝了VC6爲例說明其物理位置):

l         頭文件winsock2.h,通常處於C:"Program Files"Microsoft Visual Studio"VC98"INCLUDE;查看該頭文件可知其中又包含了windows.hpshpack4.h頭文件,因此在windows中的一些常用API都可以使用;

l         庫文件Ws2_32.lib,通常處於C:"Program Files"Microsoft Visual Studio"VC98"Lib

l         DLL文件Ws2_32.dll,通常處於C:"WINDOWS"system32,這個是可以猜到的。

6         編寫Socket程序需要的編程基礎

在開始編寫Socket程序之前,需要以下編程基礎:

l         C++語法;

l         一點點windows SDK的基礎,瞭解一些SDK的數據類型與API的調用方式;

l         一點點編譯、鏈接和執行的技術;知道cllink的最常用用法即可。

7         UDP

用最通俗的話講,所謂UDP,就是發送出去就不管的一種網絡協議。因此UDP編程的發送端只管發送就可以了,不用檢查網絡連接狀態。下面用例子來說明怎樣編寫UDP,並會詳細解釋每個API和數據類型。

7.1 UDP廣播發送程序

下面是一個用UDP發送廣播報文的例子。

#include <winsock2.h>

#include <iostream.h>

void main()

{

    SOCKET sock;   //socket套接字

    char szMsg[] = "this is a UDP test package";//被髮送的字段

    //1.啓動SOCKET庫,版本爲2.0

    WORD wVersionRequested;

    WSADATA wsaData;

    int err;  

    wVersionRequested = MAKEWORD( 2, 0 ); 

    err = WSAStartup( wVersionRequested, &wsaData );

    if ( 0 != err ) //檢查Socket初始化是否成功

    {

       cout<<"Socket2.0初始化失敗,Exit!";

       return;

    }

    //檢查Socket庫的版本是否爲2.0

    if (LOBYTE( wsaData.wVersion ) != 2 || HIBYTE( wsaData.wVersion ) != 0 )

    {

       WSACleanup( );

       return;

    }

    //2.創建socket

    sock = socket(

       AF_INET,           //internetwork: UDP, TCP, etc

       SOCK_DGRAM,        //SOCK_DGRAM說明是UDP類型

       0                  //protocol

       );

    if (INVALID_SOCKET == sock ) {

       cout<<"Socket 創建失敗,Exit!";

       return;

    }

    //3.設置該套接字爲廣播類型,

    bool opt = true;

    setsockopt(sock, SOL_SOCKET, SO_BROADCAST, reinterpret_cast<char FAR *>(&opt), sizeof(opt));

    //4.設置發往的地址

    sockaddr_in addrto;            //發往的地址 

    memset(&addrto,0,sizeof(addrto));

    addrto.sin_family = AF_INET;               //地址類型爲internetwork

    addrto.sin_addr.s_addr = INADDR_BROADCAST; //設置ip爲廣播地址

    addrto.sin_port = htons(7861);             //端口號爲7861

    int nlen=sizeof(addrto);

    unsigned int uIndex = 1;

    while(true)

    {

       Sleep(1000); //程序休眠一秒

       //向廣播地址發送消息

       if( sendto(sock, szMsg, strlen(szMsg), 0, (sockaddr*)&addrto,nlen)

           == SOCKET_ERROR )

           cout<<WSAGetLastError()<<endl;

       else

           cout<<uIndex++<<":an UDP package is sended."<<endl;

    }

    if (!closesocket(sock)) //關閉套接字

    {

       WSAGetLastError();

       return;

    }

    if (!WSACleanup())       //關閉Socket

    {

       WSAGetLastError();

       return;

    }  

}

編譯命令:

CL /c UDP_Send_Broadcast.cpp

鏈接命令(注意如果找不到該庫,則要在後面的/LIBPATH參數後加上庫的路徑):

link UDP_Send_Broadcast.obj ws2_32.lib

執行命令:

D:"Code"成品代碼"Socket"socket_src>UDP_Send_Broadcast.exe

1:an UDP package is sended.

2:an UDP package is sended.

3:an UDP package is sended.

4:an UDP package is sended.

^C

下面一一解釋代碼中出現的數據類型與API函數。有耐心的可以仔細看看,沒耐心的依葫蘆畫瓢也可以寫程序了。

7.2 SOCKET類型

SOCKETsocket套接字類型,在WINSOCK2.H中有如下定義:

typedef unsigned int    u_int;

typedef u_int           SOCKET;

可知套接字實際上就是一個無符號整型,它將被Socket環境管理和使用。套接字將被創建、設置、用來發送和接收數據,最後會被關閉。

7.3 WORD類型、MAKEWORDLOBYTEHIBYTE

WORD類型是一個16位的無符號整型,在WTYPES.H中被定義爲:

typedef unsigned short WORD;

其目的是提供兩個字節的存儲,在Socket中這兩個字節可以表示主版本號和副版本號。使用MAKEWORD宏可以給一個WORD類型賦值。例如要表示主版本號2,副版本號0,可以使用以下代碼:

WORD wVersionRequested;

wVersionRequested = MAKEWORD( 2, 0 ); 

注意低位內存存儲主版本號2,高位內存存儲副版本號0,其值爲0x0002。使用宏LOBYTE可以讀取WORD的低位字節,HIBYTE可以讀取高位字節。

7.4 WSADATA類型和LPWSADATA類型

WSADATA類型是一個結構,描述了Socket庫的一些相關信息,其結構定義如下:

typedef struct WSAData {

        WORD                    wVersion;

        WORD                    wHighVersion;

        char                    szDescription[WSADESCRIPTION_LEN+1];

        char                    szSystemStatus[WSASYS_STATUS_LEN+1];

        unsigned short          iMaxSockets;

        unsigned short          iMaxUdpDg;

        char FAR *              lpVendorInfo;

} WSADATA;

typedef WSADATA FAR *LPWSADATA;

值得注意的就是wVersion字段,存儲了Socket的版本類型。LPWSADATAWSADATA的指針類型。它們不用程序員手動填寫,而是通過Socket的初始化函數WSAStartup讀取出來。

7.5 WSAStartup函數

WSAStartup函數被用來初始化Socket環境,它的定義如下:

int PASCAL FAR WSAStartup(WORD wVersionRequired, LPWSADATA lpWSAData);

其返回值爲整型,調用方式爲PASCAL(即標準類型,PASCAL等於__stdcall),參數有兩個,第一個參數爲WORD類型,指明瞭Socket的版本號,第二個參數爲WSADATA類型的指針。

若返回值爲0,則初始化成功,若不爲0則失敗。

7.6 WSACleanup函數

這是Socket環境的退出函數。返回值爲0表示成功,SOCKET_ERROR表示失敗。

7.7 socket函數

socket的創建函數,其定義爲:

SOCKET PASCAL FAR socket (int af, int type, int protocol);

第一個參數爲int af,代表網絡地址族,目前只有一種取值是有效的,即AF_INET,代表internet地址族;

第二個參數爲int type,代表網絡協議類型,SOCK_DGRAM代表UDP協議,SOCK_STREAM代表TCP協議;

第三個參數爲int protocol,指定網絡地址族的特殊協議,目前無用,賦值0即可。

返回值爲SOCKET,若返回INVALID_SOCKET則失敗。

7.8 setsockopt函數

這個函數用來設置Socket的屬性,若不能正確設置socket屬性,則數據的發送和接收會失敗。定義如下:

int PASCAL FAR setsockopt (SOCKET s, int level, int optname,

                           const char FAR * optval, int optlen);

其返回值爲int類型,0代表成功,SOCKET_ERROR代表有錯誤發生。

第一個參數SOCKET s,代表要設置的套接字;

第二個參數int level,代表要設置的屬性所處的層次,層次包含以下取值:SOL_SOCKET代表套接字層次;IPPROTO_TCP代表TCP協議層次,IPPROTO_IP代表IP協議層次(後面兩個我都沒有用過);

第三個參數int optname,代表設置參數的名稱,SO_BROADCAST代表允許發送廣播數據的屬性,其它屬性可參考MSDN

第四個參數const char FAR * optval,代表指向存儲參數數值的指針,注意這裏可能要使用reinterpret_cast類型轉換;

第五個參數int optlen,代表存儲參數數值變量的長度。

7.9 sockaddr_inin_addr類型,inet_addrinet_ntoa函數

sockaddr_in定義了socket發送和接收數據包的地址,定義:

struct sockaddr_in {

        short   sin_family;

        u_short sin_port;

        struct in_addr sin_addr;

        char    sin_zero[8];

};

其中in_addr的定義如下:

struct in_addr {

        union {

                struct { u_char s_b1,s_b2,s_b3,s_b4; } S_un_b;

                struct { u_short s_w1,s_w2; } S_un_w;

                u_long S_addr;

        } S_un;

首先闡述in_addr的含義,很顯然它是一個存儲ip地址的聯合體(忘記union含義的請看c++書),有三種表達方式:

第一種用四個字節來表示IP地址的四個數字;

第二種用兩個雙字節來表示IP地址;

第三種用一個長整型來表示IP地址。

in_addr賦值的一種最簡單方法是使用inet_addr函數,它可以把一個代表IP地址的字符串賦值轉換爲in_addr類型,如

addrto.sin_addr.s_addr=inet_addr("192.168.0.2");

本例子中由於是廣播地址,所以沒有使用這個函數。其反函數是inet_ntoa,可以把一個in_addr類型轉換爲一個字符串。

sockaddr_in的含義比in_addr的含義要廣泛,其各個字段的含義和取值如下:

第一個字段short   sin_family,代表網絡地址族,如前所述,只能取值AF_INET

第二個字段u_short sin_port,代表IP地址端口,由程序員指定;

第三個字段struct in_addr sin_addr,代表IP地址;

第四個字段char    sin_zero[8],很搞笑,是爲了保證sockaddr_inSOCKADDR類型的長度相等而填充進來的字段。

以下代表指明瞭廣播地址,端口號爲7861的一個地址:

    sockaddr_in addrto;            //發往的地址 

    memset(&addrto,0,sizeof(addrto));

    addrto.sin_family = AF_INET;               //地址類型爲internetwork

    addrto.sin_addr.s_addr = INADDR_BROADCAST; //設置ip爲廣播地址

    addrto.sin_port = htons(7861);             //端口號爲7861

7.10           sockaddr類型

sockaddr類型是用來表示Socket地址的類型,同上面的sockaddr_in類型相比,sockaddr的適用範圍更廣,因爲sockaddr_in只適用於TCP/IP地址。Sockaddr的定義如下:

struct sockaddr {

 u_short    sa_family;

 char       sa_data[14];

};  

可知sockaddr16個字節,而sockaddr_in也有16個字節,所以sockaddr_in是可以強制類型轉換爲sockaddr的。事實上也往往使用這種方法。

7.11           Sleep函數

線程掛起函數,表示線程掛起一段時間。Sleep(1000)表示掛起一秒。定義於WINBASE.H頭文件中。WINBASE.H又被包含於WINDOWS.H中,然後WINDOWS.HWINSOCK2.H包含。所以在本例中使用Sleep函數不需要包含其它頭文件。

7.12           sendto函數

Socket中有兩套發送和接收函數,一是sendtorecvfrom;二是sendrecv。前一套在函數參數中要指明地址;而後一套需要先將套接字和一個地址綁定,然後直接發送和接收,不需綁定地址。sendto的定義如下:

int PASCAL FAR sendto (SOCKET s, const char FAR * buf, int len, int flags, const struct sockaddr FAR *to, int tolen);

第一個參數就是套接字;

第二個參數是要傳送的數據指針;

第三個參數是要傳送的數據長度(字節數);

第四個參數是傳送方式的標識,如果不需要特殊要求則可以設置爲0,其它值請參考MSDN

第五個參數是目標地址,注意這裏使用的是sockaddr的指針;

第六個參數是地址的長度;

返回值爲整型,如果成功,則返回發送的字節數,失敗則返回SOCKET_ERROR

7.13           WSAGetLastError函數

該函數用來在Socket相關API失敗後讀取錯誤碼,根據這些錯誤碼可以對照查出錯誤原因。

7.14           closesocket

關閉套接字,其參數爲SOCKET類型。成功返回0,失敗返回SOCKET_ERROR

7.15           小結

總結以上內容,寫一個UDP發送程序的步驟如下:

1.         WSAStartup函數初始化Socket環境;

2.         socket函數創建一個套接字;

3.         setsockopt函數設置套接字的屬性,例如設置爲廣播類型;很多時候該步驟可以省略;

4.         創建一個sockaddr_in,並指定其IP地址和端口號;

5.         sendto函數向指定地址發送數據,這裏的目標地址就是廣播地址;注意這裏不需要綁定,即使綁定了,其地址也會被sendto中的參數覆蓋;若使用send函數則會出錯,因爲send是面向連接的,而UDP是非連接的,只能使用sendto發送數據;

6.         closesocket函數關閉套接字;

7.         WSACleanup函數關閉Socket環境。

那麼,與之類似,一個UDP接收程序的步驟如下,注意接收方一定要bind套接字:

1.         WSAStartup函數初始化Socket環境;

2.         socket函數創建一個套接字;

3.         setsockopt函數設置套接字的屬性,例如設置爲廣播類型;

4.         創建一個sockaddr_in,並指定其IP地址和端口號;

5.         bind函數將套接字與接收的地址綁定起來,然後調用recvfrom函數或者recv接收數據; 注意這裏一定要綁定,因爲接收報文的套接字必須在網絡上有一個綁定的名稱才能保證正確接收數據;

6.         closesocket函數關閉套接字;

7.         WSACleanup函數關閉Socket環境。

廣播接收程序見源程序代碼UDP_Recv_Broadcast.cpp。編譯、鏈接、執行與UDP_Send_Broadcast類似。

7.16           UDP點對點發送接收程序

廣播發送和接收使用並不廣泛,一般來說指定發送和接收的IP比較常用。點對點方式的UDP發送和接收與上面的例子非常類似,不同的就是需要指定一個具體的IP地址。並且不需要調用setsockopt設置socket的廣播屬性。

其具體源代碼見UDP_Send_P2P.cppUDP_Recv_P2P.cpp

注意在使用這兩個程序時要設爲自己所需的IP

8         TCP

TCPUDP最大的不同之處在於TCP是一個面向連接的協議,在進行數據收發之前TCP必須進行連接,並且在收發的時候必須保持該連接。

發送方的步驟如下(省略了Socket環境的初始化、關閉等內容):

1.         socket函數創建一個套接字sock

2.         bindsock綁定到本地地址;

3.         listen偵聽sock套接字;

4.         accept函數接收客戶方的連接,返回客戶方套接字clientSocket

5.         在客戶方套接字clientSocket上使用send發送數據;

6.         closesocket函數關閉套接字sockclientSocket

而接收方的步驟如下:

1.         socket函數創建一個套接字sock

2.         創建一個指向服務方的遠程地址;

3.         connectsock連接到服務方,使用遠程地址;

4.         在套接字上使用recv接收數據;

5.         closesocket函數關閉套接字sock

值得注意的是,在服務方有兩個地址,一個是本地地址myaddr,另一個是目標地址addrto。本地地址myaddr用來和本地套接字sock綁定,目標地址被sock用來accept客戶方套接字clientSocket。這樣sockclientSocket連接成功,這兩個地址也連接上了。在服務方使用clientSocket發送數據,則會從本地地址傳送到目標地址。

在客戶方只有一個地址,即來源地址addrfrom。這個地址被用來connect遠程的服務方套接字,connect成功則本地套接字與遠程的來源地址連接了,因此可以使用該套接字接收遠程數據。其實這時客戶方套接字已經被隱性的綁定了本地地址,所以不需要顯式調用bind函數,即使調用也不會影像結果。

具體源代碼見TCP_Send.cppTCP_Recv.cpp注意將源代碼中的IP地址修改爲符合自己需要的IP。爲了減少代碼複雜性,沒有使用讀取本機IP的代碼,後續例子程序中含有此功能代碼。

8.1 bind函數

bind函數用來將一個套接字綁定到一個IP地址。一般只在服務方(即數據發送方)調用,很多函數會隱式的調用bind函數。

8.2 listen函數

從服務方監聽客戶方的連接。同一個套接字可以多次監聽。

8.3 connectaccept函數

connect是客戶方連接服務方的函數,而accept是服務方同意客戶方連接的函數。這兩個配套函數分別在各自的程序中被成功調用後就可以收發數據了。

8.4 sendrecv函數

sendrecv是用來發送和接收數據的兩個重要函數。send只能在已經連接的狀態下使用,而recv可以面向連接和非連接的狀態下使用。

send的定義如下:

int WSAAPI send(

    SOCKET s,

    const char FAR * buf,

    int len,

    int flags

    );

其參數的含義和sendto中的前四個參數一樣。而recv的定義如下:

int WSAAPI recv(

    SOCKET s,

    char FAR * buf,

    int len,

    int flags

    );

其參數含義與send中的參數含義一樣。

9         一個局域網聊天工具的編寫

掌握了以上關於socket的基本用法,編寫一個局域網聊天程序也就變得非常簡單,如同設計一個普通的對話框程序一樣。

9.1 功能設計

功能設計如下:

1.         要能夠指定聊天對象的IP和端口(端口可以內部確定);

2.         要能夠發送消息給指定聊天對象;

3.         要能夠接收聊天對象的消息;

4.         接收消息時要播放聲音;

5.         接收消息時如果當前對話框不是最前端,要閃動圖標;

6.         要有托盤圖標,可以將對話框收入托盤;

9.2 功能實現

將內部端口設爲3456,提供一個IP地址控件來設置聊天對象的IP。該控件必須能夠讀取IP地址並賦值給內部變量。將地址轉換爲in_addr類型。

發送消息需要使用一個套接字。

接收消息也需要使用一個套接字,由於發送消息也使用了一個套接字,爲了在同一個進程中同時發送和接收消息,需要使用多線程技術,將發送消息的線程設爲主線程;而接收消息的線程設爲子線程,子線程只負責接收UDP消息,在收到消息後顯示到主界面中。

接收消息時播放聲音這個功能在子線程中完成,使用sndPlaySound函數,並提供一個wav文件即可。

閃動圖標這個最白癡的功能需要使用一個Timer,在主對話框類中添加一個OnTimer函數,定時檢查當前窗口狀態變量是否爲假,若爲假就每次設置另一個圖標。若當前窗口顯示到最頂端,則設置爲默認圖標。

托盤圖標功能用網上下載的CtrayIcon類輕鬆搞定。需要提供一個自定義消息,一個彈出菜單資源。

9.3 所需資源

頭文件:winsock2.hMmsystem.h

庫文件:ws2_32.libwinmm.lib

dllWs2_32.dllwinmm.dll

wav文件:recv.wav

圖標:一個主程序圖標IDI_MAIN、四個變化圖標IDI_ICON1~4;

菜單:一個給托盤用的彈出菜單IDR_TRAYICON;

說明,Mmsystem.hwinmm.libwinmm.dll是爲了那個播放聲音的功能。

9.4 托盤功能

托盤屬於界面功能,是變更很少的需求,因此首先完成。

1.         引入TRAYICON.HTRAYICON.cpp兩個類;

2.         CLANTalkDlg類中加入一個CTrayIconm_trayIcon;屬性;

3.         CLANTalkDlg的構造函數中初始化m_trayIconm_trayIcon(IDR_TRAYICON)

4.         添加一個自定義消息WM_MY_TRAY_NOTIFICATION,即在三個地方添加消息定義、消息響應函數、消息映射;

5.         InitDialog方法中調用托盤初始化的兩個函數      m_trayIcon.SetNotificationWnd(this, WM_MY_TRAY_NOTIFICATION);    m_trayIcon.SetIcon(IDI_MAIN);

6.         重寫OnClose方法,添加彈出菜單的OnAppSuspendOnAppOpen以及OnAppAbout方法;

7.         重寫對話框的OnCancel方法。

9.5 動態圖標

動態圖標也是界面相關功能,首先完成。

1.         添加四個HICON變量m_hIcon1,m_hIcon2,m_hIcon3,m_hIcon4

2.         在構造函數中初始化這四個變量m_hIcon1 = AfxGetApp()->LoadIcon(IDI_ICON1);

3.         InitDialog中設置調用SetTimer(1,300,NULL);設置一個timerid1,間隔爲300微秒;

4.         添加一個布爾屬性m_bDynamicIcon,指示目前是否需要動態圖標,並給出一個設置函數SetDynamicIcon

5.         添加一個OnTimer函數,讓每次timer調用時根據m_bDynamicIcon的值修改圖標;

兩個地方是用來設置動態圖標的,一個是當程序收到消息並且程序不在桌面頂端時,這時設置爲動態圖標,在後面的消息接收線程中處理;二是當程序顯示到桌面頂端時,設置爲非動態;

重載OnActivate方法可以完成第二個時刻的要求。當窗口狀態爲WA_ACTIVE或者WA_CLICKACTIVESetDynamicIcon(false),否則設置SetDynamicIcon(true);

9.6 發送UDP報文功能

發送UDP報文只需在主線程中完成,需要以下步驟:

1.         初始化Socket環境,這可以在CLANTalkAppInitInstance中完成,同理關閉Socket環境在ExitInstance中完成;我們可以使用前面的方法,也可以直接調用MFC中的AfxSocketInit函數,這個函數可以確保在程序結束時自動關閉Socket環境;

2.         創建socket,考慮到報錯信息需要彈出對話框,因此不在CLANTalkDlg的構造函數中創建,而是在InitDialog中構建;發送報文的socketm_sendSock

3.         設置目的地址功能,需要一個地址賦值函數setAddress(char* szAddr);可以將一個字符串地址賦值給sockaddr_in形式的地址;在CLANTalkDlg中增加一個sockaddr_in m_addrto;屬性;

4.         讀取文本框中的文字,用sendto發送到對象地址;

5.         清空文本框,在記錄框中添加聊天記錄。

這時可以使用前面的UDP簡單接收程序來輔助測試,因爲此時還未完成報文接收功能。

9.7 接收UDP報文功能

接收UDP報文要考慮幾個問題,第一個是要創建一個子線程,在子線程中接收報文;第二是接收報文和發送報文要有互斥機制,以免衝突;第三是接收到報文要播放聲音;第四是接收報文且當前窗口不在桌面頂端要調用動態圖標功能。

按照以上需求設計步驟如下:

1.         創建接收套接字m_recvSock

2.         利用gethostnamegethostbyname等函數獲取本機IP,並將套接字bind到該地址;

3.         添加一個CwinThread* m_pRecvThread屬性,並在InitDialog中調用AfxBeginThread創建子線程;

4.         編寫子線程運行函數void RecvProcess(LPVOID pParam),這時一個全局函數,爲了方便調用CLANTalkDlg類中的各種變量與方法,將CLANTalkDlg類的指針作爲參數傳入子線程函數,並將RecvProcess設置爲CLANTalkDlg類的友元。

5.         子線程函數中完成以下功能:利用recv接收報文;保存聊天記錄;判斷當前窗口是否在前臺,並修改動態圖標屬性;播放聲音。

6.         用來記錄聊天信息的ClistBoxSort屬性要去掉,否則記錄會按內容排序,很不好看。在RC編輯器中去掉這個屬性即可。

7.         最後要注意,在主線程退出時要保證子線程退出,但此時子線程還阻塞在recv方法上,因此主線程向自己發送一條消息消除阻塞,同時改變子線程退出標誌保證子線程可以退出。

9.8 設置聊天對象IP

點擊“確認對象”按鈕時,檢測IP地址控件,如果IP地址有效,則將IP地址讀入內部屬性。這個IP地址作爲發送信息的目標地址。

這個設置只能設置發送消息的對象,所有人都可以向本機發送信息,只要他的端口是正確的。

9.9 編譯鏈接和運行

下載壓縮包後可以打開VC工程編譯鏈接,若直接運行則可以點擊LANTalkExeFile目錄中的可執行文件,這個目標包含了運行所需要的所有dll和資源文件。

當然,如果需要可以用InstallShield做一個安裝程序,不過看來是沒有必要的。

9.10           小結

這個聊天程序很簡單,但是基本上具有了一個框架,可以有最簡單的聊天功能。要在此基礎上進行擴展幾乎已經沒有什麼技術問題了。

10    使用好的Socket包可以簡化開發過程

本文中所有的技術儘量採用最原始的方式來使用。例如多線程使用的是AfxBeginThread,套接字使用了最原始的套接字,並在很多地方直接使用了SDK函數,而儘量避免了MFC等代碼框架,這是爲了方便他人掌握技術的最基本內涵。

其實在具體的編程中,當然是怎麼方便怎麼來,Socket和多線程以及界面等功能都有大量方便可用的代碼庫,複用這些代碼庫會比自己動手寫方便很多。但是,掌握了基本原理再使用這些庫,事半功倍!


發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章