WinSocket模型的探讨——Overlapped模型

时间：2021-10-16 00:07:27
重叠模型是Windows里一种重要的 I/O 模型，可以有效率的实现一些 I/O 操作，譬如文件读写、Socket读写等.

重叠模型是Windows里一种重要的 I/O 模型，可以有效率的实现一些 I/O 操作，譬如文件读写、Socket读写等，在这里我们一起来研究一下重叠模型，看看它究竟为何方神圣。

这篇文章分为以下几部分来说明：

  1. 重叠模型的概念
  2. 容易碰到的一些问题
  3. 重叠模型的基本编程方法（accept 和 AcceptEx ）
  4. 突破64个事件的等待限制
  5. 例程

好了，下面就让我们一起来学习一下重叠模型。

1、概念

对于重叠模型的概念，大家都各有说法，以我自己的角度来说，我觉得重叠其实就是一种异步处理的说法，一边向socket 投递操作，而在另一边等待结果的完成，两边互不相干，我想这就是重叠的概念。其实这个概念也不用深究，我们还是来看看如何使用重叠模型。

2、容易碰到的一些问题

为什么对Socket投递第一个WSARecv()出错并且返回错误为10045?

这个问题都是因为这个 WSARecv 的lpFlags参数引起的，这个参数一定要设置为0，而不是直接写为0，
 如果直接写为0，则会出现10014错误。要这样写：

DWORD dwFlags=0;
WSARecv(sClient, &as.m_wsaBuf, as.m_wsaBuf.len, &dwRecvBytes, &dwFlags, &as.m_Overlapped, NULL);

为什么找不到AccepctEx的声明和定义?

要使用 AcceptEx，必须加上下面两句

#include <mswsock.h>
#pragma comment(lib, "mswsock.lib")

AcceptEx 的第四个参数有什么用？

  AcceptEx 的第四个参数，指定建立连接的时候，接收客户端传过来的多少字节的数据，可以用来做验证客户端的字符串，如果这个参数为0，则AcceptEx在接受客户端连接后会立刻完成，不会接受任何客户端的数据。如果不为0，则接收客户端传过来的指定字节的数据。但这样会出现恶意连接连进来的时候，不发送任何数据，就导致了SERVER出错。

为什么程序只能收到连接，收不到从Socket传过来的数据？

程序不停的收到连接，但传过来的数据都不显示，这个问题的解决方法是：把事件数组的第一个设为和侦听Socket (sListen) 对应的事件，就是说第一个事件是给侦听Socket的，那在接受到连接后，要重新设置一下第一个事件：WSASetEvent(EventArray[0]); 以便工作线程可以继续往下走去服务其他事件。这里只知道要这样做，但具体原因不太清楚，希望高人能指点一下！

程序接收到的数据在哪里？

在使用WSARecv的时候，我们就已经传了一个WSABUF类型的变量进去，那在后面调用WSAGetOverlappedResult 之后，之前的WSABUF中的buf就是取得数据的地方，这些更仔细的说明建议大家看看Piggy的《手把手教你写 Overlapped 模型》，这篇文章对函数进行了比较详细的介绍。

3、基本编程方法：使用重叠模型，就要注意以下的一些函数。

WSASocket、AcceptEx、WSACreateEvent、WSAWaitForMultipleEvents、WSAResetEvent、WSAGetOverlappedResult、WSARecv、WSASetEvent。具体使用方法各位可以看下面的例子或者查一下MSDN，里面有介绍。

首先我们先看看重叠模型的使用步骤：

  • 创建一个带Overlapped标志的Socket句柄（其实也可以是文件句柄）；
  • 准备好一个与这个Socket句柄对应的Overlapped对象，并准备好事件句柄，以便让后面的WaitForXXX函数使用
  • 使用支持Overlapped的函数对上面的句柄作操作（向这个句柄投递请求），这些函数都有一个共同点，就是它们都有个参数是Overlapped类型的。例如AcceptEx、WriteFile、WSARecv等；
  • 在一个循环或者一个线程中使用WSAWaitForMultipleEvents来等待事件的发生；
  • 事件发生后，使用WSAGetOverlappedResult 来取得对应的结果并做处理。
  • 继续向句柄投递操作请求（向它发东西啊，收东西啊等等的！）。

到这里，给大家一个建议，对着上面的步骤写写练习，要看代码网上一堆，真要理解，还是得亲自动手才行，编程技术，动手是硬道理！

好，我们一步步来看看代码片断，对上面的步骤有点理性的认识才行。

创建带Overlapped标志的Socket句柄：

 SOCKET sListen;
 if((sListen = WSASocket(AF_INET, SOCK_STREAM, 0, NULL, 0, WSA_FLAG_Overlapped)) == INVALID_SOCKET)
 {
  OutErr("WSASocket error!");  
  return;
 }

准备好一个Overlapped对象，还有hEvent事件：

 // 准备好一个给 listenSocket 使用的 Overlapped
 ZeroMemory(&ListenOverlapped, sizeof(Overlapped));
 ListenOverlapped.hEvent = WSACreateEvent();

对这个句柄投递操作，这是一个侦听句柄，当然就是给他投递AcceptEx操作：

 AcceptEx(sListen, sClient, (PVOID) szAcceptBuffer, 0,
    sizeof(SOCKADDR_IN) + 16, sizeof(SOCKADDR_IN) + 16, &dwRecvBytes,
    &ListenOverlapped);

在一个循环或者线程中等待事件的完成，并取得结果做处理：

 while(TRUE)
 {
  // 等待侦听事件的发生
  nIdx = WSAWaitForMultipleEvents(1, &ListenOverlapped.hEvent, FALSE, INFINITE, FALSE);

  // 取得这次Overlapped的结果
  WSAGetOverlappedResult(sListen, &ListenOverlapped, &dwTransferred, FALSE, &dwFlags);

  cout << sClient << " 连接上来了！" << endl;
  cout << "连接总数为：" << ++g_nClientCnt << endl;
  // 重置一下事件
  WSAResetEvent(ListenOverlapped.hEvent);

  // 再重新向ListenSocket投递一个AcceptEx请求
  sClient = WSASocket(AF_INET, SOCK_STREAM, 0, NULL, 0, WSA_FLAG_Overlapped);
  ZeroMemory(szAcceptBuffer, 2*(sizeof(SOCKADDR_IN) + 16) );
  AcceptEx(sListen, sClient, (PVOID) szAcceptBuffer, dwRecvData,
    sizeof(SOCKADDR_IN) + 16, sizeof(SOCKADDR_IN) + 16, &dwRecvBytes,
    &ListenOverlapped);
 }

上面几步，基本上实现了AcceptEx的接受连接的整个过程，如果你都能看懂，那下面的例子程序也难不倒你了！

 4、突破64个事件的等待限制

虽说使用完成端口可以不受64个事件的限制，但我们这里既然是抱着钻研的态度，那总是要试一试的。

64个事件的等待限制是由Windows规定的，如果超过了64个，则WSAWaitForMultipleEvents函数会报错，错误码为87。要突破此限制的最基本的思路就是当线程等待的事件数超过了64，则新开线程来等待，把事件以64个为一个区，则0区、1区、2区。。。每区有64个事件，每个线程服务于一个区。这里要注意的是每个线程中的WaitForMultipleEvents要小心事件的个数、事件的起始位。详细的可以看下面的例子。

5、例程，下面的程序都是一个cpp文件拷贝过去就可以直接编译了。

 /*

  Server1: 使用 accept 来得到连接，并使用ProcessIO来处理结果，这里使用64个事件为一区，一个线程服务一个区的方法来突破了
  64个事件的限制。

*/

#include <winsock2.h>
#pragma comment(lib, "WS2_32")

#include <mswsock.h>
#pragma comment(lib, "mswsock.lib")

#include <iostream>
using namespace std;

 

#define PORT 5050
#define DATA_BUFSIZE 8192
#define DYL_MAXIMUM_WAIT_EVENTS 1024

#define OutErr(a) cout << (a) << endl
      << "出错代码：" << WSAGetLastError() << endl
      << "出错文件：" << __FILE__ << endl  
      << "出错行数：" << __LINE__ << endl

#define OutMsg(a) cout << (a) << endl;

 

void InitWinsock()
{
 // 初始化WINSOCK
 WSADATA wsd;
 if( WSAStartup(MAKEWORD(2, 2), &wsd) != 0)
 
}

SOCKET BindServer(int nPort)
{
 // 创建socket
 SOCKET sServer = WSASocket(AF_INET, SOCK_STREAM, 0, NULL, 0, WSA_FLAG_Overlapped);

 // 绑定端口
 struct sockaddr_in servAddr;
 servAddr.sin_family = AF_INET;
 servAddr.sin_port = htons(nPort);
 servAddr.sin_addr.s_addr = htonl(INADDR_ANY);

 if(bind(sServer, (struct sockaddr *)&servAddr, sizeof(servAddr)) < 0)
 {
  OutErr("bind Failed!");
  return NULL;
 }

 // 设置监听队列为200
 if(listen(sServer, WSA_MAXIMUM_WAIT_EVENTS) != 0)
 {
  OutErr("listen Failed!");
  return NULL;
 }
 return sServer;
}


DWORD EventTotal = 0;

WSAEVENT EventArray[DYL_MAXIMUM_WAIT_EVENTS];

typedef struct _SOCKET_INFORMATION {
   CHAR Buffer[DATA_BUFSIZE];
   WSABUF DataBuf;
   SOCKET Socket;
   WSAOverlapped Overlapped;
} SOCKET_INFORMATION, * LPSOCKET_INFORMATION;

LPSOCKET_INFORMATION SocketInfoArray[DYL_MAXIMUM_WAIT_EVENTS];

void CreateSocketInfo(SOCKET sClient)
{
 //if(EventTotal%WSA_MAXIMUM_WAIT_EVENTS)
 // EventTotal++;
 SocketInfoArray[EventTotal] = new SOCKET_INFORMATION();
 SocketInfoArray[EventTotal]->Socket = sClient;
 ZeroMemory(&SocketInfoArray[EventTotal]->Overlapped, sizeof(WSAOverlapped));
 SocketInfoArray[EventTotal]->Overlapped.hEvent = EventArray[EventTotal] = WSACreateEvent();
 SocketInfoArray[EventTotal]->DataBuf.buf = SocketInfoArray[EventTotal]->Buffer;
 SocketInfoArray[EventTotal]->DataBuf.len = DATA_BUFSIZE;

 memset(SocketInfoArray[EventTotal]->Buffer, 0, DATA_BUFSIZE);

 EventTotal++;
}

// 释放，注意N从0开始
void DestroySocketInfo(int n)
{
 delete SocketInfoArray[n];
 if( (n+1) != DYL_MAXIMUM_WAIT_EVENTS )
 {
  for(int i = n; i < EventTotal; i++)
  {
   SocketInfoArray[i] = SocketInfoArray[i + 1];
  }
 }
 EventTotal--;
}

DWORD WINAPI ProcessIO(LPVOID lpParam)
{
 int nSec = *((int*)lpParam);
 int nIdx = 0;
 DWORD BytesTransferred;
 DWORD Flags;
 int nTriggerCnt;

 while(1)
 {
  // 计算好事件数
  int nEvtCnt;
  if( EventTotal % WSA_MAXIMUM_WAIT_EVENTS == 0)
   nEvtCnt = 64;
  else
   nEvtCnt = EventTotal % WSA_MAXIMUM_WAIT_EVENTS;

  // 计算好事件数组的起始位置
  WSAEVENT *evtStart = EventArray + nSec*WSA_MAXIMUM_WAIT_EVENTS;
  if( (nIdx = WSAWaitForMultipleEvents(nEvtCnt, evtStart, FALSE, WSA_INFINITE, FALSE) ) == WSA_WAIT_FAILED )
  {
   OutErr("WSAWaitForMultipleEvents failed!");
   return 0;
  }

  nTriggerCnt = nIdx - WSA_WAIT_EVENT_0;

  //!! 如果是0，则说明有新的客户要连接进来
  if ( nTriggerCnt == 0 )
  
  
  nTriggerCnt = nTriggerCnt + nSec*WSA_MAXIMUM_WAIT_EVENTS;
  LPSOCKET_INFORMATION SI = SocketInfoArray[nTriggerCnt];
  WSAResetEvent(EventArray[nTriggerCnt]);
  
  // 取得重叠操作的结果
  if(WSAGetOverlappedResult(SI->Socket, &(SI->Overlapped), &BytesTransferred, FALSE, &Flags) == FALSE || BytesTransferred == 0)
  {
   //OutErr("WSAGetOverlappedResult failed!");
   cout << "socket " << SI->Socket << " 断开！" << endl;
   DestroySocketInfo(nTriggerCnt);

   continue;
  }

  cout << SI->Socket << " 发送过来的数据："  << SI->Buffer << endl;

  // 继续等待接收数据
  DWORD dwRecv;
  DWORD dwFlags = 0;

  if(WSARecv(SI->Socket, &(SI->DataBuf), 1, &dwRecv, &dwFlags, &(SI->Overlapped), NULL)  == SOCKET_ERROR)
  {
   if (WSAGetLastError() != ERROR_IO_PENDING)
   {
      OutErr("WSARecv error!");
      return 0;
   }   
  }
 }

 cout << "ProcessIO end!" << endl;

 return 0;
}

int g_nThreadCnt = 1; ///< 标示当前开了多少个线程来服务于

void main()
{
 InitWinsock();

 SOCKET sListen = BindServer(PORT);

 char szBuf[200];
 memset(szBuf, 0, 200);

 SOCKET sClient;
 if((sClient = WSASocket(AF_INET, SOCK_STREAM, 0, NULL, 0, WSA_FLAG_Overlapped)) == INVALID_SOCKET)
 {
  OutErr("WSASocket error!");  
  return;
 }

 if ((EventArray[0] = WSACreateEvent()) == WSA_INVALID_EVENT)
 {
  OutErr("WSACreateEvent error!");
  return;
 }

 EventTotal = 1;

 int nSec = 0;
 CreateThread(NULL, 0, ProcessIO, (LPVOID)&nSec, 0, NULL);

 while(1)
 {
  sClient = accept(sListen, NULL, NULL);
  if(sClient == INVALID_SOCKET)
  {
   OutErr("accept faild!");
   return;
  }

  // 先判断是否能连接，是否超过了最大连接数

  if(EventTotal + 1 > DYL_MAXIMUM_WAIT_EVENTS)
  {
   cout << "Too many Client! closing socket" << sClient << endl;
   closesocket(sClient);
   continue;
  }

  cout << sClient << " 连接进来!" << endl;
  cout << "客户端连接数：" << EventTotal + 1 << endl;

  /// 这里要做身份验证，不准其他客户端连接上来

  // 判断是否需要分区
  if( int( EventTotal/WSA_MAXIMUM_WAIT_EVENTS ) + 1 > g_nThreadCnt)
  {
   nSec = int((EventTotal + 1)/WSA_MAXIMUM_WAIT_EVENTS);
   EventArray[nSec * WSA_MAXIMUM_WAIT_EVENTS] = EventArray[0];
   EventTotal++;
   CreateThread(NULL, 0, ProcessIO, (LPVOID)&nSec, 0, NULL);
   g_nThreadCnt++;
  }
  
  CreateSocketInfo(sClient);

  DWORD dwRecv;
  DWORD dwFlags = 0;
  int nCurCnt = EventTotal - 1;

  if( WSARecv(sClient, &(SocketInfoArray[EventTotal-1]->DataBuf), 1, &dwRecv, &dwFlags,
     &(SocketInfoArray[EventTotal-1]->Overlapped), NULL) == SOCKET_ERROR )
  {
   if(WSAGetLastError() != WSA_IO_PENDING)
   {
    OutErr("WSARecv Error!");

    // 那就只能释放相关资源了  
    DestroySocketInfo(nCurCnt);
   }
  }

  if(int(EventTotal/WSA_MAXIMUM_WAIT_EVENTS) == 0)
   WSASetEvent(EventArray[0]);
  else
  {
   //!! 这里要设置一下0事件，让工作线程去服务事件数组里的其他事件
   for(int i = 0 ; i < int(EventTotal/WSA_MAXIMUM_WAIT_EVENTS); i++)
   {
    if (WSASetEvent(EventArray[i*WSA_MAXIMUM_WAIT_EVENTS]) == FALSE)
    {
     OutErr("WSASetEvent failed!");
     return;
    }
   }
  }
 }
 
}

/**
Server2.cpp: 使用了 AcceptEx 来处理连接
**/

/// 包含头文件
#include <winsock2.h>
#pragma comment(lib, "WS2_32")

#include <mswsock.h>
#pragma comment(lib, "mswsock.lib")

#include <windows.h>

#include <iostream>
using namespace std;


/// 宏定义
#define PORT 5050
#define DATA_BUFSIZE 8192

#define OutErr(a) cout << (a) << endl
      << "出错代码：" << WSAGetLastError() << endl
      << "出错文件：" << __FILE__ << endl  
      << "出错行数：" << __LINE__ << endl

#define OutMsg(a) cout << (a) << endl;


/// 全局变量定义
DWORD EventTotal = 0;

// #define WSA_MAXIMUM_WAIT_EVENTS 100

WSAEVENT EventArray[WSA_MAXIMUM_WAIT_EVENTS];

typedef struct _SOCKET_INFORMATION {
   CHAR Buffer[DATA_BUFSIZE];
   WSABUF DataBuf;
   SOCKET Socket;
   WSAOverlapped Overlapped;
} SOCKET_INFORMATION, * LPSOCKET_INFORMATION;

WSAOverlapped ListenOverlapped;

LPSOCKET_INFORMATION SocketInfoArray[WSA_MAXIMUM_WAIT_EVENTS];

/// 全局函数定义
void InitWinsock()
{
 // 初始化WINSOCK
 WSADATA wsd;
 if( WSAStartup(MAKEWORD(2, 2), &wsd) != 0)
 
}

SOCKET BindServer(int nPort)
{
 // 创建socket
 SOCKET sServer = WSASocket(AF_INET, SOCK_STREAM, 0, NULL, 0, WSA_FLAG_Overlapped);

 // 绑定端口
 struct sockaddr_in servAddr;
 servAddr.sin_family = AF_INET;
 servAddr.sin_port = htons(nPort);
 servAddr.sin_addr.s_addr = htonl(INADDR_ANY);

 if(bind(sServer, (struct sockaddr *)&servAddr, sizeof(servAddr)) < 0)
 {
  OutErr("bind Failed!");
  return NULL;
 }

 // 设置监听队列为200
 if(listen(sServer, WSA_MAXIMUM_WAIT_EVENTS) != 0)
 {
  OutErr("listen Failed!");
  return NULL;
 }
 return sServer;
}

void CreateSocketInfo(SOCKET sClient)
{
 SocketInfoArray[EventTotal] = new SOCKET_INFORMATION;
 SocketInfoArray[EventTotal]->Socket = sClient;
 ZeroMemory(&SocketInfoArray[EventTotal]->Overlapped, sizeof(WSAOverlapped));
 SocketInfoArray[EventTotal]->Overlapped.hEvent = EventArray[EventTotal] = WSACreateEvent();
 SocketInfoArray[EventTotal]->DataBuf.buf = SocketInfoArray[EventTotal]->Buffer;
 SocketInfoArray[EventTotal]->DataBuf.len = DATA_BUFSIZE;

 memset(SocketInfoArray[EventTotal]->Buffer, 0, DATA_BUFSIZE);

 EventTotal++;
}

// 释放，注意N从0开始
void DestroySocketInfo(int n)
{
 delete SocketInfoArray[n];
 if( (n+1) != WSA_MAXIMUM_WAIT_EVENTS )
 {
  for(int i = n; i < EventTotal; i++)
  {
   SocketInfoArray[i] = SocketInfoArray[i + 1];
  }
 }
 EventTotal--;
}


// 正文
void main()
{
 InitWinsock();

 SOCKET sListen, sClient;

 sListen = BindServer(PORT);

 // 准备好连接的SOCKET
 sClient = WSASocket(AF_INET, SOCK_STREAM, 0, NULL, 0, WSA_FLAG_Overlapped);
 if(sClient == SOCKET_ERROR)
 {
  OutErr("WSASocket error!");
  return;
 }

 // 给这个Listen SOCKET分配一个Overlapped，可以把这个SOCKET 想象为其他的句柄，例如文件句柄
 ZeroMemory(&ListenOverlapped, sizeof(Overlapped));
 if ((EventArray[EventTotal++] = ListenOverlapped.hEvent = WSACreateEvent()) == WSA_INVALID_EVENT)
 {
  OutErr("WSACreateEvent failed");
  return;
 }

 // 投递一个 AcceptEx 请求，注意这里的AcceptBuffer参数是指：取得的第一块从CLIENT发来的数据
 // 这块数据包括：1、本地的地址 2、连接端的地址 3、这些地址是写在这个缓冲区的后面一段的（具体可再参看MSDN）
 CHAR AcceptBuffer[2 * (sizeof(SOCKADDR_IN) + 16)];
 DWORD dwRecvBytes;
 if (AcceptEx(sListen, sClient, (PVOID) AcceptBuffer, 0,
  sizeof(SOCKADDR_IN) + 16, sizeof(SOCKADDR_IN) + 16, &dwRecvBytes,
  &ListenOverlapped) == FALSE)
 {
  if (WSAGetLastError() != ERROR_IO_PENDING)
  {
   OutErr("AcceptEx failed");
   return;
  }
 }

 // 真正的工作开始了，在下面可以异步去处理Accept、WSARecv、WSASend等请求
 int nIdx;
 DWORD BytesTransferred;
 DWORD Flags;
 while(TRUE)
 {
  // 首先等待事件的发生
  if( (nIdx = WSAWaitForMultipleEvents(EventTotal, EventArray, FALSE, INFINITE, FALSE)) == WSA_WAIT_FAILED )
  {
   OutErr("WSAWaitForMultipleEvents Error!");
   return;
  }

  // 处理连接
  if( (nIdx - WSA_WAIT_EVENT_0) == 0)
  {
   // 先看看什么结果，反正 WSAWaitForMultipleEvents 调用完成，就肯定是有某种事情发生了
   if(WSAGetOverlappedResult(sListen, &ListenOverlapped, &BytesTransferred, FALSE, &Flags) == FALSE)
   {
    OutErr("nIdx 为 0，WSAGetOverlappedResult error!");
    return;
   }
  
   if (EventTotal >= WSA_MAXIMUM_WAIT_EVENTS)
   {
    cout << "Too many connections - closing socket." << endl;
    closesocket(sClient);

    goto ReAccept;
   }
   else
   {
    // cout << AcceptBuffer << endl;
    cout << "socket " << sClient << " 连接进来" << endl;
    cout << "连接客户端总数 " << EventTotal << endl;

    // 创建相应的结构
    CreateSocketInfo(sClient);

    // 向刚进来的连接 投递一个WSARecv操作
    Flags = 0;
    if ( WSARecv(sClient, &(SocketInfoArray[EventTotal-1]->DataBuf), 1, &dwRecvBytes, &Flags,
     &(SocketInfoArray[EventTotal-1]->Overlapped), NULL) == SOCKET_ERROR)
    {
     if (WSAGetLastError() != ERROR_IO_PENDING)
     {
      OutErr("WSARecv error!");
      return;
     }
    }
   }

ReAccept:
   // 再重新向ListenSocket投递一个AcceptEx请求
   sClient = WSASocket(AF_INET, SOCK_STREAM, 0, NULL, 0, WSA_FLAG_Overlapped);
   if(sClient == INVALID_SOCKET )
   {
    OutErr("WSASocket Error!");
    return;
   }

   // 重置一下事件
   if(!WSAResetEvent(EventArray[0]))
   {
    OutErr("WSAResetEvent error!");
    return;
   }

   ZeroMemory(&ListenOverlapped, sizeof(Overlapped));
   ListenOverlapped.hEvent = EventArray[0];
   
   if (AcceptEx(sListen, sClient, (PVOID) AcceptBuffer, 0,
    sizeof(SOCKADDR_IN) + 16, sizeof(SOCKADDR_IN) + 16, &dwRecvBytes,
    &ListenOverlapped) == FALSE)
   {
    if (WSAGetLastError() != ERROR_IO_PENDING)
    {
     OutErr("AcceptEx Error!");
     return;
    }
   } 
   
   continue;
  }

  // 上面搞了一大通，终于可以开始处理真正的客户端请求了
  LPSOCKET_INFORMATION SI;
  SI = SocketInfoArray[nIdx - WSA_WAIT_EVENT_0];
  WSAResetEvent(SocketInfoArray[nIdx - WSA_WAIT_EVENT_0]);

  // 取得结果
  if (WSAGetOverlappedResult(SI->Socket, &(SI->Overlapped), &BytesTransferred, FALSE, &Flags) == FALSE
   || BytesTransferred == 0)
  {
   cout << SI->Socket << " 断开" << endl;
   DestroySocketInfo(nIdx - WSA_WAIT_EVENT_0);
   continue;
  }   

  cout << SocketInfoArray[nIdx - WSA_WAIT_EVENT_0]->Socket << " 发过来的数据：" << SocketInfoArray[nIdx - WSA_WAIT_EVENT_0]->Buffer << endl;

    // 再投递新的接收请求
  Flags = 0;
  if (WSARecv(SI->Socket, &(SI->DataBuf), 1, &dwRecvBytes, &Flags,
   &(SI->Overlapped), NULL) == SOCKET_ERROR)
  {
   if (WSAGetLastError() != ERROR_IO_PENDING)
   {
    OutErr("WSARecv() failed");
    return;
   }
  }  
 } 
}


/**
测试客户端

  Client1.cpp:
  1、服务器断开重连接
  2、测试单个客户端连接对话

**/

#include <winsock2.h>
#pragma comment(lib, "WS2_32")

#include <iostream>
using namespace std;

#define PORT 5050

#define OutErr(a) cout << (a) << endl
      << "出错代码：" << WSAGetLastError() << endl
      << "出错文件：" << __FILE__ << endl  
      << "出错行数：" << __LINE__ << endl

#define OutMsg(a) cout << (a) << endl;

void InitWinsock()
{
 // 初始化WINSOCK
 WSADATA wsd;
 if( WSAStartup(MAKEWORD(2, 2), &wsd) != 0)
 
}

SOCKET ConnectServer(char *lpszServerIP, int nPort, ULONG NonBlock)
{
 SOCKET sServer = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
 //ioctlsocket(sServer, FIONBIO, &NonBlock);
 
 struct hostent *pHost = NULL;
 struct sockaddr_in servAddr;
 servAddr.sin_family = AF_INET;
 servAddr.sin_port = htons(nPort);
 servAddr.sin_addr.s_addr = inet_addr(lpszServerIP);


 // 如果给的是主机的名字而不是IP地址
 if(servAddr.sin_addr.s_addr == INADDR_NONE)
 {
  pHost = gethostbyname( lpszServerIP );
  if(pHost == NULL)
  {
   OutErr("gethostbyname Failed!");
   return NULL;
  }
  memcpy(&servAddr.sin_addr, pHost->h_addr_list[0], pHost->h_length);
 }

 int nRet = 0;
 nRet = connect(sServer, (struct sockaddr *)&servAddr, sizeof(servAddr));
 if( nRet == SOCKET_ERROR )
 {
  OutErr("connect failed!");
  return NULL;
 }
  
 return sServer;
}


void main()
{

 InitWinsock();

connect:
 SOCKET sServer = ConnectServer("127.0.0.1", PORT, 0);


reSay:
 char szBuf[1024];
 memset(szBuf, 0, 1024);
 cout << "你要说：";
 cin >> szBuf; 

 if(send(sServer, szBuf, strlen(szBuf), 0) == SOCKET_ERROR)
  goto connect;
 else
  goto reSay;
}

 /*
测试客户端2
 Client2.cpp:测试多个客户端连接
  */

#include <winsock2.h>
#pragma comment(lib, "WS2_32")


#include <iostream>
using namespace std;

#define PORT 5050

#define OutErr(a) cout << (a) << endl
      << "出错代码：" << WSAGetLastError() << endl
      << "出错文件：" << __FILE__ << endl  
      << "出错行数：" << __LINE__ << endl

#define OutMsg(a) cout << (a) << endl;


// 全局函数定义
void InitWinsock()
{
 // 初始化WINSOCK
 WSADATA wsd;
 if( WSAStartup(MAKEWORD(2, 2), &wsd) != 0)
 
}

SOCKET ConnectServer(char *lpszServerIP, int nPort, ULONG NonBlock)
{
 SOCKET sServer = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
 //ioctlsocket(sServer, FIONBIO, &NonBlock);
 
 struct hostent *pHost = NULL;
 struct sockaddr_in servAddr;
 servAddr.sin_family = AF_INET;
 servAddr.sin_port = htons(nPort);
 servAddr.sin_addr.s_addr = inet_addr(lpszServerIP);


 // 如果给的是主机的名字而不是IP地址
 if(servAddr.sin_addr.s_addr == INADDR_NONE)
 {
  pHost = gethostbyname( lpszServerIP );
  if(pHost == NULL)
  {
   OutErr("gethostbyname Failed!");
   return NULL;
  }
  memcpy(&servAddr.sin_addr, pHost->h_addr_list[0], pHost->h_length);
 }

 int nRet = 0;
 nRet = connect(sServer, (struct sockaddr *)&servAddr, sizeof(servAddr));
 if( nRet == SOCKET_ERROR )
 {
  OutErr("connect failed!");
  return NULL;
 }
  
 return sServer;
}


void main()
{
 InitWinsock();

 SOCKET sServer[64];

 for(int i = 0; i < 63; i++)
 {
  sServer[i] = ConnectServer("127.0.0.1", PORT, 0);
  
  if(sServer[i] != NULL)
  {
   send(sServer[i], "hello", 6, 0);
  }
  Sleep(100);  
 }

 while(1)
 
}

 把上面的几个CPP文件都拷贝到自己的工程里就可以编译通过。

标签：

相关文章