包括六个UDPTCP发送接受的cpp文件,一个基于MFC的局域网聊天小工具工程,和此小工具的所有运行时库、资源和执行程序。代码的压缩包位置是http://www.blogjava.net/Files/wxb_nudt

2 Socket简介(全文摘抄)


SOCKET实际在计算机中提供了一个通信端口,可以通过这个端口与任何一个具有SOCKET接口的计算机通信。应用程序在网络上传输,接收的信息都通过这个SOCKET接口来实现。在应用开发中就像使用文件句柄一样,可以对SOCKET句柄进行读,写操作。

3 Socket的机制是什么?

Socket理解为一个可以连通网络上不同计算机程序之间的管道

管道的端口由两个因素来唯一确认,即机器的IP地址和程序所使用的端口号。

许多著名的木马程序:在网络上扫描不同的端口号就是为了获取一个可以连通的端口从而进行破坏。比较著名的端口号有http80端口和ftp21端口

Socket可以支持数据的发送和接收,它会定义一种称为套接字的变量,发送数据时首先创建套接字,然后使用该套接字的sendto等方法对准某个IP/端口进行数据发送;接收端也首先创建套接字,然后将该套接字绑定到一个IP/端口上,所有发向此端口的数据会被该套接字的recv等函数读出。如同读出文件中的数据一样。

4 所需的头文件、库文件和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

lDLL文件Ws2_32.dll,通常处于C:"WINDOWS"system32,这个是可以猜到的。

5 UDP

5.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


5.2 SOCKET类型

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

typedef unsigned int    u_int;

typedef u_int           SOCKET;

可知套接字实际上就是一个无符号整型,它将被Socket环境管理和使用。套接字将被创建、设置、用来发送和接收数据,最后会被关闭。

5.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可以读取高位字节。

5.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读取出来。

5.5 WSAStartup函数

WSAStartup函数被用来初始化Socket环境,它的定义如下:

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

其返回值为整型,调用方式为PASCAL(即标准类型,PASCAL等于__stdcall),参数有两个,第一个参数为WORD类型,指明了Socket的版本号,第二个参数为WSADATA类型的指针。

若返回值为0,则初始化成功,若不为0则失败。

5.6 WSACleanup函数

这是Socket环境的退出函数。返回值为0表示成功,SOCKET_ERROR表示失败。

5.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则失败。

5.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,代表存储参数数值变量的长度。

5.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

5.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的。事实上也往往使用这种方法。

5.11 Sleep函数

线程挂起函数,表示线程挂起一段时间。Sleep(1000)表示挂起一秒。定义于WINBASE.H头文件中。WINBASE.H又被包含于WINDOWS.H中,然后WINDOWS.HWINSOCK2.H包含。所以在本例中使用Sleep函数不需要包含其它头文件。

5.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

5.13 WSAGetLastError函数

该函数用来在Socket相关API失败后读取错误码,根据这些错误码可以对照查出错误原因。

5.14 closesocket

关闭套接字,其参数为SOCKET类型。成功返回0,失败返回SOCKET_ERROR

5.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类似。

5.16UDP点对点发送接收程序

广播发送和接收使用并不广泛,一般来说指定发送和接收的IP比较常用。点对点方式的UDP发送和接收与上面的例子非常类似,不同的就是需要指定一个具体的IP地址。并且不需要调用setsockopt设置socket的广播属性。

其具体源代码见UDP_Send_P2P.cppUDP_Recv_P2P.cpp

注意在使用这两个程序时要设为自己所需的IP

6 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的代码,后续例子程序中含有此功能代码。

6.1 bind函数

bind函数用来将一个套接字绑定到一个IP地址。一般只在服务方(即数据发送方)调用,很多函数会隐式的调用bind函数。

6.2 listen函数

从服务方监听客户方的连接。同一个套接字可以多次监听。

6.3 connectaccept函数

connect是客户方连接服务方的函数,而accept是服务方同意客户方连接的函数。这两个配套函数分别在各自的程序中被成功调用后就可以收发数据了。

6.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中的参数含义一样。

7一个局域网聊天工具的编写

掌握了以上关于socket的基本用法,编写一个局域网聊天程序也就变得非常简单,如同设计一个普通的对话框程序一样。

7.1 功能设计

功能设计如下:

1.要能够指定聊天对象的IP和端口(端口可以内部确定);

2.要能够发送消息给指定聊天对象;

3.要能够接收聊天对象的消息;

4.接收消息时要播放声音;

5.接收消息时如果当前对话框不是最前端,要闪动图标;

6.要有托盘图标,可以将对话框收入托盘;

7.2 功能实现

将内部端口设为3456,提供一个IP地址控件来设置聊天对象的IP。该控件必须能够读取IP地址并赋值给内部变量。将地址转换为in_addr类型。

发送消息需要使用一个套接字。

接收消息也需要使用一个套接字,由于发送消息也使用了一个套接字,为了在同一个进程中同时发送和接收消息,需要使用多线程技术,将发送消息的线程设为主线程;而接收消息的线程设为子线程,子线程只负责接收UDP消息,在收到消息后显示到主界面中。

接收消息时播放声音这个功能在子线程中完成,使用sndPlaySound函数,并提供一个wav文件即可

闪动图标这个最白痴的功能需要使用一个Timer,在主对话框类中添加一个OnTimer函数,定时检查当前窗口状态变量是否为假,若为假就每次设置另一个图标。若当前窗口显示到最顶端,则设置为默认图标。

托盘图标功能用网上下载的CtrayIcon类轻松搞定。需要提供一个自定义消息,一个弹出菜单资源。

7.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是为了那个播放声音的功能。

7.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方法。

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

7.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简单接收程序来辅助测试,因为此时还未完成报文接收功能。

7.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方法上,因此主线程向自己发送一条消息消除阻塞,同时改变子线程退出标志保证子线程可以退出。

7.8 设置聊天对象IP

点击“确认对象”按钮时,检测IP地址控件,如果IP地址有效,则将IP地址读入内部属性。这个IP地址作为发送信息的目标地址。

这个设置只能设置发送消息的对象,所有人都可以向本机发送信息,只要他的端口是正确的。

7.9 编译链接和运行

下载压缩包后可以打开VC工程编译链接,若直接运行则可以点击LANTalkExeFile目录中的可执行文件,这个目标包含了运行所需要的所有dll和资源文件。

当然,如果需要可以用InstallShield做一个安装程序,不过看来是没有必要的。



8 使用好的Socket包可以简化开发过程

本文中所有的技术尽量采用最原始的方式来使用。例如多线程使用的是AfxBeginThread,套接字使用了最原始的套接字,并在很多地方直接使用了SDK函数,而尽量避免了MFC等代码框架,这是为了方便他人掌握技术的最基本内涵。

其实在具体的编程中,当然是怎么方便怎么来,Socket和多线程以及界面等功能都有大量方便可用的代码库,复用这些代码库会比自己动手写方便很多。但是,掌握了基本原理再使用这些库,事半功倍!