Winsock 的I/O操作

两种I/O模式

  • 阻塞模式:执行I/O操作完成前会一直进行等待,不会将控制权交给程序。套接字 默认为阻塞模式。可以通过多线程技术进行处理。
  • 非阻塞模式:执行I/O操作时,Winsock函数会返回并交出控制权。这种模式使用 起来比较复杂,因为函数在没有运行完成就进行返回,会不断地返回 WSAEWOULDBLOCK错误。但功能强大。

Windows Socket五种I/O模型

如果你想在Windows平台上构建服务器应用,那么I/O模型是你必须考虑的。Windows操作系统提供了选择(Select)、 异步选择(WSAAsyncSelect)、事件选择(WSAEventSelect)、重叠I/O(Overlapped I/O)和完成端口(Completion Port)共五种I/O模型。每一种模型均适用于一种特定的应用场景。程序员应该对自己的应用需求非常明确,而且综合考虑到程序的扩展性和可移植性等因素,作出自己的选择。

我们假设客户端的代码如下(为代码直观,省去所有错误检查,以下同):

 1#include <WINSOCK2.H>
 2#include <stdio.h>
 3
 4#define SERVER_ADDRESS "137.117.2.148"
 5#define PORT           5150
 6#define MSGSIZE        1024
 7
 8#pragma comment(lib, "ws2_32.lib")
 9
10int main()
11{
12  WSADATA     wsaData;
13  SOCKET      sClient;
14  SOCKADDR_IN server;
15  char        szMessage[MSGSIZE];
16  int         ret;
17  
18  // Initialize Windows socket library
19  WSAStartup(0x0202, &wsaData);
20
21  // Create client socket
22  sClient = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
23
24  // Connect to server
25  memset(&server, 0, sizeof(SOCKADDR_IN));
26  server.sin_family = AF_INET;
27  server.sin_addr.S_un.S_addr = inet_addr(SERVER_ADDRESS);
28  server.sin_port = htons(PORT);
29
30  connect(sClient, (struct sockaddr *)&server, sizeof(SOCKADDR_IN));
31
32  while (TRUE)
33  {
34    printf("Send:");
35    gets(szMessage);
36
37    // Send message
38    send(sClient, szMessage, strlen(szMessage), 0);
39
40    // Receive message
41    ret = recv(sClient, szMessage, MSGSIZE, 0);
42    szMessage[ret] = '\0';
43
44    printf("Received [%d bytes]: '%s'\n", ret, szMessage);
45  }
46
47  // Clean up
48  closesocket(sClient);
49  WSACleanup();
50  return 0;
51}

客户端所做的事情相当简单,创建套接字,连接服务器,然后不停的发送和接收数据。

比较容易想到的一种服务器模型就是采用一个主线程,负责监听客户端的连接请求,当接收到某个客户端的连接请求后, 创建一个专门用于和该客户端通信的套接字和一个辅助线程。以后该客户端和服务器的交互都在这个辅助线程内完成。这种方法比较直观,程序非常简单而且可移植性好,但是不能利用平台相关的特性。例如,如果连接数增多的时候(成千上万的连接),那么线程数成倍增长,操作系统忙于频繁的线程间切换,而且大部分线程在其生命周期内都是处于非活动状态的,这大大浪费了系统的资源。所以,如果你已经知道你的代码只会运行在Windows平台上,建议采用Winsock I/O模型。

一.选择模型

Select(选择)模型是Winsock中最常见的I/O模型。之所以称其为“Select模型”,是由于它的“中心思想”便是利用select函 数,实现对I/O的管理。最初设计该模型时,主要面向的是某些使用UNIX操作系统的计算机,它们采用的是Berkeley套接字方案。 Select模型已集成到Winsock 1.1中,它使那些想避免在套接字调用过程中被无辜“锁定”的应用程序,采取一种有序的方式,同时进行对多个套接字的管理。由于Winsock 1.1向后兼容于Berkeley套接字实施方案,所以假如有一个Berkeley套接字应用使用了select函数,那么从理论角度讲,毋需对其进行任何修改,便可正常运行。(节选自《Windows网络编程》第八章)

下面的这段程序就是利用选择模型实现的Echo服务器的代码(已经不能再精简了):

  1#include <winsock.h>
  2#include <stdio.h>
  3
  4#define PORT       5150
  5#define MSGSIZE    1024
  6
  7#pragma comment(lib, "ws2_32.lib")
  8
  9int    g_iTotalConn = 0;
 10SOCKET g_CliSocketArr[FD_SETSIZE];
 11
 12DWORD WINAPI WorkerThread(LPVOID lpParameter);
 13
 14int main()
 15{
 16  WSADATA     wsaData;
 17  SOCKET      sListen, sClient;
 18  SOCKADDR_IN local, client;
 19  int         iaddrSize = sizeof(SOCKADDR_IN);
 20  DWORD       dwThreadId;
 21
 22  // Initialize Windows socket library
 23  WSAStartup(0x0202, &wsaData);
 24
 25  // Create listening socket
 26  sListen = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
 27
 28  // Bind
 29  local.sin_addr.S_un.S_addr = htonl(INADDR_ANY);
 30  local.sin_family = AF_INET;
 31  local.sin_port = htons(PORT);
 32  bind(sListen, (struct sockaddr *)&local, sizeof(SOCKADDR_IN));
 33
 34  // Listen
 35  listen(sListen, 3);
 36
 37  // Create worker thread
 38  CreateThread(NULL, 0, WorkerThread, NULL, 0, &dwThreadId);  
 39
 40  while (TRUE)
 41  {
 42    // Accept a connection
 43    sClient = accept(sListen, (struct sockaddr *)&client, &iaddrSize);
 44    printf("Accepted client:%s:%d\n", inet_ntoa(client.sin_addr), ntohs(client.sin_port));
 45
 46    // Add socket to g_CliSocketArr
 47    g_CliSocketArr[g_iTotalConn++] = sClient;
 48  }
 49  
 50  return 0;
 51}
 52
 53DWORD WINAPI WorkerThread(LPVOID lpParam)
 54{
 55  int            i;
 56  fd_set         fdread;
 57  int            ret;
 58  struct timeval tv = {1, 0};
 59  char           szMessage[MSGSIZE];
 60  
 61  while (TRUE)
 62  {
 63    FD_ZERO(&fdread);
 64    for (i = 0; i < g_iTotalConn; i++)
 65    {
 66      FD_SET(g_CliSocketArr, &fdread);
 67    }
 68
 69    // We only care read event
 70    ret = select(0, &fdread, NULL, NULL, &tv);
 71
 72    if (ret == 0)
 73    {
 74      // Time expired
 75      continue;
 76    }
 77
 78    for (i = 0; i < g_iTotalConn; i++)
 79    {
 80      if (FD_ISSET(g_CliSocketArr, &fdread))
 81      {
 82        // A read event happened on g_CliSocketArr
 83        ret = recv(g_CliSocketArr, szMessage, MSGSIZE, 0);
 84	    if (ret == 0 || (ret == SOCKET_ERROR && WSAGetLastError() == WSAECONNRESET))
 85	    {
 86	       // Client socket closed
 87	       printf("Client socket %d closed.\n", g_CliSocketArr);
 88	       closesocket(g_CliSocketArr);
 89	       if (i < g_iTotalConn - 1)
 90	       {            
 91	          g_CliSocketArr[i--] = g_CliSocketArr[--g_iTotalConn];
 92	       }
 93	    }
 94	    else
 95	    {
 96	      // We received a message from client
 97          szMessage[ret] = '\0';
 98     	  send(g_CliSocketArr, szMessage, strlen(szMessage), 0);
 99        }
100      }
101    }
102  }
103  
104  return 0;
105}

服务器的几个主要动作如下: 1.创建监听套接字,绑定,监听; 2.创建工作者线程; 3.创建一个套接字数组,用来存放当前所有活动的客户端套接字,每accept一个连接就更新一次数组; 4.接受客户端的连接。这里有一点需要注意的,就是我没有重新定义FD_SETSIZE宏,所以服务器最多支持的并发连接数为64。 而且,这里决不能无条件的accept,服务器应该根据当前的连接数来决定是否接受来自某个客户端的连接。一种比较好的实现方 案就是采用WSAAccept函数,而且让WSAAccept回调自己实现的Condition Function。如下所示:

1int CALLBACK ConditionFunc(LPWSABUF lpCallerId,LPWSABUF lpCallerData, LPQOS lpSQOS,LPQOS lpGQOS,LPWSABUF lpCalleeId, LPWSABUF lpCalleeData,GROUP FAR * g,DWORD dwCallbackData)
2{
3	if (当前连接数 < FD_SETSIZE)
4	  return CF_ACCEPT;
5	else
6	  return CF_REJECT;
7}

工作者线程里面是一个死循环,一次循环完成的动作是:

  1. 将当前所有的客户端套接字加入到读集fdread中;
  2. 调用select函数;
  3. 查看某个套接字是否仍然处于读集中,如果是,则接收数据。如果接收的数据长度为0,或者发生WSAECONNRESET错误,则表示客户端套接字主动关闭,这时需要将服务器中对应的套接字所绑定的资源释放掉,然后调整我们的套接字数组(将数组中最后一个套接字挪到当前的位置上)

除了需要有条件接受客户端的连接外,还需要在连接数为0的情形下做特殊处理,因为如果读集中没有任何套接字,select函数会立刻返回,这将导致工作者线程成为一个毫无停顿的死循环,CPU的占用率马上达到100%。

关系到套接字列表的操作都需要使用循环,在轮询的时候,需要遍历一次,再新的一轮开始时,将列表加入队列又需要遍历一次.也就是说,Select在工作一次时,需要至少遍历2次列表,这是它效率较低的原因之一.在大规模的网络连接方面,还是推荐使用IOCP或EPOLL模型.但是Select模型可以使用在诸如对战类游戏上,比如类似星际这种,因为它小巧易于实现,而且对战类游戏的网络连接量并不大.

对于Select模型想要突破Windows 64个限制的话,可以采取分段轮询,一次轮询64个.例如套接字列表为128个,在第一次轮询时,将前64个放入队列中用Select进行状态查询,待本次操作全部结束后.将后64个再加入轮询队列中进行轮询处理.这样处理需要在非阻塞式下工作.以此类推,Select也能支持无限多个.

二.异步选择

Winsock提供了一个有用的异步I/O模型。利用这个模型,应用程序可在一个套接字上,接收以Windows消息为基础的网络事件通知。具体的做法是在建好一个套接字后,调用WSAAsyncSelect函数。该模型最早出现于Winsock的1.1版本中,用于帮助应用程序开发者面向一些早期的16位Windows平台(如Windows for Workgroups),适应其“落后”的多任务消息环境。应用程序仍可从这种模型中得到好处,特别是它们用一个标准的Windows例程(常称为"WndProc"),对窗口消息进行管理的时候。该模型亦得到了Microsoft Foundation Class(微软基本类,MFC)对象CSocket的采纳。(节选自《Windows网络编程》第八章) 我还是先贴出代码,然后做详细解释:

  1#include <winsock.h>
  2#include <tchar.h>
  3
  4#define PORT      5150
  5#define MSGSIZE   1024
  6#define WM_SOCKET WM_USER+0
  7
  8#pragma comment(lib, "ws2_32.lib")
  9
 10LRESULT CALLBACK WndProc(HWND, UINT, WPARAM, LPARAM);
 11
 12int WINAPI WinMain(HINSTANCE hInstance, HINSTANCE hPrevInstance, PSTR szCmdLine, int iCmdShow)
 13{
 14  static TCHAR szAppName[] = _T("AsyncSelect Model");
 15  HWND         hwnd ;
 16  MSG          msg ;
 17  WNDCLASS     wndclass ;
 18
 19  wndclass.style         = CS_HREDRAW | CS_VREDRAW ;
 20  wndclass.lpfnWndProc   = WndProc ;
 21  wndclass.cbClsExtra    = 0 ;
 22  wndclass.cbWndExtra    = 0 ;
 23  wndclass.hInstance     = hInstance ;
 24  wndclass.hIcon         = LoadIcon (NULL, IDI_APPLICATION) ;
 25  wndclass.hCursor       = LoadCursor (NULL, IDC_ARROW) ;
 26  wndclass.hbrBackground = (HBRUSH) GetStockObject (WHITE_BRUSH) ;
 27  wndclass.lpszMenuName  = NULL ;
 28  wndclass.lpszClassName = szAppName ;
 29
 30  if (!RegisterClass(&wndclass))
 31  {
 32    MessageBox (NULL, TEXT ("This program requires Windows NT!"), szAppName, MB_ICONERROR) ;
 33    return 0 ;
 34  }
 35
 36  hwnd = CreateWindow (szAppName,                  // window class name
 37                       TEXT ("AsyncSelect Model"), // window caption
 38                       WS_OVERLAPPEDWINDOW,        // window style
 39                       CW_USEDEFAULT,              // initial x position
 40                       CW_USEDEFAULT,              // initial y position
 41                       CW_USEDEFAULT,              // initial x size
 42                       CW_USEDEFAULT,              // initial y size
 43                       NULL,                       // parent window handle
 44                       NULL,                       // window menu handle
 45                       hInstance,                  // program instance handle
 46                       NULL) ;                     // creation parameters
 47
 48  ShowWindow(hwnd, iCmdShow);
 49  UpdateWindow(hwnd);
 50
 51  while (GetMessage(&msg, NULL, 0, 0))
 52  {
 53    TranslateMessage(&msg) ;
 54    DispatchMessage(&msg) ;
 55  }
 56  
 57  return msg.wParam;
 58}
 59
 60LRESULT CALLBACK WndProc (HWND hwnd, UINT message, WPARAM wParam, LPARAM lParam)
 61{
 62  WSADATA       wsd;
 63  static SOCKET sListen;
 64  SOCKET        sClient;
 65  SOCKADDR_IN   local, client;
 66  int           ret, iAddrSize = sizeof(client);
 67  char          szMessage[MSGSIZE];
 68
 69  switch (message)
 70  {
 71  case WM_CREATE:
 72    // Initialize Windows Socket library
 73    WSAStartup(0x0202, &wsd);
 74  
 75    // Create listening socket
 76    sListen = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
 77    
 78    // Bind
 79    local.sin_addr.S_un.S_addr = htonl(INADDR_ANY);
 80    local.sin_family = AF_INET;
 81    local.sin_port = htons(PORT);
 82    bind(sListen, (struct sockaddr *)&local, sizeof(local));
 83  
 84    // Listen
 85    listen(sListen, 3);
 86
 87    // Associate listening socket with FD_ACCEPT event
 88    WSAAsyncSelect(sListen, hwnd, WM_SOCKET, FD_ACCEPT);
 89    return 0;
 90
 91  case WM_DESTROY:
 92    closesocket(sListen);
 93    WSACleanup();
 94    PostQuitMessage(0);
 95    return 0;
 96  
 97  case WM_SOCKET:
 98    if (WSAGETSELECTERROR(lParam))
 99    {
100      closesocket(wParam);
101      break;
102    }
103    
104    switch (WSAGETSELECTEVENT(lParam))
105    {
106    case FD_ACCEPT:
107      // Accept a connection from client
108      sClient = accept(wParam, (struct sockaddr *)&client, &iAddrSize);
109      // Associate client socket with FD_READ and FD_CLOSE event
110      WSAAsyncSelect(sClient, hwnd, WM_SOCKET, FD_READ | FD_CLOSE);
111      break;
112
113    case FD_READ:
114      ret = recv(wParam, szMessage, MSGSIZE, 0);
115      if (ret == 0 || ret == SOCKET_ERROR && WSAGetLastError() == WSAECONNRESET)
116      {
117        closesocket(wParam);
118      }
119      else
120      {
121        szMessage[ret] = '\0';
122        send(wParam, szMessage, strlen(szMessage), 0);
123      }
124      break;
125      
126    case FD_CLOSE:
127      closesocket(wParam);      
128      break;
129    }
130    return 0;
131  }
132  
133  return DefWindowProc(hwnd, message, wParam, lParam);
134}

在我看来,WSAAsyncSelect是最简单的一种Winsock I/O模型(之所以说它简单是因为一个主线程就搞定了)。使用Raw Windows API写过窗口类应用程序的人应该都能看得懂。这里,我们需要做的仅仅是:

  1. 在WM_CREATE消息处理函数中,初始化Windows Socket library,创建监听套接字,绑定,监听,并且调用WSAAsyncSelect函数表示我们关心在监听套接字上发生的FD_ACCEPT事件;
  2. 自定义一个消息WM_SOCKET,一旦在我们所关心的套接字(监听套接字和客户端套接字)上发生了某个事件,系统就会调用WndProc并且message参数被设置为WM_SOCKET;
  3. 在WM_SOCKET的消息处理函数中,分别对FD_ACCEPT、FD_READ和FD_CLOSE事件进行处理;
  4. 在窗口销毁消息(WM_DESTROY)的处理函数中,我们关闭监听套接字,清除Windows Socket library

下面这张用于WSAAsyncSelect函数的网络事件类型表可以让你对各个网络事件有更清楚的认识: 表1

事件 说明
FD_READ 应用程序想要接收有关是否可读的通知,以便读入数据
FD_WRITE 应用程序想要接收有关是否可写的通知,以便写入数据
FD_OOB 应用程序想接收是否有带外(OOB)数据抵达的通知
FD_ACCEPT 应用程序想接收与进入连接有关的通知
FD_CONNECT 应用程序想接收与一次连接或者多点join操作完成的通知
FD_CLOSE 应用程序想接收与套接字关闭有关的通知
FD_QOS 应用程序想接收套接字“服务质量”(QoS)发生更改的通知
FD_GROUP_QOS 应用程序想接收套接字组“服务质量”发生更改的通知(现在没什么用处,为未来套接字组的使用保留)
FD_ROUTING_INTERFACE_CHANGE 应用程序想接收在指定的方向上,与路由接口发生变化的通知
FD_ADDRESS_LIST_CHANGE 应用程序想接收针对套接字的协议家族,本地地址列表发生变化的通知

三.事件选择

Winsock提供了另一个有用的异步I/O模型。和WSAAsyncSelect模型类似的是,它也允许应用程序在一个或多个套接字上,接收以事件为基础的网络事件通知。对于表1总结的、由WSAAsyncSelect模型采用的网络事件来说,它们均可原封不动地移植到新模型。在用新模型开发的应用程序中,也能接收和处理所有那些事件。该模型最主要的差别在于网络事件会投递至一个事件对象句柄,而非投递至一个窗口例程。(节选自《Windows网络编程》第八章) 还是让我们先看代码然后进行分析:

  1#include <winsock2.h>
  2#include <stdio.h>
  3
  4#define PORT    5150
  5#define MSGSIZE 1024
  6
  7#pragma comment(lib, "ws2_32.lib")
  8
  9int      g_iTotalConn = 0;
 10SOCKET   g_CliSocketArr[MAXIMUM_WAIT_OBJECTS];
 11WSAEVENT g_CliEventArr[MAXIMUM_WAIT_OBJECTS];
 12
 13DWORD WINAPI WorkerThread(LPVOID);
 14void Cleanup(int index);
 15
 16int main()
 17{
 18  WSADATA     wsaData;
 19  SOCKET      sListen, sClient;
 20  SOCKADDR_IN local, client;
 21  DWORD       dwThreadId;
 22  int         iaddrSize = sizeof(SOCKADDR_IN);
 23
 24  // Initialize Windows Socket library
 25  WSAStartup(0x0202, &wsaData);
 26
 27  // Create listening socket
 28  sListen = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
 29
 30  // Bind
 31  local.sin_addr.S_un.S_addr = htonl(INADDR_ANY);
 32  local.sin_family = AF_INET;
 33  local.sin_port = htons(PORT);
 34  bind(sListen, (struct sockaddr *)&local, sizeof(SOCKADDR_IN));
 35
 36  // Listen
 37  listen(sListen, 3);
 38
 39  // Create worker thread
 40  CreateThread(NULL, 0, WorkerThread, NULL, 0, &dwThreadId);
 41
 42  while (TRUE)
 43  {
 44    // Accept a connection
 45    sClient = accept(sListen, (struct sockaddr *)&client, &iaddrSize);
 46    printf("Accepted client:%s:%d\n", inet_ntoa(client.sin_addr), ntohs(client.sin_port));
 47
 48    // Associate socket with network event
 49    g_CliSocketArr[g_iTotalConn] = sClient;
 50    g_CliEventArr[g_iTotalConn] = WSACreateEvent();
 51    WSAEventSelect(g_CliSocketArr[g_iTotalConn],
 52                   g_CliEventArr[g_iTotalConn],
 53                   FD_READ | FD_CLOSE);
 54    g_iTotalConn++;
 55  }
 56}
 57
 58DWORD WINAPI WorkerThread(LPVOID lpParam)
 59{
 60  int              ret, index;
 61  WSANETWORKEVENTS NetworkEvents;
 62  char             szMessage[MSGSIZE];
 63
 64  while (TRUE)
 65  {
 66    ret = WSAWaitForMultipleEvents(g_iTotalConn, g_CliEventArr, FALSE, 1000, FALSE);
 67    if (ret == WSA_WAIT_FAILED || ret == WSA_WAIT_TIMEOUT)
 68    {
 69      continue;
 70    }
 71
 72    index = ret - WSA_WAIT_EVENT_0;
 73    WSAEnumNetworkEvents(g_CliSocketArr[index], g_CliEventArr[index], &NetworkEvents);
 74
 75    if (NetworkEvents.lNetworkEvents & FD_READ)
 76    {
 77      // Receive message from client
 78      ret = recv(g_CliSocketArr[index], szMessage, MSGSIZE, 0);
 79      if (ret == 0 || (ret == SOCKET_ERROR && WSAGetLastError() == WSAECONNRESET))
 80      {
 81        Cleanup(index);
 82      }
 83      else
 84      {
 85        szMessage[ret] = '\0';
 86        send(g_CliSocketArr[index], szMessage, strlen(szMessage), 0);
 87      }
 88    }
 89
 90    if (NetworkEvents.lNetworkEvents & FD_CLOSE)
 91    {
 92       Cleanup(index);
 93    }
 94  }
 95  return 0;
 96}
 97
 98void Cleanup(int index)
 99{
100  closesocket(g_CliSocketArr[index]);
101  WSACloseEvent(g_CliEventArr[index]);
102
103  if (index < g_iTotalConn - 1)
104  {
105    g_CliSocketArr[index] = g_CliSocketArr[g_iTotalConn - 1];
106    g_CliEventArr[index] = g_CliEventArr[g_iTotalConn - 1];
107  }
108
109  g_iTotalConn--;
110}

事件选择模型也比较简单,实现起来也不是太复杂,它的基本思想是将每个套接字都和一个WSAEVENT对象对应起来,并且在关联的时候指定需要关注的哪些网络事件。一旦在某个套接字上发生了我们关注的事件(FD_READ和FD_CLOSE),与之相关联的WSAEVENT对象被Signaled。程序定义了两个全局数组,一个套接字数组,一个WSAEVENT对象数组,其大小都是MAXIMUM_WAIT_OBJECTS(64),两个数组中的元素一一对应。 同样的,这里的程序没有考虑两个问题,一是不能无条件的调用accept,因为我们支持的并发连接数有限。解决方法是将套接字按MAXIMUM_WAIT_OBJECTS分组,每MAXIMUM_WAIT_OBJECTS个套接字一组,每一组分配一个工作者线程;或者采用WSAAccept代替accept,并回调自己定义的Condition Function。第二个问题是没有对连接数为0的情形做特殊处理,程序在连接数为0的时候CPU占用率为100%。

四.重叠I/O模型

Winsock2的发布使得Socket I/O有了和文件I/O统一的接口。我们可以通过使用Win32文件操纵函数ReadFile和WriteFile来进行Socket I/O。伴随而来的,用于普通文件I/O的重叠I/O模型和完成端口模型对Socket I/O也适用了。这些模型的优点是可以达到更佳的系统性能,但是实现较为复杂,里面涉及较多的C语言技巧。例如我们在完成端口模型中会经常用到所谓的“尾随数据”。

1. 用事件通知方式实现的重叠I/O模型

  1#include <winsock2.h>
  2#include <stdio.h>
  3
  4#define PORT    5150
  5#define MSGSIZE 1024
  6
  7#pragma comment(lib, "ws2_32.lib")
  8
  9typedef struct
 10{
 11  WSAOVERLAPPED overlap;
 12  WSABUF        Buffer;
 13  char          szMessage[MSGSIZE];
 14  DWORD         NumberOfBytesRecvd;
 15  DWORD         Flags;
 16}PER_IO_OPERATION_DATA, *LPPER_IO_OPERATION_DATA;
 17
 18int                     g_iTotalConn = 0;
 19SOCKET                  g_CliSocketArr[MAXIMUM_WAIT_OBJECTS];
 20WSAEVENT                g_CliEventArr[MAXIMUM_WAIT_OBJECTS];
 21LPPER_IO_OPERATION_DATA g_pPerIODataArr[MAXIMUM_WAIT_OBJECTS];
 22
 23DWORD WINAPI WorkerThread(LPVOID);
 24void Cleanup(int);
 25
 26int main()
 27{
 28  WSADATA     wsaData;
 29  SOCKET      sListen, sClient;
 30  SOCKADDR_IN local, client;
 31  DWORD       dwThreadId;
 32  int         iaddrSize = sizeof(SOCKADDR_IN);
 33
 34  // Initialize Windows Socket library
 35  WSAStartup(0x0202, &wsaData);
 36
 37  // Create listening socket
 38  sListen = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
 39
 40  // Bind
 41  local.sin_addr.S_un.S_addr = htonl(INADDR_ANY);
 42  local.sin_family = AF_INET;
 43  local.sin_port = htons(PORT);
 44  bind(sListen, (struct sockaddr *)&local, sizeof(SOCKADDR_IN));
 45
 46  // Listen
 47  listen(sListen, 3);
 48
 49  // Create worker thread
 50  CreateThread(NULL, 0, WorkerThread, NULL, 0, &dwThreadId);
 51
 52  while (TRUE)
 53  {
 54    // Accept a connection
 55    sClient = accept(sListen, (struct sockaddr *)&client, &iaddrSize);
 56    printf("Accepted client:%s:%d\n", inet_ntoa(client.sin_addr), ntohs(client.sin_port));
 57
 58    g_CliSocketArr[g_iTotalConn] = sClient;
 59    
 60    // Allocate a PER_IO_OPERATION_DATA structure
 61    g_pPerIODataArr[g_iTotalConn] = (LPPER_IO_OPERATION_DATA)HeapAlloc(
 62      GetProcessHeap(),
 63      HEAP_ZERO_MEMORY,
 64      sizeof(PER_IO_OPERATION_DATA));
 65    g_pPerIODataArr[g_iTotalConn]->Buffer.len = MSGSIZE;
 66    g_pPerIODataArr[g_iTotalConn]->Buffer.buf = g_pPerIODataArr[g_iTotalConn]->szMessage;
 67    g_CliEventArr[g_iTotalConn] = g_pPerIODataArr[g_iTotalConn]->overlap.hEvent = WSACreateEvent();
 68
 69    // Launch an asynchronous operation
 70    WSARecv(
 71      g_CliSocketArr[g_iTotalConn],
 72      &g_pPerIODataArr[g_iTotalConn]->Buffer,
 73      1,
 74      &g_pPerIODataArr[g_iTotalConn]->NumberOfBytesRecvd,
 75      &g_pPerIODataArr[g_iTotalConn]->Flags,
 76      &g_pPerIODataArr[g_iTotalConn]->overlap,
 77      NULL);
 78    
 79    g_iTotalConn++;
 80  }
 81  
 82  closesocket(sListen);
 83  WSACleanup();
 84  return 0;
 85}
 86
 87DWORD WINAPI WorkerThread(LPVOID lpParam)
 88{
 89  int   ret, index;
 90  DWORD cbTransferred;
 91
 92  while (TRUE)
 93  {
 94    ret = WSAWaitForMultipleEvents(g_iTotalConn, g_CliEventArr, FALSE, 1000, FALSE);
 95    if (ret == WSA_WAIT_FAILED || ret == WSA_WAIT_TIMEOUT)
 96    {
 97      continue;
 98    }
 99
100    index = ret - WSA_WAIT_EVENT_0;
101    WSAResetEvent(g_CliEventArr[index]);
102
103    WSAGetOverlappedResult(
104      g_CliSocketArr[index],
105      &g_pPerIODataArr[index]->overlap,
106      &cbTransferred,
107      TRUE,
108      &g_pPerIODataArr[g_iTotalConn]->Flags);
109
110    if (cbTransferred == 0)
111    {
112      // The connection was closed by client
113      Cleanup(index);
114    }
115    else
116    {
117      // g_pPerIODataArr[index]->szMessage contains the received data
118      g_pPerIODataArr[index]->szMessage[cbTransferred] = '\0';
119      send(g_CliSocketArr[index], g_pPerIODataArr[index]->szMessage,\
120        cbTransferred, 0);
121
122      // Launch another asynchronous operation
123      WSARecv(
124        g_CliSocketArr[index],
125        &g_pPerIODataArr[index]->Buffer,
126        1,
127        &g_pPerIODataArr[index]->NumberOfBytesRecvd,
128        &g_pPerIODataArr[index]->Flags,
129        &g_pPerIODataArr[index]->overlap,
130        NULL);
131    }
132  }
133
134  return 0;
135}
136
137void Cleanup(int index)
138{
139  closesocket(g_CliSocketArr[index]);
140  WSACloseEvent(g_CliEventArr[index]);
141  HeapFree(GetProcessHeap(), 0, g_pPerIODataArr[index]);
142
143  if (index < g_iTotalConn - 1)
144  {
145    g_CliSocketArr[index] = g_CliSocketArr[g_iTotalConn - 1];
146    g_CliEventArr[index] = g_CliEventArr[g_iTotalConn - 1];
147    g_pPerIODataArr[index] = g_pPerIODataArr[g_iTotalConn - 1];
148  }
149
150  g_pPerIODataArr[--g_iTotalConn] = NULL;
151}

这个模型与上述其他模型不同的是它使用Winsock2提供的异步I/O函数WSARecv。在调用WSARecv时,指定一个WSAOVERLAPPED结构,这个调用不是阻塞的,也就是说,它会立刻返回。一旦有数据到达的时候,被指定的WSAOVERLAPPED结构中的hEvent被Signaled。由于下面这个语句 g_CliEventArr[g_iTotalConn] = g_pPerIODataArr[g_iTotalConn]->overlap.hEvent; 使得与该套接字相关联的WSAEVENT对象也被Signaled,所以WSAWaitForMultipleEvents的调用操作成功返回。我们现在应该做的就是用与调用WSARecv相同的WSAOVERLAPPED结构为参数调用WSAGetOverlappedResult,从而得到本次I/O传送的字节数等相关信息。在取得接收的数据后,把数据原封不动的发送到客户端,然后重新激活一个WSARecv异步操作。

2.用完成例程方式实现的重叠I/O模型

  1#include <WINSOCK2.H>
  2#include <stdio.h>
  3
  4#define PORT    5150
  5#define MSGSIZE 1024
  6
  7#pragma comment(lib, "ws2_32.lib")
  8
  9typedef struct
 10{
 11  WSAOVERLAPPED overlap;
 12  WSABUF        Buffer;
 13  char          szMessage[MSGSIZE];
 14  DWORD         NumberOfBytesRecvd;
 15  DWORD         Flags; 
 16  SOCKET        sClient;
 17}PER_IO_OPERATION_DATA, *LPPER_IO_OPERATION_DATA;
 18
 19DWORD WINAPI WorkerThread(LPVOID);
 20void CALLBACK CompletionROUTINE(DWORD, DWORD, LPWSAOVERLAPPED, DWORD);
 21
 22SOCKET g_sNewClientConnection;
 23BOOL   g_bNewConnectionArrived = FALSE;
 24
 25int main()
 26{
 27  WSADATA     wsaData;
 28  SOCKET      sListen;
 29  SOCKADDR_IN local, client;
 30  DWORD       dwThreadId;
 31  int         iaddrSize = sizeof(SOCKADDR_IN);
 32
 33  // Initialize Windows Socket library
 34  WSAStartup(0x0202, &wsaData);
 35
 36  // Create listening socket
 37  sListen = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
 38
 39  // Bind
 40  local.sin_addr.S_un.S_addr = htonl(INADDR_ANY);
 41  local.sin_family = AF_INET;
 42  local.sin_port = htons(PORT);
 43  bind(sListen, (struct sockaddr *)&local, sizeof(SOCKADDR_IN));
 44
 45  // Listen
 46  listen(sListen, 3);
 47
 48  // Create worker thread
 49  CreateThread(NULL, 0, WorkerThread, NULL, 0, &dwThreadId);
 50
 51  while (TRUE)
 52  {
 53    // Accept a connection
 54    g_sNewClientConnection = accept(sListen, (struct sockaddr *)&client, &iaddrSize);
 55    g_bNewConnectionArrived = TRUE;
 56    printf("Accepted client:%s:%d\n", inet_ntoa(client.sin_addr), ntohs(client.sin_port));
 57  }
 58}
 59
 60DWORD WINAPI WorkerThread(LPVOID lpParam)
 61{
 62  LPPER_IO_OPERATION_DATA lpPerIOData = NULL;
 63  while (TRUE)
 64  {
 65    if (g_bNewConnectionArrived)
 66    {
 67      // Launch an asynchronous operation for new arrived connection
 68      lpPerIOData = (LPPER_IO_OPERATION_DATA)HeapAlloc(
 69        GetProcessHeap(),
 70        HEAP_ZERO_MEMORY,
 71        sizeof(PER_IO_OPERATION_DATA));
 72      lpPerIOData->Buffer.len = MSGSIZE;
 73      lpPerIOData->Buffer.buf = lpPerIOData->szMessage;
 74      lpPerIOData->sClient = g_sNewClientConnection;
 75      
 76      WSARecv(lpPerIOData->sClient,
 77        &lpPerIOData->Buffer,
 78        1,
 79        &lpPerIOData->NumberOfBytesRecvd,
 80        &lpPerIOData->Flags,
 81        &lpPerIOData->overlap,
 82        CompletionROUTINE);      
 83      
 84      g_bNewConnectionArrived = FALSE;
 85    }
 86
 87    SleepEx(1000, TRUE);
 88  }
 89  return 0;
 90}
 91
 92void CALLBACK CompletionROUTINE(DWORD dwError,
 93                                DWORD cbTransferred,
 94                                LPWSAOVERLAPPED lpOverlapped,
 95                                DWORD dwFlags)
 96{
 97  LPPER_IO_OPERATION_DATA lpPerIOData = (LPPER_IO_OPERATION_DATA)lpOverlapped;
 98  
 99  if (dwError != 0 || cbTransferred == 0)
100  {
101    // Connection was closed by client
102    closesocket(lpPerIOData->sClient);
103    HeapFree(GetProcessHeap(), 0, lpPerIOData);
104  }
105  else
106  {
107    lpPerIOData->szMessage[cbTransferred] = '\0';
108    send(lpPerIOData->sClient, lpPerIOData->szMessage, cbTransferred, 0);
109    
110    // Launch another asynchronous operation
111    memset(&lpPerIOData->overlap, 0, sizeof(WSAOVERLAPPED));
112    lpPerIOData->Buffer.len = MSGSIZE;
113    lpPerIOData->Buffer.buf = lpPerIOData->szMessage;    
114
115    WSARecv(lpPerIOData->sClient,
116      &lpPerIOData->Buffer,
117      1,
118      &lpPerIOData->NumberOfBytesRecvd,
119      &lpPerIOData->Flags,
120      &lpPerIOData->overlap,
121      CompletionROUTINE);
122  }
123}

用完成例程来实现重叠I/O比用事件通知简单得多。在这个模型中,主线程只用不停的接受连接即可;辅助线程判断有没有新的客户端连接被建立,如果有,就为那个客户端套接字激活一个异步的WSARecv操作,然后调用SleepEx使线程处于一种可警告的等待状态,以使得I/O完成后CompletionROUTINE可以被内核调用。如果辅助线程不调用SleepEx,则内核在完成一次I/O操作后,无法调用完成例程(因为完成例程的运行应该和当初激活WSARecv异步操作的代码在同一个线程之内)。 完成例程内的实现代码比较简单,它取出接收到的数据,然后将数据原封不动的发送给客户端,最后重新激活另一个WSARecv异步操作。注意,在这里用到了“尾随数据”。我们在调用WSARecv的时候,参数lpOverlapped实际上指向一个比它大得多的结构PER_IO_OPERATION_DATA,这个结构除了WSAOVERLAPPED以外,还被我们附加了缓冲区的结构信息,另外还包括客户端套接字等重要的信息。这样,在完成例程中通过参数lpOverlapped拿到的不仅仅是WSAOVERLAPPED结构,还有后边尾随的包含客户端套接字和接收数据缓冲区等重要信息。这样的C语言技巧在我后面介绍完成端口的时候还会使用到。

五.完成端口模型

“完成端口”模型是迄今为止最为复杂的一种I/O模型。然而,假若一个应用程序同时需要管理为数众多的套接字,那么采用这种模型,往往可以达到最佳的系统性能!但不幸的是,该模型只适用于Windows NT和Windows 2000操作系统。因其设计的复杂性,只有在你的应用程序需要同时管理数百乃至上千个套接字的时候,而且希望随着系统内安装的CPU数量的增多,应用程序的性能也可以线性提升,才应考虑采用“完成端口”模型。要记住的一个基本准则是,假如要为Windows NT或Windows 2000开发高性能的服务器应用,同时希望为大量套接字I/O请求提供服务(Web服务器便是这方面的典型例子),那么I/O完成端口模型便是最佳选择!(节选自《Windows网络编程》第八章) 完成端口模型是我最喜爱的一种模型。虽然其实现比较复杂(其实我觉得它的实现比用事件通知实现的重叠I/O简单多了),但其效率是惊人的。我在T公司的时候曾经帮同事写过一个邮件服务器的性能测试程序,用的就是完成端口模型。结果表明,完成端口模型在多连接(成千上万)的情况下,仅仅依靠一两个辅助线程,就可以达到非常高的吞吐量。下面我还是从代码说起:

  1#include <WINSOCK2.H>
  2#include <stdio.h>
  3
  4#define PORT    5150
  5#define MSGSIZE 1024
  6
  7#pragma comment(lib, "ws2_32.lib")
  8
  9typedef enum
 10{
 11  RECV_POSTED
 12}OPERATION_TYPE;
 13
 14typedef struct
 15{
 16  WSAOVERLAPPED  overlap;
 17  WSABUF         Buffer;
 18  char           szMessage[MSGSIZE];
 19  DWORD          NumberOfBytesRecvd;
 20  DWORD          Flags;
 21  OPERATION_TYPE OperationType;
 22}PER_IO_OPERATION_DATA, *LPPER_IO_OPERATION_DATA;
 23
 24DWORD WINAPI WorkerThread(LPVOID);
 25
 26int main()
 27{
 28  WSADATA                 wsaData;
 29  SOCKET                  sListen, sClient;
 30  SOCKADDR_IN             local, client;
 31  DWORD                   i, dwThreadId;
 32  int                     iaddrSize = sizeof(SOCKADDR_IN);
 33  HANDLE                  CompletionPort = INVALID_HANDLE_VALUE;
 34  SYSTEM_INFO             systeminfo;
 35  LPPER_IO_OPERATION_DATA lpPerIOData = NULL;
 36
 37  // Initialize Windows Socket library
 38  WSAStartup(0x0202, &wsaData);
 39
 40  // Create completion port
 41  CompletionPort = CreateIoCompletionPort(INVALID_HANDLE_VALUE, NULL, 0, 0);
 42
 43  // Create worker thread
 44  GetSystemInfo(&systeminfo);
 45  for (i = 0; i < systeminfo.dwNumberOfProcessors; i++)
 46  {
 47    CreateThread(NULL, 0, WorkerThread, CompletionPort, 0, &dwThreadId);
 48  }
 49  
 50  // Create listening socket
 51  sListen = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
 52
 53  // Bind
 54  local.sin_addr.S_un.S_addr = htonl(INADDR_ANY);
 55  local.sin_family = AF_INET;
 56  local.sin_port = htons(PORT);
 57  bind(sListen, (struct sockaddr *)&local, sizeof(SOCKADDR_IN));
 58
 59  // Listen
 60  listen(sListen, 3);
 61
 62  while (TRUE)
 63  {
 64    // Accept a connection
 65    sClient = accept(sListen, (struct sockaddr *)&client, &iaddrSize);
 66    printf("Accepted client:%s:%d\n", inet_ntoa(client.sin_addr), ntohs(client.sin_port));
 67
 68    // Associate the newly arrived client socket with completion port
 69    CreateIoCompletionPort((HANDLE)sClient, CompletionPort, (DWORD)sClient, 0);
 70    
 71    // Launch an asynchronous operation for new arrived connection
 72    lpPerIOData = (LPPER_IO_OPERATION_DATA)HeapAlloc(
 73      GetProcessHeap(),
 74      HEAP_ZERO_MEMORY,
 75      sizeof(PER_IO_OPERATION_DATA));
 76    lpPerIOData->Buffer.len = MSGSIZE;
 77    lpPerIOData->Buffer.buf = lpPerIOData->szMessage;
 78    lpPerIOData->OperationType = RECV_POSTED;
 79    WSARecv(sClient,
 80      &lpPerIOData->Buffer,
 81      1,
 82      &lpPerIOData->NumberOfBytesRecvd,
 83      &lpPerIOData->Flags,
 84      &lpPerIOData->overlap,
 85      NULL);
 86  }
 87
 88  PostQueuedCompletionStatus(CompletionPort, 0xFFFFFFFF, 0, NULL);
 89  CloseHandle(CompletionPort);
 90  closesocket(sListen);
 91  WSACleanup();
 92  return 0;
 93}
 94
 95DWORD WINAPI WorkerThread(LPVOID CompletionPortID)
 96{
 97  HANDLE                  CompletionPort=(HANDLE)CompletionPortID;
 98  DWORD                   dwBytesTransferred;
 99  SOCKET                  sClient;
100  LPPER_IO_OPERATION_DATA lpPerIOData = NULL;
101
102  while (TRUE)
103  {
104    GetQueuedCompletionStatus(
105      CompletionPort,
106      &dwBytesTransferred,
107      &sClient,
108      (LPOVERLAPPED *)&lpPerIOData,
109      INFINITE);
110    if (dwBytesTransferred == 0xFFFFFFFF)
111    {
112      return 0;
113    }
114    
115    if (lpPerIOData->OperationType == RECV_POSTED)
116    {
117      if (dwBytesTransferred == 0)
118      {
119        // Connection was closed by client
120        closesocket(sClient);
121        HeapFree(GetProcessHeap(), 0, lpPerIOData);        
122      }
123      else
124      {
125        lpPerIOData->szMessage[dwBytesTransferred] = '\0';
126        send(sClient, lpPerIOData->szMessage, dwBytesTransferred, 0);
127        
128        // Launch another asynchronous operation for sClient
129        memset(lpPerIOData, 0, sizeof(PER_IO_OPERATION_DATA));
130        lpPerIOData->Buffer.len = MSGSIZE;
131        lpPerIOData->Buffer.buf = lpPerIOData->szMessage;
132        lpPerIOData->OperationType = RECV_POSTED;
133        WSARecv(sClient,
134          &lpPerIOData->Buffer,
135          1,
136          &lpPerIOData->NumberOfBytesRecvd,
137          &lpPerIOData->Flags,
138          &lpPerIOData->overlap,
139          NULL);
140      }
141    }
142  }
143  return 0;
144}

首先,说说主线程:

  1. 创建完成端口对象
  2. 创建工作者线程(这里工作者线程的数量是按照CPU的个数来决定的,这样可以达到最佳性能)
  3. 创建监听套接字,绑定,监听,然后程序进入循环
  4. 在循环中,我做了以下几件事情: (1). 接受一个客户端连接 (2). 将该客户端套接字与完成端口绑定到一起(还是调用CreateIoCompletionPort,但这次的作用不同),注意,按道理来讲,此时传递给CreateIoCompletionPort的第三个参数应该是一个完成键,一般来讲,程序都是传递一个单句柄数据结构的地址,该单句柄数据包含了和该客户端连接有关的信息,由于我们只关心套接字句柄,所以直接将套接字句柄作为完成键传递; (3). 触发一个WSARecv异步调用,这次又用到了“尾随数据”,使接收数据所用的缓冲区紧跟在WSAOVERLAPPED对象之后,此外,还有操作类型等重要信息。

在工作者线程的循环中,我们

  1. 调用GetQueuedCompletionStatus取得本次I/O的相关信息(例如套接字句柄、传送的字节数、单I/O数据结构的地址等等)
  2. 通过单I/O数据结构找到接收数据缓冲区,然后将数据原封不动的发送到客户端
  3. 再次触发一个WSARecv异步操作

五种I/O模型的比较

我会从以下几个方面来进行比较 *有无每线程64连接数限制 如果在选择模型中没有重新定义FD_SETSIZE宏,则每个fd_set默认可以装下64个SOCKET。同样的,受MAXIMUM_WAIT_OBJECTS宏的影响,事件选择、用事件通知实现的重叠I/O都有每线程最大64连接数限制。如果连接数成千上万,则必须对客户端套接字进行分组,这样,势必增加程序的复杂度。 相反,异步选择、用完成例程实现的重叠I/O和完成端口不受此限制。

*线程数 除了异步选择以外,其他模型至少需要2个线程。一个主线程和一个辅助线程。同样的,如果连接数大于64,则选择模型、事件选择和用事件通知实现的重叠I/O的线程数还要增加。

*实现的复杂度 我的个人看法是,在实现难度上,异步选择<选择<用完成例程实现的重叠I/O<事件选择<完成端口<用事件通知实现的重叠I/O

*性能 由于选择模型中每次都要重设读集,在select函数返回后还要针对所有套接字进行逐一测试,我的感觉是效率比较差;完成端口和用完成例程实现的重叠I/O基本上不涉及全局数据,效率应该是最高的,而且在多处理器情形下完成端口还要高一些;事件选择和用事件通知实现的重叠I/O在实现机制上都是采用WSAWaitForMultipleEvents,感觉效率差不多;至于异步选择,不好比较。所以我的结论是:选择<用事件通知实现的重叠I/O<事件选择<用完成例程实现的重叠I/O<完成端口.



◆先看定义:

1typedef unsigned int u_int;
2typedef u_int SOCKET;

◆Socket相当于进行网络通信两端的插座,只要对方的Socket和自己的Socket有通信联接,双方就可以发送和接收数据了。其定义类似于文件句柄的定义。

◆Socket有五种不同的类型:

1、流式套接字(stream socket) 定义:

1#define SOCK_STREAM 1 

流式套接字提供了双向、有序的、无重复的以及无记录边界的数据流服务,适合处理大量数据。它是面向联结的,必须建立数据传输链路,同时还必须对传输的数据进行验证,确保数据的准确性。因此,系统开销较大。

2、 数据报套接字(datagram socket)

定义:

1#define SOCK_DGRAM 2 

数据报套接字也支持双向的数据流,但不保证传输数据的准确性,但保留了记录边界。由于数据报套接字是无联接的,例如广播时的联接,所以并不保证接收端是否正在侦听。数据报套接字传输效率比较高。

3、原始套接字(raw-protocol interface)

定义:

1#define SOCK_RAW 3 

原始套接字保存了数据包中的完整IP头,前面两种套接字只能收到用户数据。因此可以通过原始套接字对数据进行分析。 其它两种套接字不常用,这里就不介绍了。

◆Socket开发所必须需要的文件(以WinSock V2.0为例):

头文件:Winsock2.h

库文件:WS2_32.LIB

动态库:W32_32.DLL

一些重要的定义

1、数据类型的基本定义:这个大家一看就懂。

1typedef unsigned char u_char;
2typedef unsigned short u_short;
3typedef unsigned int u_int;
4typedef unsigned long u_long;

2、 网络地址的数据结构,有一个老的和一个新的的,请大家留意,如果想知道为什么,请发邮件给Bill Gate。其实就是计算机的IP地址,不过一般不用用点分开的IP地址,当然也提供一些转换函数。

◆ 旧的网络地址结构的定义,为一个4字节的联合:

 1struct in_addr 
 2 {
 3	 union 
 4	 {
 5		 struct { u_char s_b1,s_b2,s_b3,s_b4; } S_un_b;
 6		 struct { u_short s_w1,s_w2; } S_un_w;
 7		 u_long S_addr;
 8	 } S_un;
 9 #define s_addr S_un.S_addr /* can be used for most tcp & ip code */
10 //下面几行省略,反正没什么用处。
11 };

其实完全不用这么麻烦,请看下面:

◆ 新的网络地址结构的定义: 非常简单,就是一个无符号长整数 unsigned long。举个例子:IP地址为127.0.0.1的网络地址是什么呢?请看定义:

1#define INADDR_LOOPBACK 0x7f000001

3、 套接字地址结构

(1)、sockaddr结构:

1struct sockaddr {
2  u_short sa_family; /* address family */
3  char sa_data[14]; /* up to 14 bytes of direct address */
4};

sa_family为网络地址类型,一般为AF_INET,表示该socket在Internet域中进行通信,该地址结构随选择的协议的不同而变化,因此一般情况下另一个与该地址结构大小相同的sockaddr_in结构更为常用,sockaddr_in结构用来标识TCP/IP协议下的地址。换句话说,这个结构是通用socket地址结构,而下面的sockaddr_in是专门针对Internet域的socket地址结构。

(2)、sockaddr_in结构

1struct sockaddr_in {
2  short sin_family;
3  u_short sin_port;
4  struct in_addr sin_addr;
5  char sin_zero[8];
6};

sin _family为网络地址类型,必须设定为AF_INET。sin_port为服务端口,注意不要使用已固定的服务端口,如HTTP的端口80等。如果端口设置为0,则系统会自动分配一个唯一端口。sin_addr为一个unsigned long的IP地址。sin_zero为填充字段,纯粹用来保证结构的大小。

◆ 将常用的用点分开的IP地址转换为unsigned long类型的IP地址的函数:

1unsigned long inet_addr(const char FAR * cp )

用法:

1unsigned long addr=inet_addr("192.1.8.84")

◆ 如果将sin_addr设置为INADDR_ANY,则表示所有的IP地址,也即所有的计算机。

1#define INADDR_ANY (u_long)0x00000000

4、 主机地址:

先看定义:

1struct hostent {
2 char FAR * h_name; /* official name of host */
3 char FAR * FAR * h_aliases; /* alias list */
4 short h_addrtype; /* host address type */
5 short h_length; /* length of address */
6 char FAR * FAR * h_addr_list; /* list of addresses */
7 #define h_addr h_addr_list[0] /* address, for backward compat */
8 };

h_name为主机名字。 h_aliases为主机别名列表。 h_addrtype为地址类型。 h_length为地址类型。 h_addr_list为IP地址,如果该主机有多个网卡,就包括地址的列表。 另外还有几个类似的结构,这里就不一一介绍了。

5、 常见TCP/IP协议的定义:

1#define IPPROTO_IP 0 
2#define IPPROTO_ICMP 1 
3#define IPPROTO_IGMP 2 
4#define IPPROTO_TCP 6 
5#define IPPROTO_UDP 17 
6#define IPPROTO_RAW 255 

具体是什么协议,大家一看就知道了。

套接字的属性

为了灵活使用套接字,我们可以对它的属性进行设定。

1、 属性内容:

 1//允许调试输出
 2#define SO_DEBUG 0x0001 /* turn on debugging info recording */
 3//是否监听模式
 4#define SO_ACCEPTCONN 0x0002 /* socket has had listen() */
 5//套接字与其他套接字的地址绑定
 6#define SO_REUSEADDR 0x0004 /* allow local address reuse */
 7//保持连接
 8#define SO_KEEPALIVE 0x0008 /* keep connections alive */
 9//不要路由出去
10#define SO_DONTROUTE 0x0010 /* just use interface addresses */
11//设置为广播
12#define SO_BROADCAST 0x0020 /* permit sending of broadcast msgs */
13//使用环回不通过硬件
14#define SO_USELOOPBACK 0x0040 /* bypass hardware when possible */
15//当前拖延值
16#define SO_LINGER 0x0080 /* linger on close if data present */
17//是否加入带外数据
18#define SO_OOBINLINE 0x0100 /* leave received OOB data in line */
19//禁用LINGER选项
20#define SO_DONTLINGER (int)(~SO_LINGER)
21//发送缓冲区长度
22#define SO_SNDBUF 0x1001 /* send buffer size */
23//接收缓冲区长度
24#define SO_RCVBUF 0x1002 /* receive buffer size */
25//发送超时时间
26#define SO_SNDTIMEO 0x1005 /* send timeout */
27//接收超时时间
28#define SO_RCVTIMEO 0x1006 /* receive timeout */
29//错误状态
30#define SO_ERROR 0x1007 /* get error status and clear */
31//套接字类型
32#define SO_TYPE 0x1008 /* get socket type */

2、 读取socket属性:

1int getsockopt(SOCKET s, int level, int optname, char FAR * optval, int FAR * optlen)

s为欲读取属性的套接字。level为套接字选项的级别,大多数是特定协议和套接字专有的。如IP协议应为 IPPROTO_IP。

optname为读取选项的名称 optval为存放选项值的缓冲区指针。 optlen为缓冲区的长度 用法:

1int ttl=0; //读取TTL值
2int rc = getsockopt( s, IPPROTO_IP, IP_TTL, (char *)&ttl, sizeof(ttl));
3//来自MS platform SDK 2003

3、 设置socket属性:

1int setsockopt(SOCKET s,int level, int optname,const char FAR * optval, int optlen)

s为欲设置属性的套接字。 level为套接字选项的级别,用法同上。 optname为设置选项的名称 optval为存放选项值的缓冲区指针。 optlen为缓冲区的长度

用法:

1int ttl=32; //设置TTL值
2int rc = setsockopt( s, IPPROTO_IP, IP_TTL, (char *)&ttl, sizeof(ttl));

套接字的使用步骤

1、启动Winsock:对Winsock DLL进行初始化,协商Winsock的版本支持并分配必要的资源。(服务器端和客户端)

1int WSAStartup( WORD wVersionRequested, LPWSADATA lpWSAData )
2wVersionRequested为打算加载Winsock的版本,一般如下设置:
3wVersionRequested=MAKEWORD(2,0)

或者直接赋值:

1wVersionRequested=2

LPWSADATA为初始化Socket后加载的版本的信息,定义如下:

1typedef struct WSAData {
2	WORD wVersion;
3	WORD wHighVersion;
4	char szDescription[WSADESCRIPTION_LEN+1];
5	char szSystemStatus[WSASYS_STATUS_LEN+1];
6	unsigned short iMaxSockets;
7	unsigned short iMaxUdpDg;
8	char FAR * lpVendorInfo;
9 } WSADATA, FAR * LPWSADATA;

如果加载成功后数据为:

wVersion=2 表示加载版本为2.0。 wHighVersion=514 表示当前系统支持socket最高版本为2.2。 szDescription=“WinSock 2.0” szSystemStatus=“Running” 表示正在运行。 iMaxSockets=0 表示同时打开的socket最大数,为0表示没有限制。 iMaxUdpDg=0 表示同时打开的数据报最大数,为0表示没有限制。 lpVendorInfo 没有使用,为厂商指定信息预留。

该函数使用方法:

1WORD wVersion=MAKEWORD(2,0);
2WSADATA wsData;
3int nResult= WSAStartup(wVersion,&wsData);
4if(nResult !=0)
5{
6  //错误处理
7}

2、创建套接字:(服务器端和客户端)

1SOCKET socket( int af, int type, int protocol );

af为网络地址类型,一般为AF_INET,表示在Internet域中使用。 type为套接字类型,前面已经介绍了。 protocol为指定网络协议,一般为IPPROTO_IP。 用法:

1SOCKET sock=socket(AF_INET,SOCK_STREAM,IPPROTO_IP);
2if(sock==INVALID_SOCKET)
3{
4//错误处理
5}

3、套接字的绑定:将本地地址绑定到所创建的套接字上。(服务器端和客户端)

1int bind( SOCKET s, const struct sockaddr FAR * name, int namelen )

s为已经创建的套接字。 name为socket地址结构,为sockaddr结构,如前面讨论的,我们一般使用sockaddr_in 结构,在使用再强制转换为sockaddr结构。 namelen为地址结构的长度。

用法:

1sockaddr_in addr;
2addr. sin_family=AF_INET;
3addr. sin_port= htons(0); //保证字节顺序
4addr. sin_addr.s_addr= inet_addr("192.1.8.84")
5int nResult=bind(s,(sockaddr*)&addr,sizeof(sockaddr));
6if(nResult==SOCKET_ERROR)
7{
8//错误处理
9}

4、 套接字的监听:(服务器端)

1int listen(SOCKET s, int backlog )

s为一个已绑定但未联接的套接字。 backlog为指定正在等待联接的最大队列长度,这个参数非常重要,因为服务器一般可 以提供多个连接。 用法:

1int nResult=listen(s,5) //最多5个连接
2if(nResult==SOCKET_ERROR)
3{
4//错误处理
5}

5、套接字等待连接::(服务器端)

1SOCKET accept( SOCKET s, struct sockaddr FAR * addr, int FAR * addrlen )

s为处于监听模式的套接字。 sockaddr为接收成功后返回客户端的网络地址。 addrlen为网络地址的长度。

用法:

1sockaddr_in addr;
2SOCKET s_d=accept(s,(sockaddr*)&addr,sizeof(sockaddr));
3if(s==INVALID_SOCKET)
4{
5//错误处理
6}

6、套接字的连结:将两个套接字连结起来准备通信。(客户端)

1int connect(SOCKET s, const struct sockaddr FAR * name, int namelen )

s为欲连结的已创建的套接字。 name为欲连结的socket地址。 namelen为socket地址的结构的长度。

用法:

1sockaddr_in addr;
2addr. sin_family=AF_INET;
3addr. sin_port=htons(0); //保证字节顺序
4addr. sin_addr.s_addr= htonl(INADDR_ANY) //保证字节顺序
5int nResult=connect(s,(sockaddr*)&addr,sizeof(sockaddr));
6if(nResult==SOCKET_ERROR)
7{
8//错误处理
9}

7、套接字发送数据:(服务器端和客户端)

1int send(SOCKET s, const char FAR * buf, int len, int flags )

s为服务器端监听的套接字。 buf为欲发送数据缓冲区的指针。 len为发送数据缓冲区的长度。 flags为数据发送标记。 返回值为发送数据的字符数。

◆这里讲一下这个发送标记,下面8中讨论的接收标记也一样:

flag取值必须为0或者如下定义的组合:0表示没有特殊行为。

1#define MSG_OOB 0x1 /* process out-of-band data */
2#define MSG_PEEK 0x2 /* peek at incoming message */
3#define MSG_DONTROUTE 0x4 /* send without using routing tables */

MSG_OOB表示数据应该带外发送,所谓带外数据就是TCP紧急数据。 MSG_PEEK表示使有用的数据复制到缓冲区内,但并不从系统缓冲区内删除。 MSG_DONTROUTE表示不要将包路由出去。

用法:

1char buf[]="xiaojin";
2int nResult=send(s,buf,strlen(buf));
3if(nResult==SOCKET_ERROR)
4{
5//错误处理
6}

8、 套接字的数据接收:(客户端)

1int recv( SOCKET s, char FAR * buf, int len, int flags )

s为准备接收数据的套接字。 buf为准备接收数据的缓冲区。 len为准备接收数据缓冲区的大小。 flags为数据接收标记。 返回值为接收的数据的字符数。

用法:

1char mess[1000];
2int nResult =recv(s,mess,1000,0);
3if(nResult==SOCKET_ERROR)
4{
5//错误处理
6}

9、中断套接字连接:通知服务器端或客户端停止接收和发送数据。(服务器端和客户端)

1int shutdown(SOCKET s, int how)

s为欲中断连接的套接字。 How为描述禁止哪些操作,取值为:SD_RECEIVE、SD_SEND、SD_BOTH。

1#define SD_RECEIVE 0x00
2#define SD_SEND 0x01
3#define SD_BOTH 0x02

用法:

1int nResult= shutdown(s,SD_BOTH);
2if(nResult==SOCKET_ERROR)
3{
4//错误处理
5}

10、 关闭套接字:释放所占有的资源。(服务器端和客户端)

1int closesocket( SOCKET s )

s为欲关闭的套接字。

用法:

1int nResult=closesocket(s);
2if(nResult==SOCKET_ERROR)
3{
4//错误处理
5}

与socket有关的一些函数介绍

1、读取当前错误值:每次发生错误时,如果要对具体问题进行处理,那么就应该调用这个函数取得错误代码。

1 int WSAGetLastError(void );
2 #define h_errno WSAGetLastError()

错误值请自己阅读Winsock2.h。

2、将主机的unsigned long值转换为网络字节顺序(32位):为什么要这样做呢?因为不同的计算机使用不同的字节顺序存储数据。因此任何从Winsock函数对IP地址和端口号的引用和传给Winsock函数的IP地址和端口号均时按照网络顺序组织的。

1u_long htonl(u_long hostlong);

举例:

1 htonl(0)=0
2 htonl(80)= 1342177280

3、将unsigned long数从网络字节顺序转换位主机字节顺序,是上面函数的逆函数。

1 u_long ntohl(u_long netlong);

举例:

1ntohl(0)=0
2 ntohl(1342177280)= 80

4、将主机的unsigned short值转换为网络字节顺序(16位):原因同2:

1 u_short htons(u_short hostshort);

举例:

1 htonl(0)=0
2 htonl(80)= 20480

5、将unsigned short数从网络字节顺序转换位主机字节顺序,是上面函数的逆函数。

1u_short ntohs(u_short netshort);

举例:

1ntohs(0)=0
2 ntohsl(20480)= 80

6、将用点分割的IP地址转换位一个in_addr结构的地址,这个结构的定义见笔记(一),实际上就是一个unsigned long值。计算机内部处理IP地址可是不认识如192.1.8.84之类的数据。

1unsigned long inet_addr( const char FAR * cp );

举例:

1 inet_addr("192.1.8.84")=1409810880
2 inet_addr("127.0.0.1")= 16777343

如果发生错误,函数返回INADDR_NONE值。

7、将网络地址转换位用点分割的IP地址,是上面函数的逆函数。

1char FAR * inet_ntoa( struct in_addr in );

举例:

1char * ipaddr=NULL;
2 char addr[20];
3 in_addr inaddr;
4 inaddr. s_addr=16777343;
5 ipaddr= inet_ntoa(inaddr);
6 strcpy(addr,ipaddr); 

这样addr的值就变为127.0.0.1。 注意意不要修改返回值或者进行释放动作。如果函数失败就会返回NULL值。

8、获取套接字的本地地址结构:

1int getsockname(SOCKET s, struct sockaddr FAR * name, int FAR * namelen );

s为套接字 name为函数调用后获得的地址值 namelen为缓冲区的大小。

9、获取与套接字相连的端地址结构:

1 int getpeername(SOCKET s, struct sockaddr FAR * name, int FAR * namelen );

s为套接字 name为函数调用后获得的端地址值 namelen为缓冲区的大小。

10、获取计算机名:

1 int gethostname( char FAR * name, int namelen );

name是存放计算机名的缓冲区 namelen是缓冲区的大小 用法:

1 char szName[255];
2 memset(szName,0,255);
3 if(gethostname(szName,255)==SOCKET_ERROR)
4 {
5     //错误处理
6 }

返回值为:szNmae="xiaojin"

11、根据计算机名获取主机地址:

1 struct hostent FAR * gethostbyname( const char FAR * name );

name为计算机名。 用法:

 1hostent * host;
 2 char* ip;
 3 host= gethostbyname("xiaojin");
 4 if(host->h_addr_list[0])
 5 {
 6   struct in_addr addr;
 7    memmove(&addr, host->h_addr_list[0]4);
 8   //获得标准IP地址
 9   ip=inet_ ntoa (addr);
10 }

返回值为:

1hostent->h_name="xiaojin"
2hostent->h_addrtype=2 //AF_INET
3hostent->length=4
4ip="127.0.0.1"

Winsock 的I/O操作:

1、 两种I/O模式 阻塞模式:执行I/O操作完成前会一直进行等待,不会将控制权交给程序。套接字 默认为阻塞模式。可以通过多线程技术进行处理。 非阻塞模式:执行I/O操作时,Winsock函数会返回并交出控制权。这种模式使用 起来比较复杂,因为函数在没有运行完成就进行返回,会不断地返回 WSAEWOULDBLOCK错误。但功能强大。 为了解决这个问题,提出了进行I/O操作的一些I/O模型,下面介绍最常见的三种:

2、select模型:

  通过调用select函数可以确定一个或多个套接字的状态,判断套接字上是否有数据,或 者能否向一个套接字写入数据。

1int select( int nfds, fd_set FAR * readfds, fd_set FAR * writefds, 
2 fd_set FAR *exceptfds, const struct timeval FAR * timeout );

◆先来看看涉及到的结构的定义: a、 d_set结构:

1#define FD_SETSIZE 64?
2typedef struct fd_set {
3 u_int fd_count; /* how many are SET? */
4 SOCKET fd_array[FD_SETSIZE]; /* an array of SOCKETs */
5 } fd_set;

fd_count为已设定socket的数量 fd_array为socket列表,FD_SETSIZE为最大socket数量,建议不小于64。这是微软建 议的。

B、timeval结构:

1struct timeval {
2 long tv_sec; /* seconds */
3 long tv_usec; /* and microseconds */
4 };

tv_sec为时间的秒值。 tv_usec为时间的毫秒值。 这个结构主要是设置select()函数的等待值,如果将该结构设置为(0,0),则select()函数 会立即返回。

◆再来看看select函数各参数的作用:

nfds:没有任何用处,主要用来进行系统兼容用,一般设置为0。 readfds:等待可读性检查的套接字组。 writefds;等待可写性检查的套接字组。 exceptfds:等待错误检查的套接字组。 timeout:超时时间。

函数失败的返回值: 调用失败返回SOCKET_ERROR,超时返回0。 readfds、writefds、exceptfds三个变量至少有一个不为空,同时这个不为空的套接字组 种至少有一个socket,道理很简单,否则要select干什么呢。

举例:测试一个套接字是否可读:

 1fd_set fdread;
 2//FD_ZERO定义
 3// #define FD_ZERO(set) (((fd_set FAR *)(set))->fd_count=0)
 4FD_ZERO(&fdread);
 5FD_SET(s,&fdread) //加入套接字,详细定义请看winsock2.h
 6if(select(0,%fdread,NULL,NULL,NULL)>0
 7{
 8  //成功
 9  if(FD_ISSET(s,&fread) //是否存在fread中,详细定义请看winsock2.h
10  {
11    //是可读的
12  }
13}

◆I/O操作函数:主要用于获取与套接字相关的操作参数。

1int ioctlsocket(SOCKET s, long cmd, u_long FAR * argp ); 

s为I/O操作的套接字。 cmd为对套接字的操作命令。 argp为命令所带参数的指针。

常见的命令:

1//确定套接字自动读入的数据量
2#define FIONREAD _IOR(''''f'''', 127, u_long) /* get # bytes to read */
3//允许或禁止套接字的非阻塞模式,允许为非0,禁止为0
4#define FIONBIO _IOW(''''f'''', 126, u_long) /* set/clear non-blocking i/o */
5//确定是否所有带外数据都已被读入
6#define SIOCATMARK _IOR(''''s'''', 7, u_long) /* at oob mark? */

3、WSAAsynSelect模型: WSAAsynSelect模型也是一个常用的异步I/O模型。应用程序可以在一个套接字上接收以 WINDOWS消息为基础的网络事件通知。该模型的实现方法是通过调用WSAAsynSelect函 数 自动将套接字设置为非阻塞模式,并向WINDOWS注册一个或多个网络时间,并提供一 个通知时使用的窗口句柄。当注册的事件发生时,对应的窗口将收到一个基于消息的通知。

1 int WSAAsyncSelect( SOCKET s, HWND hWnd, u_int wMsg, long lEvent); 

s为需要事件通知的套接字 hWnd为接收消息的窗口句柄 wMsg为要接收的消息 lEvent为掩码,指定应用程序感兴趣的网络事件组合,主要如下:

 1#define FD_READ_BIT 0
 2#define FD_READ (1 << FD_READ_BIT)
 3#define FD_WRITE_BIT 1
 4#define FD_WRITE (1 << FD_WRITE_BIT)
 5#define FD_OOB_BIT 2
 6#define FD_OOB (1 << FD_OOB_BIT)
 7#define FD_ACCEPT_BIT 3
 8#define FD_ACCEPT (1 << FD_ACCEPT_BIT)
 9#define FD_CONNECT_BIT 4
10#define FD_CONNECT (1 << FD_CONNECT_BIT)
11#define FD_CLOSE_BIT 5
12#define FD_CLOSE (1 << FD_CLOSE_BIT)

用法:要接收读写通知:

1int nResult= WSAAsyncSelect(s,hWnd,wMsg,FD_READ|FD_WRITE)
2if(nResult==SOCKET_ERROR)
3{
4  //错误处理
5}

取消通知:

1 int nResult= WSAAsyncSelect(s,hWnd,00) 

当应用程序窗口hWnd收到消息时,wMsg.wParam参数标识了套接字,lParam的低字标明 了网络事件,高字则包含错误代码。

4、WSAEventSelect模型 WSAEventSelect模型类似WSAAsynSelect模型,但最主要的区别是网络事件发生时会被发 送到一个事件对象句柄,而不是发送到一个窗口。

使用步骤如下: a、 创建事件对象来接收网络事件:

1#define WSAEVENT HANDLE
2#define LPWSAEVENT LPHANDLE
3WSAEVENT WSACreateEvent( void );

该函数的返回值为一个事件对象句柄,它具有两种工作状态:已传信(signaled)和未传信 (nonsignaled)以及两种工作模式:人工重设(manual reset)和自动重设(auto reset)。默认未 未传信的工作状态和人工重设模式。

b、将事件对象与套接字关联,同时注册事件,使事件对象的工作状态从未传信转变未 已传信。

1 int WSAEventSelect( SOCKET s,WSAEVENT hEventObject,long lNetworkEvents ); 

s为套接字 hEventObject为刚才创建的事件对象句柄 lNetworkEvents为掩码,定义如上面所述

c、I/O处理后,设置事件对象为未传信

1BOOL WSAResetEvent( WSAEVENT hEvent );

Hevent为事件对象

成功返回TRUE,失败返回FALSE。

d、等待网络事件来触发事件句柄的工作状态:

1DWORD WSAWaitForMultipleEvents( DWORD cEvents,const WSAEVENT FAR * lphEvents, BOOL fWaitAll,DWORD dwTimeout, BOOL fAlertable );

lpEvent为事件句柄数组的指针 cEvent为为事件句柄的数目,其最大值为WSA_MAXIMUM_WAIT_EVENTS fWaitAll指定等待类型:TRUE:当lphEvent数组重所有事件对象同时有信号时返回; FALSE:任一事件有信号就返回。 dwTimeout为等待超时(毫秒) fAlertable为指定函数返回时是否执行完成例程

对事件数组中的事件进行引用时,应该用WSAWaitForMultipleEvents的返回值,减去 预声明值WSA_WAIT_EVENT_0,得到具体的引用值。例如:

1nIndex=WSAWaitForMultipleEvents();
2MyEvent=EventArray[Index- WSA_WAIT_EVENT_0];

e、判断网络事件类型:

1int WSAEnumNetworkEvents( SOCKET s,
2WSAEVENT hEventObject, LPWSANETWORKEVENTS lpNetworkEvents );
3s为套接字
4hEventObject为需要重设的事件对象
5lpNetworkEvents为记录网络事件和错误代码,其结构定义如下:
6typedef struct _WSANETWORKEVENTS {
7  long lNetworkEvents;
8  int iErrorCode[FD_MAX_EVENTS];
9} WSANETWORKEVENTS, FAR * LPWSANETWORKEVENTS;

f、关闭事件对象句柄:

1BOOL WSACloseEvent(WSAEVENT hEvent);

调用成功返回TRUE,否则返回FALSE。

本文转自 http://www.cppblog.com/xushaohua/articles/37653.html,已经对代码做过Markdown格式整理,如有侵权,请告之删除。