c++ iocp

// IOCP_TCPIP_Socket_Server.cpp
 
#include <WinSock2.h>
#include <Windows.h>
#include <vector>
#include <iostream>
 
using namespace std;
 
#pragma comment(lib, "Ws2_32.lib")  // Socket编程需用的动态链接库
#pragma comment(lib, "Kernel32.lib") // IOCP需要用到的动态链接库
 
/**
 * 结构体名称：PER_IO_DATA
 * 结构体功能：重叠I/O需要用到的结构体，临时记录IO数据
 **/
const int DataBuffSize  = 2 * 1024;
typedef struct
{
 OVERLAPPED overlapped;
 WSABUF databuff;
 char buffer[ DataBuffSize ];
 int BufferLen;
 int operationType;
}PER_IO_OPERATEION_DATA, *LPPER_IO_OPERATION_DATA, *LPPER_IO_DATA, PER_IO_DATA;
 
/**
 * 结构体名称：PER_HANDLE_DATA
 * 结构体存储：记录单个套接字的数据，包括了套接字的变量及套接字的对应的客户端的地址。
 * 结构体作用：当服务器连接上客户端时，信息存储到该结构体中，知道客户端的地址以便于回访。
 **/
typedef struct
{
 SOCKET socket;
 SOCKADDR_STORAGE ClientAddr;
}PER_HANDLE_DATA, *LPPER_HANDLE_DATA;
 
// 定义全局变量
const int DefaultPort = 6000;  
vector < PER_HANDLE_DATA* > clientGroup;  // 记录客户端的向量组
 
HANDLE hMutex = CreateMutex(NULL, FALSE, NULL);
DWORD WINAPI ServerWorkThread(LPVOID CompletionPortID);
DWORD WINAPI ServerSendThread(LPVOID IpParam);
 
// 开始主函数
int main()
{
// 加载socket动态链接库
 WORD wVersionRequested = MAKEWORD(2, 2); // 请求2.2版本的WinSock库
 WSADATA wsaData; // 接收Windows Socket的结构信息
 DWORD err = WSAStartup(wVersionRequested, &wsaData);
 
 if (0 != err){ // 检查套接字库是否申请成功
  cerr << "Request Windows Socket Library Error!
";
  system("pause");
  return -1;
 }
 if(LOBYTE(wsaData.wVersion) != 2 || HIBYTE(wsaData.wVersion) != 2){// 检查是否申请了所需版本的套接字库
  WSACleanup();
  cerr << "Request Windows Socket Version 2.2 Error!
";
  system("pause");
  return -1;
 }
 
// 创建IOCP的内核对象
 /**
  * 需要用到的函数的原型：
  * HANDLE WINAPI CreateIoCompletionPort(
     *    __in   HANDLE FileHandle,  // 已经打开的文件句柄或者空句柄，一般是客户端的句柄
     *    __in   HANDLE ExistingCompletionPort, // 已经存在的IOCP句柄
     *    __in   ULONG_PTR CompletionKey, // 完成键，包含了指定I/O完成包的指定文件
     *    __in   DWORD NumberOfConcurrentThreads // 真正并发同时执行最大线程数，一般推介是CPU核心数*2
     * );
  **/
 HANDLE completionPort = CreateIoCompletionPort( INVALID_HANDLE_VALUE, NULL, 0, 0);
 if (NULL == completionPort){ // 创建IO内核对象失败
  cerr << "CreateIoCompletionPort failed. Error:" << GetLastError() << endl;
  system("pause");
  return -1;
 }
 
// 创建IOCP线程--线程里面创建线程池
 
 // 确定处理器的核心数量
 SYSTEM_INFO mySysInfo;
 GetSystemInfo(&mySysInfo);
 
 // 基于处理器的核心数量创建线程
 for(DWORD i = 0; i < (mySysInfo.dwNumberOfProcessors * 2); ++i){
  // 创建服务器工作器线程，并将完成端口传递到该线程
  HANDLE ThreadHandle = CreateThread(NULL, 0, ServerWorkThread, completionPort, 0, NULL);
  if(NULL == ThreadHandle){
   cerr << "Create Thread Handle failed. Error:" << GetLastError() << endl;
  system("pause");
   return -1;
  }
  CloseHandle(ThreadHandle);
 }
 
// 建立流式套接字
 SOCKET srvSocket = socket(AF_INET, SOCK_STREAM, 0);
 
// 绑定SOCKET到本机
 SOCKADDR_IN srvAddr;
 srvAddr.sin_addr.S_un.S_addr = htonl(INADDR_ANY);
 srvAddr.sin_family = AF_INET;
 srvAddr.sin_port = htons(DefaultPort);
 int bindResult = bind(srvSocket, (SOCKADDR*)&srvAddr, sizeof(SOCKADDR));
 if(SOCKET_ERROR == bindResult){
  cerr << "Bind failed. Error:" << GetLastError() << endl;
  system("pause");
  return -1;
 }
 
// 将SOCKET设置为监听模式
 int listenResult = listen(srvSocket, 10);
 if(SOCKET_ERROR == listenResult){
  cerr << "Listen failed. Error: " << GetLastError() << endl;
  system("pause");
  return -1;
 }
 
// 开始处理IO数据
 cout << "本服务器已准备就绪，正在等待客户端的接入...
";
 
 // 创建用于发送数据的线程
 HANDLE sendThread = CreateThread(NULL, 0, ServerSendThread, 0, 0, NULL);
 
 while(true){
  PER_HANDLE_DATA * PerHandleData = NULL;
  SOCKADDR_IN saRemote;
  int RemoteLen;
  SOCKET acceptSocket;
 
  // 接收连接，并分配完成端，这儿可以用AcceptEx()
  RemoteLen = sizeof(saRemote);
  acceptSocket = accept(srvSocket, (SOCKADDR*)&saRemote, &RemoteLen);
  if(SOCKET_ERROR == acceptSocket){ // 接收客户端失败
   cerr << "Accept Socket Error: " << GetLastError() << endl;
   system("pause");
   return -1;
  }
  
  // 创建用来和套接字关联的单句柄数据信息结构
  PerHandleData = (LPPER_HANDLE_DATA)GlobalAlloc(GPTR, sizeof(PER_HANDLE_DATA)); // 在堆中为这个PerHandleData申请指定大小的内存
  PerHandleData -> socket = acceptSocket;
  memcpy (&PerHandleData -> ClientAddr, &saRemote, RemoteLen);
  clientGroup.push_back(PerHandleData);  // 将单个客户端数据指针放到客户端组中
 
  // 将接受套接字和完成端口关联
  CreateIoCompletionPort((HANDLE)(PerHandleData -> socket), completionPort, (DWORD)PerHandleData, 0);
 
  
  // 开始在接受套接字上处理I/O使用重叠I/O机制
  // 在新建的套接字上投递一个或多个异步
  // WSARecv或WSASend请求，这些I/O请求完成后，工作者线程会为I/O请求提供服务 
  // 单I/O操作数据(I/O重叠)
  LPPER_IO_OPERATION_DATA PerIoData = NULL;
  PerIoData = (LPPER_IO_OPERATION_DATA)GlobalAlloc(GPTR, sizeof(PER_IO_OPERATEION_DATA));
  ZeroMemory(&(PerIoData -> overlapped), sizeof(OVERLAPPED));
  PerIoData->databuff.len = 1024;
  PerIoData->databuff.buf = PerIoData->buffer;
  PerIoData->operationType = 0; // read
 
  DWORD RecvBytes;
  DWORD Flags = 0;
  WSARecv(PerHandleData->socket, &(PerIoData->databuff), 1, &RecvBytes, &Flags, &(PerIoData->overlapped), NULL);
 }
 
 system("pause");
 return 0;
}
 
// 开始服务工作线程函数
DWORD WINAPI ServerWorkThread(LPVOID IpParam)
{
 HANDLE CompletionPort = (HANDLE)IpParam;
 DWORD BytesTransferred;
 LPOVERLAPPED IpOverlapped;
 LPPER_HANDLE_DATA PerHandleData = NULL;
 LPPER_IO_DATA PerIoData = NULL;
 DWORD RecvBytes;
 DWORD Flags = 0;
 BOOL bRet = false;
 
 while(true){
  bRet = GetQueuedCompletionStatus(CompletionPort, &BytesTransferred, (PULONG_PTR)&PerHandleData, (LPOVERLAPPED*)&IpOverlapped, INFINITE);
  if(bRet == 0){
   cerr << "GetQueuedCompletionStatus Error: " << GetLastError() << endl;
   return -1;
  }
  PerIoData = (LPPER_IO_DATA)CONTAINING_RECORD(IpOverlapped, PER_IO_DATA, overlapped);
  
  // 检查在套接字上是否有错误发生
  if(0 == BytesTransferred){
   closesocket(PerHandleData->socket);
   GlobalFree(PerHandleData);
   GlobalFree(PerIoData);
   continue;
  }
  
  // 开始数据处理，接收来自客户端的数据
  WaitForSingleObject(hMutex,INFINITE);
  cout << "A Client says: " << PerIoData->databuff.buf << endl;
  ReleaseMutex(hMutex);
 
  // 为下一个重叠调用建立单I/O操作数据
  ZeroMemory(&(PerIoData->overlapped), sizeof(OVERLAPPED)); // 清空内存
  PerIoData->databuff.len = 1024;
  PerIoData->databuff.buf = PerIoData->buffer;
  PerIoData->operationType = 0; // read
  WSARecv(PerHandleData->socket, &(PerIoData->databuff), 1, &RecvBytes, &Flags, &(PerIoData->overlapped), NULL);
 }
 
 return 0;
}
 
 
// 发送信息的线程执行函数
DWORD WINAPI ServerSendThread(LPVOID IpParam)
{
 while(1){
  char talk[200];
  gets(talk);
  int len;
  for (len = 0; talk[len] != ''; ++len){
   // 找出这个字符组的长度
  }
  talk[len] = '
';
  talk[++len] = '';
  printf("I Say:");
  cout << talk;
  WaitForSingleObject(hMutex,INFINITE);
  for(int i = 0; i < clientGroup.size(); ++i){
   send(clientGroup[i]->socket, talk, 200, 0); // 发送信息
  }
  ReleaseMutex(hMutex); 
 }
 return 0;
}






//client
#include <iostream>
#include <cstdio>
#include <string>
#include <cstring>
#include <winsock2.h>
#include <Windows.h>
 
using namespace std;
 
#pragma comment(lib, "Ws2_32.lib")  // Socket编程需用的动态链接库
 
SOCKET sockClient;  // 连接成功后的套接字
HANDLE bufferMutex;  // 令其能互斥成功正常通信的信号量句柄
const int DefaultPort = 6000;
 
int main()
{
// 加载socket动态链接库(dll)
 WORD wVersionRequested;
 WSADATA wsaData; // 这结构是用于接收Wjndows Socket的结构信息的
 wVersionRequested = MAKEWORD( 2, 2 ); // 请求2.2版本的WinSock库
 int err = WSAStartup( wVersionRequested, &wsaData );
 if ( err != 0 ) { // 返回值为零的时候是表示成功申请WSAStartup
  return -1;
 }
 if ( LOBYTE( wsaData.wVersion ) != 2 || HIBYTE( wsaData.wVersion ) != 2 ) { // 检查版本号是否正确
  WSACleanup( );
  return -1; 
 }
 
// 创建socket操作，建立流式套接字，返回套接字号sockClient
  sockClient = socket(AF_INET, SOCK_STREAM, 0);
  if(sockClient == INVALID_SOCKET) { 
  printf("Error at socket():%ld
", WSAGetLastError()); 
        WSACleanup(); 
        return -1; 
      } 
 
// 将套接字sockClient与远程主机相连
 // int connect( SOCKET s,  const struct sockaddr* name,  int namelen);
 // 第一个参数：需要进行连接操作的套接字
 // 第二个参数：设定所需要连接的地址信息
 // 第三个参数：地址的长度
 SOCKADDR_IN addrSrv;
 addrSrv.sin_addr.S_un.S_addr = inet_addr("127.0.0.1");  // 本地回路地址是127.0.0.1; 
 addrSrv.sin_family = AF_INET;
 addrSrv.sin_port = htons(DefaultPort);
 while(SOCKET_ERROR == connect(sockClient, (SOCKADDR*)&addrSrv, sizeof(SOCKADDR))){
  // 如果还没连接上服务器则要求重连
  cout << "服务器连接失败，是否重新连接？（Y/N):";
  char choice;
  while(cin >> choice && (!((choice != 'Y' && choice == 'N') || (choice == 'Y' && choice != 'N')))){
   cout << "输入错误，请重新输入:";
   cin.sync();
   cin.clear();
  }
  if (choice == 'Y'){
   continue;
  }
  else{
   cout << "退出系统中...";
   system("pause");
   return 0;
  }
 }
 cin.sync();
 cout << "本客户端已准备就绪，用户可直接输入文字向服务器反馈信息。
";
 
 send(sockClient, "
Attention: A Client has enter...
", 200, 0);
 
 bufferMutex = CreateSemaphore(NULL, 1, 1, NULL); 
 
 DWORD WINAPI SendMessageThread(LPVOID IpParameter);
 DWORD WINAPI ReceiveMessageThread(LPVOID IpParameter);
 
 HANDLE sendThread = CreateThread(NULL, 0, SendMessageThread, NULL, 0, NULL);  
 HANDLE receiveThread = CreateThread(NULL, 0, ReceiveMessageThread, NULL, 0, NULL);  
 
    
    WaitForSingleObject(sendThread, INFINITE);  // 等待线程结束
 closesocket(sockClient);
 CloseHandle(sendThread);
 CloseHandle(receiveThread);
 CloseHandle(bufferMutex);
 WSACleanup(); // 终止对套接字库的使用
 
 printf("End linking...
");
 printf("
");
 system("pause");
 return 0;
}
 
 
DWORD WINAPI SendMessageThread(LPVOID IpParameter)
{
 while(1){
  string talk;
  getline(cin, talk);
  WaitForSingleObject(bufferMutex, INFINITE);  // P（资源未被占用）  
  if("quit" == talk){
   talk.push_back('');
   send(sockClient, talk.c_str(), 200, 0);
   break;
  }
  else{
   talk.append("
");
  }
  printf("
I Say:("quit"to exit):");
  cout << talk;
  send(sockClient, talk.c_str(), 200, 0); // 发送信息
  ReleaseSemaphore(bufferMutex, 1, NULL);  // V（资源占用完毕） 
 }
 return 0;
}
 
 
DWORD WINAPI ReceiveMessageThread(LPVOID IpParameter)
{
 while(1){ 
  char recvBuf[300];
  recv(sockClient, recvBuf, 200, 0);
  WaitForSingleObject(bufferMutex, INFINITE);  // P（资源未被占用）  
 
  printf("%s Says: %s", "Server", recvBuf);  // 接收信息
  
  ReleaseSemaphore(bufferMutex, 1, NULL);  // V（资源占用完毕） 
 }
 return 0;
}